../../third_party/nvfuser/python_tests/test_torchscript.py