split.hpp

#pragma once
#include <span>
#include <tuple>
#include "arm_simd/helpers/vec128.hpp"
 
#ifdef __ARM_FEATURE_MVE
#define simd mve
#else
#define simd neon
#endif
 
namespace argon {

template <typename scalar_type>
std::pair<std::span<scalar_type>, std::span<scalar_type>> split_head_aligned(std::span<scalar_type> span) {
  constexpr size_t lanes = sizeof(simd::Vec128_t<scalar_type>) / sizeof(scalar_type);
  return {span.first(span.size() & ~(lanes - 1)), span.last(span.size() & (lanes - 1))};
}

template <typename scalar_type>
std::pair<std::span<scalar_type>, std::span<scalar_type>> split_tail_aligned(std::span<scalar_type> span) {
  constexpr size_t lanes = sizeof(simd::Vec128_t<scalar_type>) / sizeof(scalar_type);
  return {span.first(span.size() & (lanes - 1)), span.last(span.size() & ~(lanes - 1))};
}
}  // namespace argon
#undef simd