size.hpp

#pragma once
#include "arm_simd/helpers/vec128.hpp"
 
#ifdef __ARM_FEATURE_MVE
#define simd mve
#else
#define simd neon
#endif
 
namespace argon::helpers {
template <typename T>
constexpr size_t vectorizeable_size(size_t size) {
  constexpr size_t lanes = sizeof(simd::Vec128_t<T>) / sizeof(T);
  return size & ~(lanes - 1);
}

template <typename T>
constexpr size_t vectorized_size(size_t size) {
  constexpr size_t lanes = sizeof(simd::Vec128_t<T>) / sizeof(T);
  return vectorizeable_size<T>(size) / lanes;
}
}  // namespace argon::helpers
#undef simd