From 7013588ebeefaba105849098af07d3ef7208eaec Mon Sep 17 00:00:00 2001
From: Wesley Maxey <71408887+wmaxey@users.noreply.github.com>
Date: Thu, 25 Apr 2024 15:37:48 -0700
Subject: [PATCH] Add NVTX to Windows build. (#277)

---
 windows/image/installers/install-cuda.ps1 | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/windows/image/installers/install-cuda.ps1 b/windows/image/installers/install-cuda.ps1
index 3cbb04c1..1c2f34c9 100644
--- a/windows/image/installers/install-cuda.ps1
+++ b/windows/image/installers/install-cuda.ps1
@@ -59,7 +59,8 @@ $cudaComponents =
     "cupti_$componentTag",
     "nvrtc_$componentTag",
     "nvrtc_dev_$componentTag",
-    "nvml_dev_$componentTag"
+    "nvml_dev_$componentTag",
+    "nvtx_$componentTag"
     
 Invoke-WebRequest -Uri "$cudaVersionUrl" -OutFile "./cuda_network.exe" -UseBasicParsing
 Start-Process -Wait -PassThru -FilePath .\cuda_network.exe -ArgumentList "-s $cudaComponents"