4343
4444if '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' ]
4748if 'PYTORCH_NVFUSER_DISABLE' not in os .environ :
4849 os .environ ['PYTORCH_NVFUSER_DISABLE' ] = ""
4950os .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