maximum Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec256/vec256_int.h lines 1311–1323

template <>
Vectorized<int64_t> inline maximum(
    const Vectorized<int64_t>& a,
    const Vectorized<int64_t>& b) {
#ifndef CPU_CAPABILITY_AVX2
  return emulate(a, b, [](int64_t a_point, int64_t b_point) {
    return std::max(a_point, b_point);
  });
#else
  __m256i cmp = _mm256_cmpgt_epi64(a, b);
  return _mm256_blendv_epi8(b, a, cmp);
#endif
}