Hi all,
I profile a model for inference with Torchscript. The profiling info shows method names like kernel 1, kernel 3. Why? How could I get profiling info with normal kernel names such conv, relu, etc.
Thanks.
Du
Hi all,
I profile a model for inference with Torchscript. The profiling info shows method names like kernel 1, kernel 3. Why? How could I get profiling info with normal kernel names such conv, relu, etc.
Thanks.
Du
Are you using the nvtx markers?
If so, you can pass your desired names to these ranges, which will be shown in the profiler later.