File tree Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -478,8 +478,9 @@ def test_create_supervised_trainer_on_cuda_amp_scaler():
478478 )
479479
480480
481- @pytest .mark .skipif (not torch .cuda .is_available (), reason = "Skip if no GPU" )
482- @pytest .mark .skipif (not find_spec ("apex" ), reason = "Skip if no APEX" )
481+ # @pytest.mark.skipif(not torch.cuda.is_available(), reason="Skip if no GPU")
482+ # @pytest.mark.skipif(not find_spec("apex"), reason="Skip if no APEX")
483+ @pytest .mark .skip (reason = "Temporarily disabled, as it fails because of an issue from apex side" )
483484def test_create_supervised_trainer_on_cuda_apex ():
484485 model_device = trainer_device = "cuda"
485486 _test_create_supervised_trainer_wrong_accumulation (
Original file line number Diff line number Diff line change 11import os
2- from importlib .util import find_spec
32
43import numpy as np
54import pytest
@@ -500,8 +499,10 @@ def test_distrib_xla_nprocs(xmp_executor):
500499 xmp_executor (_test_distrib_xla_nprocs , args = (), nprocs = n )
501500
502501
503- @pytest .mark .skipif (not torch .cuda .is_available (), reason = "Skip if no GPU" )
504- @pytest .mark .skipif (not find_spec ("apex" ), reason = "Skip if no APEX" )
502+ # Enable this test when apex issue is fixed
503+ # @pytest.mark.skipif(not torch.cuda.is_available(), reason="Skip if no GPU")
504+ # @pytest.mark.skipif(not find_spec("apex"), reason="Skip if no APEX")
505+ @pytest .mark .skip (reason = "Temporarily disabled, as it fails because of an issue from apex side" )
505506def test_apex_average_on_cuda ():
506507 device = "cuda"
507508 _test_apex_average (device , amp_mode = "apex" , opt_level = "O0" )
You can’t perform that action at this time.
0 commit comments