mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
263412e536
Summary: `is_complex_t` is a bad name. For example in std, there are `std::is_same` but not `std::is_same_t`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/39906 Reviewed By: mrshenli Differential Revision: D22665013 Pulled By: anjali411 fbshipit-source-id: 4b71745f5e2ea2d8cf5845d95ada4556c87e040d
43 lines
921 B
C++
43 lines
921 B
C++
#if !defined(C10_INTERNAL_INCLUDE_COMPLEX_REMAINING_H)
|
|
#error "c10/util/complex_utils.h is not meant to be individually included. Include c10/util/complex.h instead."
|
|
#endif
|
|
|
|
#include <limits>
|
|
|
|
|
|
namespace c10 {
|
|
|
|
template <typename T>
|
|
struct is_complex : public std::false_type {};
|
|
|
|
template <typename T>
|
|
struct is_complex<std::complex<T>> : public std::true_type {};
|
|
|
|
template <typename T>
|
|
struct is_complex<c10::complex<T>> : public std::true_type {};
|
|
|
|
|
|
// Extract double from std::complex<double>; is identity otherwise
|
|
// TODO: Write in more idiomatic C++17
|
|
template <typename T>
|
|
struct scalar_value_type {
|
|
using type = T;
|
|
};
|
|
template <typename T>
|
|
struct scalar_value_type<std::complex<T>> {
|
|
using type = T;
|
|
};
|
|
template <typename T>
|
|
struct scalar_value_type<c10::complex<T>> {
|
|
using type = T;
|
|
};
|
|
|
|
}
|
|
|
|
namespace std {
|
|
|
|
template <typename T>
|
|
class numeric_limits<c10::complex<T>> : public numeric_limits<T> {};
|
|
|
|
}
|