[inductor] Fix MKL issue with test_indirect_device_assert

jansel · jansel · commit b38992727278 · 2023-08-30T18:58:51.000Z
ghstack-source-id: b91f2e7 Pull Request resolved: #108172
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")
