Skip to content

Commit

Permalink
Rolling back PR openxla#12076: [XLA:CPU][oneDNN] AVX512 vectorization
Browse files Browse the repository at this point in the history
Reason: Broke JAX Arm CI
```
E     AssertionError: [b"'-prefer-256-bit' is not a recognized feature for this target (ignoring feature)"
```

Reverts dea0475

PiperOrigin-RevId: 631195048
  • Loading branch information
penpornk authored and copybara-github committed May 6, 2024
1 parent 33ceb95 commit d37ae92
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions xla/service/cpu/simple_orc_jit.cc
Original file line number Diff line number Diff line change
Expand Up @@ -299,10 +299,6 @@ SimpleOrcJIT::InferTargetMachineForJIT(
const llvm::TargetOptions& target_options,
llvm::CodeGenOptLevel opt_level) {
std::vector<std::string> attrs = DetectMachineAttributes();
// Default preference is 256-bit vectorization because of the attribute
// `+prefer-256-bit`. Drop `prefer-256-bit` from the attributes by negation
// for higher target machine features, for example, avx-512 vectorization.
attrs.push_back("-prefer-256-bit");
llvm::SmallVector<std::string, 0> llvm_attrs(attrs.begin(), attrs.end());
std::unique_ptr<llvm::TargetMachine> target_machine(
llvm::EngineBuilder()
Expand Down

0 comments on commit d37ae92

Please sign in to comment.