add missing ISA constants

also refactors enable_instructions test

Author: ndgrigorian

mkl/_mkl_service.pxd

@@ -73,10 +73,12 @@ cdef extern from "mkl.h":
     int MKL_ENABLE_AVX512_E3
     int MKL_ENABLE_AVX512_E4
     int MKL_ENABLE_AVX512_E1
+    int MKL_ENABLE_AVX512_E5
     int MKL_ENABLE_AVX512
     int MKL_ENABLE_AVX2
     int MKL_ENABLE_AVX2_E1
     int MKL_ENABLE_SSE4_2
+    int MKL_ENABLE_AVX10
 
     # MPI Implementation Constants
     int MKL_BLACS_CUSTOM

mkl/_mkl_service.pyx

@@ -773,10 +773,12 @@ cdef object __enable_instructions(isa=None) except *:
             "avx512_e3": mkl.MKL_ENABLE_AVX512_E3,
             "avx512_e2": mkl.MKL_ENABLE_AVX512_E2,
             "avx512_e1": mkl.MKL_ENABLE_AVX512_E1,
+            "avx512_e5": mkl.MKL_ENABLE_AVX512_E5,
             "avx512": mkl.MKL_ENABLE_AVX512,
             "avx2_e1": mkl.MKL_ENABLE_AVX2_E1,
             "avx2": mkl.MKL_ENABLE_AVX2,
             "sse4_2": mkl.MKL_ENABLE_SSE4_2,
+            "avx10": mkl.MKL_ENABLE_AVX10,
         },
     }
     cdef int c_mkl_isa = __mkl_str_to_int(isa, __variables["input"])

mkl/tests/test_mkl_service.py

@@ -247,16 +247,24 @@ def test_cbwr_get_auto_branch():
     mkl.cbwr_get_auto_branch()
 
 
-def test_enable_instructions_avx512():
-    mkl.enable_instructions("avx512")
-
-
-def test_enable_instructions_avx2():
-    mkl.enable_instructions("avx2")
+instructions = [
+    "single_path",
+    "avx512_e4",
+    "avx512_e3",
+    "avx512_e2",
+    "avx512_e1",
+    "avx512_e5",
+    "avx512",
+    "avx2_e1",
+    "avx2",
+    "sse4_2",
+    "avx10",
+]
 
 
-def test_enable_instructions_sse4_2():
-    mkl.enable_instructions("sse4_2")
+@pytest.mark.parametrize("isa", instructions)
+def test_enable_instructions(isa):
+    mkl.enable_instructions(isa)
 
 
 def test_set_env_mode():