convert_to_int32 Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec256/vec256_int.h lines 1493–1505

template <typename T>
std::
    enable_if_t<std::is_same_v<T, int8_t>, Vectorized<int32_t>> inline convert_to_int32(
        const int8_t* ptr,
        int count = Vectorized<int32_t>::size()) {
  if (count == Vectorized<int32_t>::size()) {
    return _mm256_cvtepi8_epi32(
        _mm_loadl_epi64(reinterpret_cast<const __m128i*>(ptr)));
  } else {
    auto a = Vectorized<int8_t>::loadu(ptr, count);
    return _mm256_cvtepi8_epi32(_mm256_castsi256_si128(a));
  }
}