[SWDEV-539119] [release/2.8] Add fast_tanh support (#2484)

jataylo · web-flow · commit 4febbd8a99f7 · 2025-08-12T13:42:38.000-07:00
Perf improvement for triton tanh
diff --git a/.ci/docker/ci_commit_pins/triton.txt b/.ci/docker/ci_commit_pins/triton.txt
@@ -1 +1 @@
-711e2a92522e0a9921ce58ae658571ca55c49b97
+6581064e5be2ac4ab0c198c1bb0e533a1900bbe3
diff --git a/torch/_inductor/codegen/triton.py b/torch/_inductor/codegen/triton.py
@@ -1232,7 +1232,7 @@ def tan(x):
     @staticmethod
     @maybe_upcast_float32()
     def tanh(x):
-        return f"libdevice.tanh({x})"
+        return f"libdevice.fast_tanhf({x})"
 
     @staticmethod
     @maybe_upcast_float32()
