From a2923951221ab342d27e4c9ccccf9290aad752fb Mon Sep 17 00:00:00 2001
From: Mark Saroufim <marksaroufim@meta.com>
Date: Mon, 18 Nov 2024 11:27:00 -0800
Subject: [PATCH] Fix ncu command

---
 .github/workflows/nvidia_workflow.yml | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/.github/workflows/nvidia_workflow.yml b/.github/workflows/nvidia_workflow.yml
index e218ad0..6cdc913 100644
--- a/.github/workflows/nvidia_workflow.yml
+++ b/.github/workflows/nvidia_workflow.yml
@@ -45,14 +45,13 @@ jobs:
           echo "Triton detected, installing triton"
           pip install triton
         fi
-
     - name: Run script with NSight Compute profiling
       run: |
         # First run normally and capture output
         python "${{ github.event.inputs.filename }}" > training.log 2>&1
         
         # Then run with NSight Compute profiling
-        ncu --set full --export ncu_profile "${{ github.event.inputs.filename }}" > ncu.log 2>&1
+        ncu --set full --export ncu_profile $(which python) "${{ github.event.inputs.filename }}" > ncu.log 2>&1
 
     - name: Upload training artifacts
       uses: actions/upload-artifact@v3