blob: 28ab78370637d8ab0a3025f40a8f7851e7c2c2a1 [file] [log] [blame]
import sys
sys.argv.append("--jit_executor=legacy")
import os
os.environ['PYTORCH_CUDA_FUSER_DISABLE_FALLBACK'] = '1'
os.environ['PYTORCH_CUDA_FUSER_DISABLE_FMA'] = '1'
os.environ['PYTORCH_CUDA_FUSER_JIT_OPT_LEVEL'] = '0'
from test_jit_cuda_fuser import *
if __name__ == '__main__':
run_tests()