Unfortunately, the non-unified profile is not too useful. I am mostly interested in knowing which kernel you are running. Perhaps you could run https://pytorch.org/docs/stable/autograd.html#profiler 2 instead?