mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
75cfc0be21
Partial work from #90847, in the direction of solving #89205. Most of the content is from #90847, but this is only for CPU, so hopefully it does not increase the build time by a lot. tag: @albanD, @malfet Pull Request resolved: https://github.com/pytorch/pytorch/pull/93153 Approved by: https://github.com/malfet, https://github.com/Skylion007
47 lines
1.1 KiB
C++
47 lines
1.1 KiB
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 c10
|
|
|
|
namespace std {
|
|
|
|
template <typename T>
|
|
class numeric_limits<c10::complex<T>> : public numeric_limits<T> {};
|
|
|
|
template <typename T>
|
|
bool isnan(const c10::complex<T>& v) {
|
|
return std::isnan(v.real()) || std::isnan(v.imag());
|
|
}
|
|
|
|
} // namespace std
|