Skip to content

Commit

Permalink
[xla:cpu] Temporarily disable fast F64 Tanh due to incorrect results.
Browse files Browse the repository at this point in the history
jax-ml/jax#23590

PiperOrigin-RevId: 674332130
  • Loading branch information
penpornk authored and Google-ML-Automation committed Sep 13, 2024
1 parent f4f5e76 commit 8fcf359
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion xla/service/cpu/llvm_ir_runtime.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down

0 comments on commit 8fcf359

Please sign in to comment.