From d37ae923c37152350d168c2fc2520bc93441bc0a Mon Sep 17 00:00:00 2001
From: Penporn Koanantakool <penporn@google.com>
Date: Mon, 6 May 2024 14:58:30 -0700
Subject: [PATCH] Rolling back PR #12076: [XLA:CPU][oneDNN] AVX512
 vectorization Reason: Broke JAX Arm CI ``` E     AssertionError:
 [b"'-prefer-256-bit' is not a recognized feature for this target (ignoring
 feature)" ```

Reverts dea04758ca33fced4ad7128110cde7d792da37e4

PiperOrigin-RevId: 631195048
---
 xla/service/cpu/simple_orc_jit.cc | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/xla/service/cpu/simple_orc_jit.cc b/xla/service/cpu/simple_orc_jit.cc
index cda9d70138bed..21ce0280906c1 100644
--- a/xla/service/cpu/simple_orc_jit.cc
+++ b/xla/service/cpu/simple_orc_jit.cc
@@ -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()