Files
pytorch/test/test_jit_profiling.py
Nikolay Korovaiko edc5ef1afb run the simple executor for jit tests by default, add profiling jobs … (#37017)
Summary:
…for fusion tests

fix flake8 warnings

fix ci failures

fix test_determination.py
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37017

Differential Revision: D21238446

Pulled By: Krovatkin

fbshipit-source-id: 393e6135883dc5ac57bdff580de96c66829d454c
2020-04-28 19:16:52 -07:00

11 lines
261 B
Python

import sys
sys.argv.append("--ge_config=profiling")
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)