mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-23 15:46:09 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm-main
rocm_jax/build/collect-profile-requirements.txt

5 lines
101 B
Plaintext
Raw Permalink Normal View History

profiler_test.py fixes and add coverage to Cloud TPU CI * Add deps to test requirements, including in new `collect-profile-requirements.txt` (to avoid adding tensorflow to `test-requirements.txt`). * Use correct Python executable `ProfilerTest.test_remote_profiler` (`python` sometimes defaults to python2) * Run computations for longer in `ProfilerTest.test_remote_profiler`, othewise `collect_profile` sometimes misses it.
2023-06-09 17:58:42 +00:00
tensorflow
tensorboard-plugin-profile
# Needed for the profile plugin to work without error
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 11:56:09 -07:00
protobuf
Reference in New Issue Copy Permalink