Merge pull request #52 from msaroufim/msaroufim/removefunctorchtest

Fix functorch import
albanD · Jun 16, 2024 · ec47458 · ec47458
2 parents 0141210 + 2d790cd
commit ec47458
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/run_test.py b/run_test.py
@@ -28,7 +28,7 @@
 from torch.testing._internal.common_utils import run_tests
 from bug_zoo import BugZoo
 from empty_tensor import EmptyTensorTest
-from functorch import FunctorchTest
+from functorch_test import FunctorchTest
 from inner_autograd_tensor import InnerAutogradTensorTest
 from logging_mode import TracerTensorTest
 from negative_tensor import NegativeTensorTest

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""",
         )