From 3913b02e96674c026fbce090ea735109f185d639 Mon Sep 17 00:00:00 2001 From: Masaki Kozuki Date: Tue, 17 Dec 2024 01:08:27 +0900 Subject: [PATCH] make comment of `_inplace_to_out_of_place` verbose (#1556) --- thunder/torch/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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]] = {}