torch_trace
Profile Python scripts with torch.profiler to identify GPU kernels and aten ops by time and input shapes; use it first when Nsight is unavailable.
Instructions
Run a Python script under torch.profiler in the user's interpreter. Returns kernels
(real GPU kernels by device time, pct over total kernel time) and ops (aten ops with
input shapes, inclusive time). Works without Nsight installed; use this first.
Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| cwd | No | ||
| top | No | ||
| args | No | ||
| script | Yes | ||
| timeout | No |