Update

[ghstack-poisoned]
pytorch · May 24, 2024 · 4efbed7 · 4efbed7
1 parent 26bd3b9
commit 4efbed7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/torch/_inductor/codegen/cpp.py b/torch/_inductor/codegen/cpp.py
@@ -1071,6 +1071,12 @@ def exp(x):
     def exp2(x):
         return f"{x}.exp2()"
 
+    @staticmethod
+    def expm1(x):
+        # decompose for a better performance
+        vec_one = f"decltype({x})(1)"
+        return f"{x}.exp() - {vec_one}"
+
     @staticmethod
     def erf(x):
         return f"{x}.erf()"