File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -1586,12 +1586,15 @@ def test_pytorch_profiler_nested(tmpdir):
1586
1586
1587
1587
for n in ('a' , 'b' , 'c' ):
1588
1588
pa [n ] = [e .name for e in pa [n ]]
1589
- if LooseVersion (torch .__version__ ) >= LooseVersion ("1.7.1 " ):
1589
+ if LooseVersion (torch .__version__ ) >= LooseVersion ("1.7.0 " ):
1590
1590
pa [n ] = [e .replace ("aten::" , "" ) for e in pa [n ]]
1591
1591
assert pa [n ] == expected_ [n ]
1592
1592
1593
1593
1594
- @RunIf (min_gpus = 1 , special = True )
1594
+ @pytest .mark .skipif (torch .cuda .device_count () < 1 , reason = "test requires GPU machine" )
1595
+ @pytest .mark .skipif (
1596
+ not os .getenv ("PL_RUNNING_SPECIAL_TESTS" , '0' ) == '1' , reason = "test should be run outside of pytest"
1597
+ )
1595
1598
def test_pytorch_profiler_nested_emit_nvtx (tmpdir ):
1596
1599
"""
1597
1600
This test check emit_nvtx is correctly supported
You can’t perform that action at this time.
0 commit comments