search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.49k stars 480 forks source link

add profiler hints in paralloader #8244

Closed JackCaoG closed 1 month ago

JackCaoG commented 1 month ago

I verified that these traces show up in xprof.