Remove XLA tanh fix cherry-pick, to avoid CI breakages when the XLA commit is

bumped.
This commit is contained in:
Peter Hawkins 2024-09-16 20:05:25 +00:00
parent 8c84e1637e
commit 4fac852a90
2 changed files with 0 additions and 17 deletions

View File

@ -1,14 +0,0 @@
diff --git a/xla/service/cpu/llvm_ir_runtime.cc b/xla/service/cpu/llvm_ir_runtime.cc
index 89b40b915caa3..25541c16bfd61 100644
--- a/xla/service/cpu/llvm_ir_runtime.cc
+++ b/xla/service/cpu/llvm_ir_runtime.cc
@@ -410,7 +410,8 @@ void RewriteIRRuntimeFunctions(llvm::Module* module,
rewrite_calls(kTanhV8F32SymbolName, GenerateVF32Tanh, /*vector_width=*/8);
rewrite_calls(kTanhV16F32SymbolName, GenerateVF32Tanh, /*vector_width=*/16);
- rewrite_calls("tanh", GenerateVF64Tanh, /*vector_width=*/1);
+ // TODO(penporn): Re-enable after fixing JAX issue #23590.
+ // rewrite_calls("tanh", GenerateVF64Tanh, /*vector_width=*/1);
rewrite_calls("expf", GenerateVF32Exp, /*vector_width=*/1);
rewrite_calls("llvm.exp.f32", GenerateVF32Exp, /*vector_width=*/1);

View File

@ -30,9 +30,6 @@ def repo():
sha256 = XLA_SHA256,
strip_prefix = "xla-{commit}".format(commit = XLA_COMMIT),
urls = tf_mirror_urls("https://github.com/openxla/xla/archive/{commit}.tar.gz".format(commit = XLA_COMMIT)),
patch_file = [
"//third_party/xla:tanh.patch",
],
)
# For development, one often wants to make changes to the TF repository as well