Skip to content

Commit c3fc367

Browse files
committed
add CNR constant for avx10
1 parent b63c9c1 commit c3fc367

4 files changed

Lines changed: 10 additions & 1 deletion

File tree

CHANGELOG.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
77
## [dev] (MM/DD/YYYY)
88

99
### Added
10-
* Added support for ISA constants `"avx10"` and `"avx512_e5"` [gh-175](https://github.com/IntelPython/mkl-service/pull/175)
10+
* Added support for ISA constants `"avx10"` and `"avx512_e5"` and CNR constants `"avx10"` and `"avx10,strict"` [gh-175](https://github.com/IntelPython/mkl-service/pull/175)
1111

1212
### Removed
1313
* Dropped support for Python 3.9 [gh-118](https://github.com/IntelPython/mkl-service/pull/118)

mkl/_mkl_service.pxd

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ cdef extern from "mkl.h":
5959
int MKL_CBWR_AVX2
6060
int MKL_CBWR_AVX512
6161
int MKL_CBWR_AVX512_E1
62+
int MKL_CBWR_AVX10
6263

6364
int MKL_CBWR_SUCCESS
6465
int MKL_CBWR_ERR_INVALID_SETTINGS

mkl/_mkl_service.pyx

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -684,6 +684,8 @@ cdef object __cbwr_set(branch=None) except *:
684684
"avx512,strict": mkl.MKL_CBWR_AVX512 | mkl.MKL_CBWR_STRICT,
685685
"avx512_e1": mkl.MKL_CBWR_AVX512_E1,
686686
"avx512_e1,strict": mkl.MKL_CBWR_AVX512_E1 | mkl.MKL_CBWR_STRICT,
687+
"avx10": mkl.MKL_CBWR_AVX10,
688+
"avx10,strict": mkl.MKL_CBWR_AVX10 | mkl.MKL_CBWR_STRICT,
687689
},
688690
"output": {
689691
mkl.MKL_CBWR_SUCCESS: "success",
@@ -721,6 +723,8 @@ cdef inline __cbwr_get(cnr_const=None) except *:
721723
mkl.MKL_CBWR_AVX512 | mkl.MKL_CBWR_STRICT: "avx512,strict",
722724
mkl.MKL_CBWR_AVX512_E1: "avx512_e1",
723725
mkl.MKL_CBWR_AVX512_E1 | mkl.MKL_CBWR_STRICT: "avx512_e1,strict",
726+
mkl.MKL_CBWR_AVX10: "avx10",
727+
mkl.MKL_CBWR_AVX10 | mkl.MKL_CBWR_STRICT: "avx10,strict",
724728
mkl.MKL_CBWR_ERR_INVALID_INPUT: "err_invalid_input",
725729
},
726730
}
@@ -749,6 +753,8 @@ cdef object __cbwr_get_auto_branch() except *:
749753
mkl.MKL_CBWR_AVX512 | mkl.MKL_CBWR_STRICT: "avx512,strict",
750754
mkl.MKL_CBWR_AVX512_E1: "avx512_e1",
751755
mkl.MKL_CBWR_AVX512_E1 | mkl.MKL_CBWR_STRICT: "avx512_e1,strict",
756+
mkl.MKL_CBWR_AVX10: "avx10",
757+
mkl.MKL_CBWR_AVX10 | mkl.MKL_CBWR_STRICT: "avx10,strict",
752758
mkl.MKL_CBWR_SUCCESS: "success",
753759
mkl.MKL_CBWR_ERR_INVALID_INPUT: "err_invalid_input",
754760
},

mkl/tests/test_mkl_service.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,13 +223,15 @@ def check_cbwr(branch, cnr_const):
223223
"avx2",
224224
"avx512",
225225
"avx512_e1",
226+
"avx10",
226227
]
227228

228229

229230
strict = [
230231
"avx2,strict",
231232
"avx512,strict",
232233
"avx512_e1,strict",
234+
"avx10,strict",
233235
]
234236

235237

0 commit comments

Comments
 (0)