update _inplace_to_out_place

Lightning-AI · Dec 12, 2024 · 510d925 · 510d925
1 parent e3f9443
commit 510d925
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/thunder/torch/__init__.py b/thunder/torch/__init__.py
@@ -1910,6 +1910,8 @@ def tanhshrink(a: TensorLike, /) -> TensorLike:
     return a - tanh(a)
 
 
+_inplace_to_out_of_place[tanhshrink] = tanhshrink, -1
+
 #
 # Elementwise binary operations
 #