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

Remove xla_cpu_enable_mlir_lowering

Browse Source 
This is now replaced with --xla_cpu_use_xla_runtime

PiperOrigin-RevId: 500955199
This commit is contained in:
Tres Popp 2023-01-10 04:18:30 -08:00 committed by jax authors
parent b5be92e449
commit edd09c65f3
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/_src/test_util.py
View File

@ -309,7 +309,7 @@ def skip_on_xla_cpu_mlir(test_method):
@functools.wraps(test_method)
def test_method_wrapper(self, *args, **kwargs):
xla_flags = os.getenv('XLA_FLAGS') or ''
if '--xla_cpu_use_xla_runtime' in xla_flags or '--xla_cpu_enable_mlir_lowering' in xla_flags:
if '--xla_cpu_use_xla_runtime' in xla_flags:
test_name = getattr(test_method, '__name__', '[unknown test]')
raise unittest.SkipTest(
f'{test_name} not supported on XLA:CPU MLIR')