[JIT] RUN_CUDA -> RUN_NVFUSER to fix nvfuser ci

davidberard98 · davidberard98 · commit 857f21c99e20 · 2022-04-01T13:58:44.000-07:00
fixes merge issue between #75016 and #73322 ghstack-source-id: a79f74a Pull Request resolved: #75134
diff --git a/test/test_jit_cuda_fuser.py b/test/test_jit_cuda_fuser.py
@@ -69,6 +69,12 @@ def is_pre_volta():
 
 TEST_BF16 = torch.cuda.is_bf16_supported()
 
+# TODO: remove this - just to avoid conflicts with 73322 and 75016
+try:
+    RUN_NVFUSER
+except NameError:
+    RUN_NVFUSER = RUN_CUDA and not TEST_WITH_ROCM and not IS_WINDOWS
+
 class TestCudaFuser(JitTestCase):
 
     special_values = torch.tensor(
@@ -4274,7 +4280,7 @@ def reduce_scalar(temp):
         reduce_scalar(res).backward()
         torch._C._jit_set_nvfuser_guard_mode(old_guard)
 
-    @unittest.skipIf(not RUN_CUDA, "requires CUDA")
+    @unittest.skipIf(not RUN_NVFUSER, "requires CUDA")
     @unittest.skipIf(GRAPH_EXECUTOR != ProfilingMode.PROFILING,
                      "Requires fusion optimization pass to be effective")
     def test_cuda_fusion_guard_backward(self):
