Skip to content

Commit

Permalink
Fix ncu command
Browse files Browse the repository at this point in the history
  • Loading branch information
msaroufim committed Nov 18, 2024
1 parent 46aa35a commit a292395
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions .github/workflows/nvidia_workflow.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit a292395

Please sign in to comment.