mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump ProfilerTest.test_remote_profiler to run computations for 10 seconds.

Browse Source 
I hate to make a unit test run for this long, but I'm still seeing
cases where the remote profiler will miss the computations and hang
forever. (Why is it so slow?) This should hopefully be enough that it always catches some
computations.
This commit is contained in:
Skye Wanderman-Milne 2023-07-27 23:54:42 +00:00
parent 7c0ef8660d
commit 11ff93951c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/profiler_test.py
View File

@ -236,7 +236,7 @@ class ProfilerTest(unittest.TestCase):
thread_profiler.start()
start_time = time.time()
y = jnp.zeros((5, 5))
while time.time() - start_time < 5:
while time.time() - start_time < 10:
y = jnp.dot(y, y)
jax.profiler.stop_server()
thread_profiler.join()