AVXUtils.hpp

Go to the documentation of this file.

 
#include <fmt/core.h>
#include <fmt/ranges.h>
#include <immintrin.h>
 
#include <array>
 
#ifndef TPGLIBS_AVXUTILS_HPP_
#define TPGLIBS_AVXUTILS_HPP_
 
namespace tpglibs {

inline __m256i
_mm256_div_epi16(const __m256i& va, const int16_t& b)
{
  // NaiveUtils.hpp has the lay version.
  __m256i vb = _mm256_set1_epi16(32768 / b);
  return _mm256_mulhrs_epi16(va, vb);
}

inline void
_mm256_print_epi16(const __m256i& input)
{
  std::array<int16_t, 16> prints;
  _mm256_storeu_si256(reinterpret_cast<__m256i*>(prints.begin()), input);
 
  fmt::print("{}", prints);
}
 
} // namespace tpglibs
 
#endif // TPGLIBS_AVXUTILS_HPP_