bool.hpp

#pragma once
#include "argon/features.h"
#include "arm_simd/helpers/make_unsigned.hpp"
 
#ifdef __ARM_FEATURE_MVE
#define simd mve
#else
#define simd neon
#endif
 
namespace argon {
// clang-format off

template <typename T> struct Bool {
  using type = typename simd::make_unsigned<T>::type;
};

#if ARGON_HAS_SINGLE_FLOAT
template <> struct Bool<float32x4_t> { using type = uint32x4_t; };
 
#ifndef ARGON_PLATFORM_MVE
template <> struct Bool<float32x2_t> { using type = uint32x2_t; };
#endif
#endif
 
#if ARGON_HAS_DOUBLE_FLOAT
template <> struct Bool<float64x2_t> { using type = uint64x2_t; };
 
#ifndef ARGON_PLATFORM_MVE
template <> struct Bool<float64x1_t> { using type = uint64x1_t; };
#endif
#endif
 
#if ARGON_HAS_HALF_FLOAT
template <> struct Bool<float16x8_t> { using type = uint16x8_t; };
 
#ifndef ARGON_PLATFORM_MVE
template <> struct Bool<float16x4_t> { using type = uint16x4_t; };
#endif
#endif

template <typename T>
using Bool_t = Bool<T>::type;
// clang-format on
}  // namespace argon
#undef simd