make comment of _inplace_to_out_of_place verbose (#1556)

Lightning-AI · Dec 16, 2024 · 3913b02 · 3913b02
1 parent 458ec6a
commit 3913b02
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/thunder/torch/__init__.py b/thunder/torch/__init__.py
@@ -91,7 +91,8 @@
 # torch operation definitions
 #
 
-# in-place sym -> out-of-place (= functional) sym with index of `inplace` argument
+# in-place sym -> out-of-place (= functional) sym with index of `inplace: bool` argument
+# If an in-place op doesn't have `inplace: bool` argument, set -1.
 _inplace_to_out_of_place: dict[Callable, tuple[Callable, int]] = {}