From 554188e88e7dd6d2c8dee830a8dfb8cfff914e6e Mon Sep 17 00:00:00 2001
From: Trent Nelson <trent@solana.com>
Date: Tue, 17 Dec 2019 19:00:12 -0500
Subject: [PATCH] Colo - Node install scripts missing latest user requests
 (#7540)

* Enable user GPU profiling while installing CUDA

* Install heaptrack
---
 net/datacenter-node-install/setup-cuda.sh      | 3 +++
 net/datacenter-node-install/setup-dc-node-1.sh | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/net/datacenter-node-install/setup-cuda.sh b/net/datacenter-node-install/setup-cuda.sh
index f2fbbfb967..72190d307d 100644
--- a/net/datacenter-node-install/setup-cuda.sh
+++ b/net/datacenter-node-install/setup-cuda.sh
@@ -48,3 +48,6 @@ apt install -y gcc make dkms
 for rf in "${RUN_FILES[@]}"; do
   sh "$rf" --silent --driver --toolkit
 done
+
+# Allow normal users to use CUDA profiler
+echo 'options nvidia "NVreg_RestrictProfilingToAdminUsers=0"' > /etc/modprobe.d/nvidia-enable-user-profiling.conf
diff --git a/net/datacenter-node-install/setup-dc-node-1.sh b/net/datacenter-node-install/setup-dc-node-1.sh
index 1c51e7b0b1..4c69400a6c 100644
--- a/net/datacenter-node-install/setup-dc-node-1.sh
+++ b/net/datacenter-node-install/setup-dc-node-1.sh
@@ -31,7 +31,7 @@ EOF
 
 apt install -y build-essential pkg-config clang cmake sysstat perf \
   linux-generic-hwe-18.04-edge linux-tools-generic-hwe-18.04-edge \
-  iftop
+  iftop heaptrack
 
 "$HERE"/../scripts/install-docker.sh
 usermod -aG docker "$SETUP_USER"