Skip to content

Commit 250d0aa

Browse files
Revert "Update test_results.py"
This reverts commit ea74906.
1 parent 15f4b9e commit 250d0aa

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/core/test_results.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,8 @@
2626
from pytorch_lightning.trainer.states import TrainerState
2727
from tests.helpers import BoringDataModule, BoringModel
2828
from tests.helpers.runif import RunIf
29+
import os
30+
import numpy
2931

3032

3133
def _setup_ddp(rank, worldsize):
@@ -50,6 +52,7 @@ def _ddp_test_fn(rank, worldsize, result_cls: Result):
5052
def test_result_reduce_ddp():
5153
"""Make sure result logging works with DDP"""
5254
tutils.reset_seed()
55+
os.environ["MKL_SERVICE_FORCE_INTEL"] = "1"
5356
tutils.set_random_master_port()
5457

5558
worldsize = 2

0 commit comments

Comments
 (0)