Profiler (TracerViewer in specific) can not load planes if their names do not contain CPU/GPU/TPU

[Niz13]

Hello,

I've written a new profiler for a device that isn't CPU/GPU/TPU.

The profiler inherits from ProfilerInterface.
In CollectData(XSpace* space), I'm trying to add a new plane to represent this new device.
Unless I name the plane "/device:GPU:0" or "/host:CPU", the plane does not show up in the TracerViewer of the profiler.

I even renamed the CPU plane to something else, and it stopped showing up.

Somewhere in the Tensorboard code there must be a hard-coded check that ignores anything but CPU/GPU/TPU.

[bileschi]

Routing to profiler experts - @ckluk , can you help direct us in a fruitful direction here? Thanks.

[trisolaran]

tensorflow/core/profiler/convert/xplane_to_trace_events.cc:133
we are looking at specific prefix.
ideally we should define some prefix for custom device. but it is not done yet.

[trisolaran]

we will add a prefix that you can use. tensorflow::profiler::kCustomPlanePrefix which will be defined in xplane_schema.h
we will convert events in these planes to trace viewer.
depends on how you build your tf/tb, or which version you use (nightly/stable), this will be available to you at certain point of time in future.
before that, you can use kGpuPlanePrefix as WAR.