diff options
author | Sayed Adel <seiko@imavr.com> | 2020-10-27 00:04:20 +0000 |
---|---|---|
committer | Sayed Adel <seiko@imavr.com> | 2020-10-27 11:46:58 +0000 |
commit | 8cc5009857391ab2a11fb1af1042c0b83544e97d (patch) | |
tree | f064a082a6701e8bed978eaa47ca73271b1e2a1f | |
parent | d54a45b28fdaad24eedf8a2ffee19d9afc7f2e69 (diff) | |
download | numpy-8cc5009857391ab2a11fb1af1042c0b83544e97d.tar.gz |
MAINT, TST: Add _SIMD attribute for each enabled SIMD extension
-rw-r--r-- | numpy/core/src/_simd/_simd.c | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/numpy/core/src/_simd/_simd.c b/numpy/core/src/_simd/_simd.c index e5cb582b3..d2e26d821 100644 --- a/numpy/core/src/_simd/_simd.c +++ b/numpy/core/src/_simd/_simd.c @@ -40,6 +40,11 @@ PyMODINIT_FUNC PyInit__simd(void) Py_DECREF(simd_mod); \ goto err; \ } \ + Py_INCREF(simd_mod); \ + if (PyModule_AddObject(m, target_name, simd_mod) < 0) { \ + Py_DECREF(simd_mod); \ + goto err; \ + } \ } #define ATTACH_BASELINE_MODULE(MAKE_MSVC_HAPPY) \ @@ -52,6 +57,11 @@ PyMODINIT_FUNC PyInit__simd(void) Py_DECREF(simd_mod); \ goto err; \ } \ + Py_INCREF(simd_mod); \ + if (PyModule_AddObject(m, "baseline", simd_mod) < 0) { \ + Py_DECREF(simd_mod); \ + goto err; \ + } \ } NPY__CPU_DISPATCH_CALL(NPY_CPU_HAVE, ATTACH_MODULE, MAKE_MSVC_HAPPY) |