Files
pytorch/test/test_jit_legacy.py

16 lines
433 B
Python

# Owner(s): ["oncall: jit"]
import sys
sys.argv.append("--jit-executor=legacy")
from torch.testing._internal.common_utils import parse_cmd_line_args, run_tests
if __name__ == '__main__':
# The value of GRAPH_EXECUTOR depends on command line arguments so make sure they're parsed
# before instantiating tests.
parse_cmd_line_args()
from test_jit import * # noqa: F403, F401
if __name__ == '__main__':
run_tests()