rocm_jax/collect-profile-requirements.txt at rocm-main - rocm_jax - Gitea For EOELAB

mirrors/rocm_jax

mirror of https://github.com/ROCm/jax.git synced 2025-04-14 10:56:06 +00:00

Skye Wanderman-Milne 6c909760d5 Cloud TPU CI: make sure we update test deps and upgrade protobuf version

`profiler_test.py:ProfilerTest.test_remote_profiler` fails with the
protobuf upgrade. However, I was seeing mysterious hangs without this,
and in general I think we should be testing with up-to-date deps given
that we don't pin. I'm gonna continue working on getting the Cloud TPU
CI green.

2023-07-19 16:17:47 -07:00

5 lines

101 B

Plaintext

Raw Permalink Blame History

 tensorflow
 tensorboard-plugin-profile
 # Needed for the profile plugin to work without error
 protobuf