is_void_v Class — pytorch Architecture

Source Code

aten/src/ATen/native/cpu/Loops.h lines 84–97

template <typename func_t,
    std::enable_if_t<!std::is_void_v<typename function_traits<func_t>::result_type>>* = nullptr>
inline void
execute_op(char* C10_RESTRICT data[], const int64_t* strides, int64_t i, int64_t n, func_t&& op) {
  using traits = function_traits<func_t>;
  using result_type = typename traits::result_type;
  for (; i < n; i++) {
    result_type* out_ptr = (result_type*)(data[0] + i * strides[0]);
    *out_ptr = std::apply(op, dereference<traits>(
        &data[1],
        &strides[1],
        i));
  }
}