is_zarch_implemented Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec256/zarch/vec256_zarch.h lines 22–29

template <typename T>
constexpr bool is_zarch_implemented() {
  return (
      std::is_same_v<T, float> || std::is_same_v<T, double> ||
      std::is_same_v<T, int8_t> || std::is_same_v<T, uint8_t> ||
      std::is_same_v<T, uint16_t> || std::is_same_v<T, int16_t> ||
      std::is_same_v<T, int32_t> || std::is_same_v<T, int64_t>);
}