diff --git a/test/_nvfuser/__init__.py b/test/_nvfuser/__init__.py new file mode 100644 index 00000000000..e69de29bb2d diff --git a/test/_nvfuser/test_dynamo.py b/test/_nvfuser/test_dynamo.py new file mode 120000 index 00000000000..140d1d845e7 --- /dev/null +++ b/test/_nvfuser/test_dynamo.py @@ -0,0 +1 @@ +../../third_party/nvfuser/python_tests/test_dynamo.py \ No newline at end of file diff --git a/test/_nvfuser/test_python_frontend.py b/test/_nvfuser/test_python_frontend.py new file mode 120000 index 00000000000..a022a886483 --- /dev/null +++ b/test/_nvfuser/test_python_frontend.py @@ -0,0 +1 @@ +../../third_party/nvfuser/python_tests/test_python_frontend.py \ No newline at end of file diff --git a/test/_nvfuser/test_torchscript.py b/test/_nvfuser/test_torchscript.py new file mode 120000 index 00000000000..24384a27422 --- /dev/null +++ b/test/_nvfuser/test_torchscript.py @@ -0,0 +1 @@ +../../third_party/nvfuser/python_tests/test_torchscript.py \ No newline at end of file diff --git a/test/test_jit_cuda_fuser.py b/test/test_jit_cuda_fuser.py index 4d5c89d0d2a..c735bd996ab 100644 --- a/test/test_jit_cuda_fuser.py +++ b/test/test_jit_cuda_fuser.py @@ -1,7 +1,7 @@ # Owner(s): ["module: nvfuser"] try: - from _nvfuser.test_torchscript import run_tests # noqa: F403 + from _nvfuser.test_torchscript import * # noqa: F403,F401 except ImportError: def run_tests(): return diff --git a/test/test_nvfuser_dynamo.py b/test/test_nvfuser_dynamo.py index a64da982c8f..d4a67db02d8 100644 --- a/test/test_nvfuser_dynamo.py +++ b/test/test_nvfuser_dynamo.py @@ -1,7 +1,7 @@ # Owner(s): ["module: nvfuser"] try: - from _nvfuser.test_dynamo import run_tests # noqa: F403 + from _nvfuser.test_dynamo import * # noqa: F403,F401 except ImportError: def run_tests(): return diff --git a/test/test_nvfuser_frontend.py b/test/test_nvfuser_frontend.py index 59da68c524a..c530209a7a8 100644 --- a/test/test_nvfuser_frontend.py +++ b/test/test_nvfuser_frontend.py @@ -1,7 +1,7 @@ # Owner(s): ["module: nvfuser"] try: - from _nvfuser.test_python_frontend import run_tests # noqa: F403 + from _nvfuser.test_python_frontend import * # noqa: F403,F401 except ImportError: def run_tests(): return diff --git a/third_party/nvfuser/python_tests/__init__.py b/third_party/nvfuser/python_tests/__init__.py new file mode 100644 index 00000000000..e69de29bb2d diff --git a/third_party/nvfuser/python_tests/test_torchscript.py b/third_party/nvfuser/python_tests/test_torchscript.py index 310bb29f5f4..7eccdc0f21f 100644 --- a/third_party/nvfuser/python_tests/test_torchscript.py +++ b/third_party/nvfuser/python_tests/test_torchscript.py @@ -25,8 +25,6 @@ from torch.testing._internal.jit_utils import clone_inputs, get_traced_sample_va from torch.testing._internal.jit_metaprogramming_utils import create_traced_fn from torch.testing import FileCheck -from jit.test_fuser_common import TestFuserCommon # noqa: F401 - import itertools import numpy as np import math