diff --git a/thunder/torch/__init__.py b/thunder/torch/__init__.py index 57b64bfee0..38d0353c1f 100644 --- 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]] = {}