Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Skip float64 test_nextafter on TPU. #21108

Merged
merged 1 commit into from
May 9, 2024

Conversation

justinjfu
Copy link
Collaborator

Skip float64 test for test_nextafter on TPU. X64 lowering is not currently enabled on the backend.

Error: https://github.com/google/jax/actions/runs/8958798888/job/24603466593

...
jax/_src/compiler.py:333: in compile_or_get_cached
    return _compile_and_write_cache(
jax/_src/compiler.py:500: in _compile_and_write_cache
    executable = backend_compile(
jax/_src/profiler.py:335: in wrapper
    return func(*args, **kwargs)
jax/_src/compiler.py:238: in backend_compile
    return backend.compile(built_c, compile_options=options)
E   jaxlib.xla_extension.XlaRuntimeError: UNIMPLEMENTED: While rewriting computation to not contain X64 element types, XLA encountered an HLO for which this rewriting is not implemented: %bitcast-convert.45 = s64[4]{0} bitcast-convert(f64[4]{0} %broadcast.39), metadata={op_name="jit(wrapped)/jit(main)/while/body/nextafter" source_file="/home/runner/actions-runner/_work/jax/jax/tests/pallas/pallas_test.py" source_line=1551}

@justinjfu justinjfu requested a review from superbobry May 7, 2024 21:48
@google-ml-butler google-ml-butler bot added kokoro:force-run pull ready Ready for copybara import and testing labels May 7, 2024
@justinjfu justinjfu merged commit eb0b1b0 into google:main May 9, 2024
13 checks passed
@justinjfu justinjfu deleted the skip_pallas_test_64 branch May 10, 2024 17:30
This pull request was closed.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
pull ready Ready for copybara import and testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants