diff --git a/include/recoil/split/rans_split_decoder.h b/include/recoil/split/rans_split_decoder.h index f5e4ce6..6095f1f 100644 --- a/include/recoil/split/rans_split_decoder.h +++ b/include/recoil/split/rans_split_decoder.h @@ -5,9 +5,12 @@ #include "recoil/rans_coded_data.h" #include "recoil/rans_decoder.h" #include "recoil/simd/rans_decoder_avx2_32x32.h" -//#include "recoil/simd/rans_decoder_avx512_32x32.h" #include +#ifdef AVX512 +#include "recoil/simd/rans_decoder_avx512_32x32.h" +#endif + namespace Recoil { template; +#ifdef AVX512 + using MyRansDecoder = RansDecoder_AVX512_32x32; +#else using MyRansDecoder = RansDecoder_AVX2_32x32; - //using MyRansDecoder = RansDecoder_AVX512_32x32; +#endif + public: std::vector result; diff --git a/include/recoil/split/rans_symbol_split_decoder.h b/include/recoil/split/rans_symbol_split_decoder.h index 3cdab18..b5f1cb3 100644 --- a/include/recoil/split/rans_symbol_split_decoder.h +++ b/include/recoil/split/rans_symbol_split_decoder.h @@ -6,10 +6,13 @@ #include "recoil/rans_coded_data.h" #include "recoil/rans_decoder.h" #include "recoil/simd/rans_decoder_avx2_32x32.h" -//#include "recoil/simd/rans_decoder_avx512_32x32.h" #include #include +#ifdef AVX512 +#include "recoil/simd/rans_decoder_avx512_32x32.h" +#endif + namespace Recoil { template; +#ifdef AVX512 + using MyRansDecoder = RansDecoder_AVX512_32x32; +#else using MyRansDecoder = RansDecoder_AVX2_32x32; - //using MyRansDecoder = RansDecoder_AVX512_32x32; +#endif public: std::vector result;