mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00
Remove XLA tanh fix cherry-pick, to avoid CI breakages when the XLA commit is
bumped.
This commit is contained in:
parent
8c84e1637e
commit
4fac852a90
14
third_party/xla/tanh.patch
vendored
14
third_party/xla/tanh.patch
vendored
@ -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);
|
3
third_party/xla/workspace.bzl
vendored
3
third_party/xla/workspace.bzl
vendored
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user