Skip to content

Commit eadcb1c

Browse files
committed
format code
1 parent 599e06d commit eadcb1c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/test_jit_cuda_fuser.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,8 @@
4343

4444
if 'PYTORCH_NVFUSER_ENABLE' not in os.environ:
4545
os.environ['PYTORCH_NVFUSER_ENABLE'] = ""
46-
os.environ['PYTORCH_NVFUSER_ENABLE'] = 'linear_decomposition,conv_decomposition,graph_op_fusion,' + os.environ['PYTORCH_NVFUSER_ENABLE']
46+
os.environ['PYTORCH_NVFUSER_ENABLE'] = 'linear_decomposition,conv_decomposition,graph_op_fusion,' + \
47+
os.environ['PYTORCH_NVFUSER_ENABLE']
4748
if 'PYTORCH_NVFUSER_DISABLE' not in os.environ:
4849
os.environ['PYTORCH_NVFUSER_DISABLE'] = ""
4950
os.environ['PYTORCH_NVFUSER_DISABLE'] = 'fallback,fma,' + os.environ['PYTORCH_NVFUSER_DISABLE']
@@ -3827,6 +3828,7 @@ def _view_test_generator(self, ndims, test_fn):
38273828
total += 1
38283829
test_fn(all_views[idx], all_views[jdx], torch.float, 'cuda', 1e-6)
38293830

3831+
@unittest.skipIf(ALIAS_TEST_DISABLED, "skipping this test since view is disabled now")
38303832
@unittest.skipIf(not RUN_NVFUSER, "requires CUDA")
38313833
@unittest.skipIf(GRAPH_EXECUTOR != ProfilingMode.PROFILING,
38323834
"Requires fusion optimization pass to be effective")
@@ -4151,7 +4153,6 @@ def t(x):
41514153
t_jit = torch.jit.script(t)
41524154
self._run_helper(t_jit, t, x)
41534155

4154-
@unittest.skipIf(ALIAS_TEST_DISABLED, "skipping this test since index_select's backend PR hasn't been merged")
41554156
@unittest.skipIf(not RUN_NVFUSER, "requires CUDA")
41564157
@unittest.skipIf(GRAPH_EXECUTOR != ProfilingMode.PROFILING,
41574158
"Requires fusion optimization pass to be effective")
@@ -4170,7 +4171,6 @@ def t(x_kj, idx_kj, sbf):
41704171
t_jit = torch.jit.script(t)
41714172
self._run_helper(t_jit, t, lookup_tv, indies_tv, sbf)
41724173

4173-
@unittest.skipIf(ALIAS_TEST_DISABLED, "skipping this test since index_select's backend PR hasn't been merged")
41744174
@unittest.skipIf(not RUN_NVFUSER, "requires CUDA")
41754175
@unittest.skipIf(GRAPH_EXECUTOR != ProfilingMode.PROFILING,
41764176
"Requires fusion optimization pass to be effective")

0 commit comments

Comments
 (0)