TPU VM + Pytorch/XLA profiling shows zero device compute

[coleridge72]

I've been using the profiler on a TPU VM with Pytorch following this guide

The overview page shows zero "device compute" in the graph and summary. I notice this is actually also the case in the screenshot of the guide. The trace viewer does show activity happening on the cores though.

Could this be a bug? or maybe I am just missing something.

Thanks

Specs:

torch==1.10.0
torch-xla==1.10
torchvision==0.11.1
tensorboard==2.7.0
tensorboard-data-server==0.6.1
tensorboard-plugin-profile==2.5.0
tensorboard-plugin-wit==1.8.0