1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
#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
|
