the xprof_kernel_demo can't see the runtime perf counter

[pathfinder-pf]

ref: https://github.com/openxla/xprof/blob/master/docs/xprof_kernel_demo.ipynb

hardware
Libtpu version: 0.0.40
Accelerator type: 7x

TPU Chips

Chip	Type	Devices	PID
/dev/vfio/3	TPU7x chip	2	None
/dev/vfio/5	TPU7x chip	2	None
/dev/vfio/6	TPU7x chip	2	None
/dev/vfio/7	TPU7x chip	2	None

software version

drjax                          0.1.4
flax                           0.12.7
jax                            0.10.0
jaxlib                         0.10.0
jaxtyping                      0.3.3
libtpu                         0.0.40

[Becker-ZH]

Hi @pathfinder-pf , thanks for trying out the colab and kernel debugging features!

Perf counter collection needs a newer version of libtpu. Could you please update your environment to the latest stable releases if possible? (try libTPU 0.0.42 or later/nightly)

• If use JAX, the stable jax should already pinned libTPU to be 0.0.42.* (https://github.com/jax-ml/jax/blob/main/setup.py)
• XProf version for frontend visualization: 2.22.3

Thank you!

[pathfinder-pf]

thanks, i will try it later @Becker-ZH

[pathfinder-pf]

why there are three MXU line on tpu 7x?