AI Dynamics

Global AI News Aggregator

PyTorch NVTX Profiler: Emit Markers for Operations

By

@soumithchintala

–

torch.autograd.profiler.emit_nvtx.
It emits nvtx markers for every torch op. https://
docs.pytorch.org/docs/stable/au
tograd.html#torch.autograd.profiler.emit_nvtx
…

→ View original post on X — @soumithchintala,

Commentaires

Leave a Reply

Your email address will not be published. Required fields are marked *

MORE ARTICLES