[inductor] Fix MKL issue with test_indirect_device_assert (#108172)

jansel · pytorchmergebot · commit 8a089f632e9d · 2023-08-30T23:13:22.000Z
Pull Request resolved: #108172 Approved by: https://github.com/peterbell10
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")
