[xla:cpu] Disable SVE LLVM codegen by default on AArch64 CPUs. #23931
+33
−1
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
[xla:cpu] Disable SVE LLVM codegen by default on AArch64 CPUs.
There are many missing SVE lowerings in LLVM, especially for bf16 type. This causes program termination on SVE-available machines such as Google Axion. Example error:
Since most AArch64 machines still use 128-bit registers, SVE and NEON shouldn't have significant performance difference, so we disable SVE codegen in public builds for the time being.
After JAX uses an XLA commit that has changes from this PR, the following JAX tests will pass on Axion:
Add
--test_env=XLA_FLAGS=--xla_cpu_max_isa=""
to the options to get the errors back.