Revert "Merge pull request #1 from shuyingsunshine21/shuyingsunshine2…

…1-checkpoint_consolidate" This reverts commit c5053da, reversing changes made to 0d23d75.
shuyingsunshine21 · Mar 24, 2021 · d64f90c · d64f90c
1 parent 7635b4f
commit d64f90c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/utilities/test_all_gather_grad.py b/tests/utilities/test_all_gather_grad.py
@@ -44,6 +44,7 @@ def _test_all_gather_ddp(rank, world_size):
 @RunIf(skip_windows=True)
 def test_all_gather_ddp():
     world_size = 3
+    os.environ["MKL_SERVICE_FORCE_INTEL"] = "1"
     torch.multiprocessing.spawn(_test_all_gather_ddp, args=(world_size, ), nprocs=world_size)