TPU Support

[rdyro]

I'm a big fan of your tool and I was hoping we could add TPU support (I work for Google).

I wrote a prototype here: https://github.com/rdyro/nvtop-with-tpu

My current strategy involves calling Python tpu-info via popen and sscanf parsing the output.

Would you be open to upstreaming the TPU support in some version (perhaps a Python-less one eventually)?

For Python-less access to TPU data, the TPU metrics are available through a GRPC server, but a C client dependency might be heavy. Another possibility is a C++ or Go library loaded via dlopen .

[Syllo]

Hello,

I'm glad nvtop is appreciated 😉

Sure thing, that would be great!

I'd rather if we stick to loading a library dynamically (the way it's done for other devices, e.g. à la libnvidia-ml). But I'm not against having conditional build for TPUs using a sub-module (or external lib) that we would link with.

[rdyro]

Great, thanks! I'll work on a prototype with a dynamic library and I can make a PR with actual code.

[rdyro]

Hey, I made a PR here. Any feedback on C and about the cmake approach would be greatly appreciated!