pytorch
7551136b - Add NVTX markers that dump additional information for nvprim_nvfuser Dynamo graphs (#88259)

Commit
2 years ago
Add NVTX markers that dump additional information for nvprim_nvfuser Dynamo graphs (#88259) dump information on graphs that NVFuser JIT compiles: - the markers show the list of ops, args, and inputs that make up the graph also dumps information on FX nodes that are not touched by NVFuser: - the markers show the op, name, and arg list of the node Pull Request resolved: https://github.com/pytorch/pytorch/pull/88259 Approved by: https://github.com/IvanYashchuk, https://github.com/jjsjann123, https://github.com/mruberry
Author
Committer
Parents
Loading