pytorch
a4486fe7 - [ROCm] Print name irrespective of seq number assignment for roctx traces (#45229)

Commit
4 years ago
[ROCm] Print name irrespective of seq number assignment for roctx traces (#45229) Summary: Recent changes to the seq_num correlation behavior in profiler (PR https://github.com/pytorch/pytorch/issues/42565) has changed the behavior for emit_nvtx(record_shapes=True) which doesn't print the name of the operator properly. Created PR to dump out the name in roctx traces, irrespective of the sequence number assigned only for ROCm. cc: jeffdaily sunway513 Pull Request resolved: https://github.com/pytorch/pytorch/pull/45229 Reviewed By: zou3519 Differential Revision: D23932902 Pulled By: albanD fbshipit-source-id: c782667ff002b70b51f1cc921afd1b1ac533b39d
Author
Parents
Loading