fix wrongly exchanged variable

Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com>
pytorch · May 12, 2023 · ef7e9c6 · ef7e9c6
1 parent db8e514
commit ef7e9c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgen/api/autograd.py b/torchgen/api/autograd.py
@@ -326,7 +326,7 @@ def is_reference_for_foreach(
         f.func.name.name.base.split("_foreach_")[-1] == function_schema.name.name.base
         and (
             not function_schema.name.name.inplace
-            or str(function_schema.name) in _foreach_with_inplace_ref
+            or str(f.name) in _foreach_with_inplace_ref
         )
         and all(
             ref_arg.type in (arg.type, getattr(arg.type, "elem", None))