Files
pytorch/test/test_jit_simple.py
Nikolay Korovaiko 96885f73ed make test_jit infer the profiling mode, add a job for simple executor (#38374)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/38374

Differential Revision: D21567658

Pulled By: Krovatkin

fbshipit-source-id: c0eb44cf6c842d5feebabf8c7d99c1b4aa6c4960
2020-05-13 23:55:40 -07:00

11 lines
258 B
Python

import sys
sys.argv.append("--ge_config=simple")
from test_jit import *
if __name__ == '__main__':
run_tests()
if not PY2:
import test_jit_py3
suite = unittest.findTestCases(test_jit_py3)
unittest.TextTestRunner().run(suite)