MAINT: manually include 16bit file on MSVC

numpy · Apr 26, 2024 · 4045b72 · 4045b72
1 parent c3fec33
commit 4045b72
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/numpy/_core/src/npysort/x86_simd_qsort_16bit.dispatch.cpp b/numpy/_core/src/npysort/x86_simd_qsort_16bit.dispatch.cpp
@@ -2,6 +2,13 @@
 #ifndef __CYGWIN__
 
 #include "x86-simd-sort/src/x86simdsort-static-incl.h"
+/*
+ * MSVC doesn't set the macro __AVX512VBMI2__ which is required for the 16-bit
+ * functions and therefore we need to manually include this file here
+ */
+#ifdef _MSC_VER
+#include "x86-simd-sort/src/avx512-16bit-qsort.hpp"
+#endif
 
 namespace np { namespace qsort_simd {