Skip to content

Commit a879955

Browse files
Copilotcaic99
andcommitted
Update documentation to reflect new PyTorch profiler API
Co-authored-by: caic99 <78061359+caic99@users.noreply.github.com>
1 parent 737878f commit a879955

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

PYTORCH_PROFILER_INTEGRATION.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ ls -la ./profiler_results/pytorch_profiler_trace.json
2424

2525
## Implementation Details
2626

27-
The profiler uses PyTorch's `torch::autograd::profiler::RecordProfile` and automatically:
27+
The profiler uses PyTorch's modern `torch::profiler` API and automatically:
2828
- Creates the output directory if it doesn't exist
2929
- Profiles all forward pass operations in DeepPotPT and DeepSpinPT
3030
- Saves profiling results to a JSON file when the object is destroyed

0 commit comments

Comments
 (0)