test_jit_cuda_fuser_profiling.py

import sys
sys.argv.append("--jit_executor=profiling")

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()