Skip to content

Commit f7921f4

Browse files
jataylotvukovic-amd
authored andcommitted
[SWDEV-539119] [release/2.8] Add fast_tanh support (#2484)
Perf improvement for triton tanh
1 parent 0da7d02 commit f7921f4

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
711e2a92522e0a9921ce58ae658571ca55c49b97
1+
6581064e5be2ac4ab0c198c1bb0e533a1900bbe3

torch/_inductor/codegen/triton.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1232,7 +1232,7 @@ def tan(x):
12321232
@staticmethod
12331233
@maybe_upcast_float32()
12341234
def tanh(x):
1235-
return f"libdevice.tanh({x})"
1235+
return f"libdevice.fast_tanhf({x})"
12361236

12371237
@staticmethod
12381238
@maybe_upcast_float32()

0 commit comments

Comments
 (0)