tensorflow / tensorflow/tensorboard

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

Open
#5,131 3 comments 0 reactions 2 assignees View on GitHub

@bileschi is already working on this.

Since Jul 15, 2021.

core:backend plugin:profile stat:awaiting response theme:usability type:feature
Dominant language
TypeScript
Stars
7.2k
Forks
1.7k
Avg merge
4d 22h
Merged PRs (30d)
1

Description

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.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.