I am using the model provided by Torchvision.
Is it possible only with NVTX?
I use nvprof and I could know whether the cudaDeviceSynchronize(), cudaStreamSynchronize() function was called, but it’s not known who is calling synchronize().