From 250d0aaaa2e6c6a6a3407bc6c8b83c0fe2479c0b Mon Sep 17 00:00:00 2001 From: shuyingsunshine21 <80445420+shuyingsunshine21@users.noreply.github.com> Date: Wed, 24 Mar 2021 02:06:48 -0700 Subject: [PATCH] Revert "Update test_results.py" This reverts commit ea749068785bbad689a12066544893b1605f20c5. --- tests/core/test_results.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/core/test_results.py b/tests/core/test_results.py index f25ab0c40a..74c4a0c212 100644 --- a/tests/core/test_results.py +++ b/tests/core/test_results.py @@ -26,6 +26,8 @@ from pytorch_lightning.core.step_result import Result from pytorch_lightning.trainer.states import TrainerState from tests.helpers import BoringDataModule, BoringModel from tests.helpers.runif import RunIf +import os +import numpy def _setup_ddp(rank, worldsize): @@ -50,6 +52,7 @@ def _ddp_test_fn(rank, worldsize, result_cls: Result): def test_result_reduce_ddp(): """Make sure result logging works with DDP""" tutils.reset_seed() + os.environ["MKL_SERVICE_FORCE_INTEL"] = "1" tutils.set_random_master_port() worldsize = 2