Merge pull request #22099 from greglucas/numpy-cpu-avx512

timhoffm · web-flow · commit 82e593984cb6 · 2022-01-04T08:22:46.000+01:00
CI: Disable numpy avx512 instructions
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -9,6 +9,7 @@ on:
 
 env:
   NO_AT_BRIDGE: 1  # Necessary for GTK3 interactive test.
+  NPY_DISABLE_CPU_FEATURES: "AVX512F,AVX512CD,AVX512VL,AVX512BW,AVX512DQ,AVX512_SKX"
   OPENBLAS_NUM_THREADS: 1
   PYTHONFAULTHANDLER: 1
 
diff --git a/azure-pipelines.yml b/azure-pipelines.yml
@@ -129,6 +129,7 @@ stages:
         displayName: 'print env'
 
       - bash: |
+          export NPY_DISABLE_CPU_FEATURES="AVX512F,AVX512CD,AVX512VL,AVX512BW,AVX512DQ,AVX512_SKX"
           PYTHONFAULTHANDLER=1 python -m pytest --junitxml=junit/test-results.xml -raR --maxfail=50 --timeout=300 --durations=25 --cov-report= --cov=lib -n 2 ||
             [[ "$PYTHON_VERSION" = 'Pre' ]]
         displayName: 'pytest'
diff --git a/lib/matplotlib/tests/test_backend_tk.py b/lib/matplotlib/tests/test_backend_tk.py
@@ -36,7 +36,8 @@ def test_func():
         try:
             proc = subprocess.run(
                 [sys.executable, "-c", f"{source}\n{func.__name__}()"],
-                env={**os.environ, "MPLBACKEND": "TkAgg"},
+                env={**os.environ, "MPLBACKEND": "TkAgg",
+                     "NPY_DISABLE_CPU_FEATURES": ""},
                 timeout=_test_timeout,
                 stdout=subprocess.PIPE,
                 stderr=subprocess.PIPE,
diff --git a/lib/matplotlib/tests/test_sphinxext.py b/lib/matplotlib/tests/test_sphinxext.py
@@ -28,7 +28,8 @@ def test_tinypages(tmpdir):
     # coverage anyways); hide them using GCOV_ERROR_FILE.
     proc = Popen(
         cmd, stdout=PIPE, stderr=PIPE, universal_newlines=True,
-        env={**os.environ, "MPLBACKEND": "", "GCOV_ERROR_FILE": os.devnull})
+        env={**os.environ, "MPLBACKEND": "", "GCOV_ERROR_FILE": os.devnull,
+             "NPY_DISABLE_CPU_FEATURES": ""})
     out, err = proc.communicate()
 
     # Build the pages with warnings turned into errors
@@ -128,7 +129,8 @@ def build_sphinx_html(source_dir, doctree_dir, html_dir, extra_args=None):
     cmd = [sys.executable, '-msphinx', '-W', '-b', 'html',
            '-d', str(doctree_dir), str(source_dir), str(html_dir), *extra_args]
     proc = Popen(cmd, stdout=PIPE, stderr=PIPE, universal_newlines=True,
-                 env={**os.environ, "MPLBACKEND": ""})
+                 env={**os.environ, "MPLBACKEND": "",
+                      "NPY_DISABLE_CPU_FEATURES": ""})
     out, err = proc.communicate()
 
     assert proc.returncode == 0, \
