Skip the test for stream_executor like all the other tests in this file

PiperOrigin-RevId: 496107041
google · Dec 17, 2022 · b1415bb · b1415bb
1 parent e6e1836
commit b1415bb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/debugging_primitives_test.py b/tests/debugging_primitives_test.py
@@ -812,6 +812,10 @@ def f2(x):
       self.assertEqual(output(), "140\n")
 
   def test_nested_pjit_debug_print(self):
+    if xla_bridge.get_backend().runtime_type == 'stream_executor':
+      raise self.skipTest(
+          'Host callback not supported for runtime type: stream_executor.')
+
     if not jax.config.jax_array:
       self.skipTest("This test only works with jax.Array.")