#pragma once

#include <ATen/NumericUtils.h>

namespace at {
namespace native {

// std:: does not have clamp functors
template <typename T>
struct minimum {
    __device__ T operator()(const T& a, const T& b) const { return (_isnan(a) || a < b) ? a : b; }
};

template <typename T>
struct maximum {
    __device__ T operator()(const T& a, const T& b) const { return (_isnan(a) || a > b) ? a : b; }
};

}} // at::native