[inductor] Fix MKL issue with test_indirect_device_assert (pytorch#10…

…8172) Pull Request resolved: pytorch#108172 Approved by: https://github.com/peterbell10
heysaeed · Aug 30, 2023 · 8a089f6 · 8a089f6
1 parent b2fe5eb
commit 8a089f6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test/inductor/test_torchinductor.py b/test/inductor/test_torchinductor.py
@@ -7563,6 +7563,7 @@ def test_indirect_device_assert(self):
                     ],
                     stdout=subprocess.PIPE,
                     stderr=subprocess.PIPE,
+                    env={**os.environ, "MKL_THREADING_LAYER": "GNU"},
                 )
                 stderr = proc.communicate()[1]
                 self.assertTrue(
@@ -7576,8 +7577,10 @@ def test_indirect_device_assert(self):
                 [sys.executable, test_path, "first_arg", "2", "False", "True"],
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,
+                env={**os.environ, "MKL_THREADING_LAYER": "GNU"},
             )
             stderr = proc.communicate()[1]
+
             self.assertTrue(
                 any(
                     "index out of bounds" in err.decode("utf-8")