fix tracer tensor error

albanD · Jun 16, 2024 · 2d790cd · 2d790cd
1 parent 24d5683
commit 2d790cd
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tracer_tensor.py b/tracer_tensor.py
@@ -173,9 +173,9 @@ def test_basic(self):
             str(g.graph),
             """\
 graph():
-    %arg_1 : [#users=1] = placeholder[target=arg_1]
-    %arg_2 : [#users=1] = placeholder[target=arg_2]
-    %add_tensor : [#users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%arg_1, %arg_2), kwargs = {})
+    %arg_1 : [num_users=1] = placeholder[target=arg_1]
+    %arg_2 : [num_users=1] = placeholder[target=arg_2]
+    %add_tensor : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%arg_1, %arg_2), kwargs = {})
     return add_tensor""",
         )
 
@@ -186,9 +186,9 @@ def test_constant(self):
             str(g.graph),
             """\
 graph():
-    %arg_1 : [#users=1] = placeholder[target=arg_1]
-    %_tensor_constant0 : [#users=1] = get_attr[target=_tensor_constant0]
-    %add_tensor : [#users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%_tensor_constant0, %arg_1), kwargs = {})
+    %arg_1 : [num_users=1] = placeholder[target=arg_1]
+    %_tensor_constant0 : [num_users=1] = get_attr[target=_tensor_constant0]
+    %add_tensor : [num_users=1] = call_function[target=torch.ops.aten.add.Tensor](args = (%_tensor_constant0, %arg_1), kwargs = {})
     return add_tensor""",
         )