argon_for.hpp

#pragma once
#include <type_traits>
#include "argon/features.h"
#include "arm_simd/helpers/concepts.hpp"
#include "arm_simd/helpers/scalar.hpp"
 
#ifdef __ARM_FEATURE_MVE
#define simd mve
#else
#define simd neon
#endif
 
template <typename T>
class ArgonHalf;
 
template <typename T>
  requires std::is_arithmetic_v<T>
class Argon;
 
namespace argon::helpers {

template <typename T>
struct ArgonFor;

template <simd::is_quadword T>
struct ArgonFor<T> {
  using type = Argon<simd::Scalar_t<T>>;
};
 
#ifndef ARGON_PLATFORM_MVE
template <simd::is_doubleword T>
struct ArgonFor<T> {
  using type = ArgonHalf<simd::Scalar_t<T>>;
};
#endif

template <typename T>
using ArgonFor_t = typename ArgonFor<std::remove_cv_t<T>>::type;
}  // namespace argon::helpers
#undef simd