check_both_nan Class — pytorch Architecture

Source Code

aten/src/ATen/test/vec_test_all_types.h lines 286–292

template <typename T>
bool check_both_nan([[maybe_unused]] T x, [[maybe_unused]] T y) {
    if constexpr (std::is_floating_point_v<T> || c10::is_reduced_floating_point_v<T>) {
        return std::isnan(x) && std::isnan(y);
    }
    return false;
}