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

Added debug_callback to the list of exclusions in jax2tf/tests/primitives_test.py

Browse Source 
PiperOrigin-RevId: 641149152
This commit is contained in:
Sergei Lebedev 2024-06-07 00:00:36 -07:00 committed by jax authors
parent c01c98400d
commit 5d6413cecc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/experimental/jax2tf/tests/primitives_test.py
View File

@ -179,7 +179,7 @@ class JaxPrimitiveTest(tf_test_util.JaxToTfTestCase):
# TODO: Remove once tensorflow is 2.10.0 everywhere.
if p.name == "optimization_barrier":
continue
if p.name == "debug_callback":
if p.name == "debug_callback" or p.name == "debug_print":
# TODO(sharadmv,necula): enable debug callbacks in TF
continue
if p.name == "pallas_call":