Skip to content

Commit 45fcb69

Browse files
Alok-Joshiclaude
andcommitted
Run a plain torch script under ncu in the Windows repro step
Launch a minimal torch script directly with ncu, without nsight-python, to tell whether the shutdown abort needs nsight-python's attach mode or occurs whenever Nsight Compute profiles a torch process. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Signed-off-by: Alok Joshi <aljoshi@nvidia.com>
1 parent 6576b00 commit 45fcb69

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

‎.github/workflows/ci.yaml‎

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -206,6 +206,16 @@ jobs:
206206
# The repro: one profiled kernel.
207207
python repro_nsight.py
208208
Write-Host "=== exit $LASTEXITCODE | torch on cuda, nsight.analyze.kernel"
209+
# No nsight-python at all: ncu launches a plain torch script itself.
210+
$app = @(
211+
'import torch',
212+
'a = torch.randn(1024, 1024, device="cuda")',
213+
'b = a @ a',
214+
'print("simple app done", b.sum().item())'
215+
)
216+
Set-Content -Path simple_python_app.py -Value $app
217+
ncu -f -o simple_python_app python simple_python_app.py
218+
Write-Host "=== exit $LASTEXITCODE | ncu python simple_python_app.py"
209219
exit 0
210220
- name: "Running Tests"
211221
shell: pwsh

0 commit comments

Comments
 (0)