Open mahmoodn opened 1 year ago
That kernel might be part of the TensorRT library, which ships in binary form.
Thanks for the reply. I have recently encountered a problem with resnet50 and Nvbit which I have raised the issue here. I think you also have worked with these two in your tests with MLPerf benchmarks. Have you seen a similar thing before? I have stuck at that problem and tools are closed source.
Hi Using nsight compute, I see a kernel named
res2_sm_80_kernel
for resnet50.The command is
I searched the code to see the kernel code, but it seems that this name is part of another Nvidia library. Also with the name
sm_80
I expect that this name has been formed by an engine, so this complete name doesn't exist in the libraries.I also wasn't able to find the higher level C++ function that invokes this kernel. Any idea about how to find details of this kernel?