Skip to content

Commit 6c095b2

Browse files
Revert "Update test_metric_result_integration.py"
This reverts commit bf70e43.
1 parent 250d0aa commit 6c095b2

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/core/test_metric_result_integration.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
import torch.distributed as dist
1717
import torch.multiprocessing as mp
1818
from torchmetrics import Metric
19+
import numpy
20+
import os
1921

2022
import tests.helpers.utils as tutils
2123
from pytorch_lightning.core.step_result import Result
@@ -96,6 +98,7 @@ def test_result_reduce_ddp():
9698
tutils.set_random_master_port()
9799

98100
worldsize = 2
101+
os.environ["MKL_SERVICE_FORCE_INTEL"] = "1"
99102
mp.spawn(_ddp_test_fn, args=(worldsize, ), nprocs=worldsize)
100103

101104

0 commit comments

Comments
 (0)