Skip to content

Commit 79678b9

Browse files
disabled apex tests (#2308)
1 parent a93e481 commit 79678b9

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

tests/ignite/engine/test_create_supervised.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff 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")
483484
def test_create_supervised_trainer_on_cuda_apex():
484485
model_device = trainer_device = "cuda"
485486
_test_create_supervised_trainer_wrong_accumulation(

tests/ignite/metrics/test_accumulation.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import os
2-
from importlib.util import find_spec
32

43
import numpy as np
54
import 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")
505506
def test_apex_average_on_cuda():
506507
device = "cuda"
507508
_test_apex_average(device, amp_mode="apex", opt_level="O0")

0 commit comments

Comments
 (0)