diff options
Diffstat (limited to 'src/QuantUtilsAvx2.cc')
| -rw-r--r-- | src/QuantUtilsAvx2.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/QuantUtilsAvx2.cc b/src/QuantUtilsAvx2.cc index ecd0be2..875f1dc 100644 --- a/src/QuantUtilsAvx2.cc +++ b/src/QuantUtilsAvx2.cc @@ -23,7 +23,7 @@ void QuantizeAvx2( uint8_t* dst, int len, const TensorQuantizationParams& qparams) { -#if defined(__AVX2__) && defined(__FMA__) +#if defined(__AVX2__) && (defined(__FMA__) || defined(_MSC_VER)) constexpr int VLEN = 8; std::size_t i = 0; __m256 inverse_scale_v = _mm256_set1_ps(1.f / qparams.scale); |