p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!22904 fix cuda10 complie error 

Merge pull request !22904 from zhouyaqiang0/complex_ops
This commit is contained in:
i-robot 2021-09-04 12:24:11 +00:00 committed by Gitee
parent 75d7a7c8ab bc519edba2
commit 1f3a5cb2ea
1 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/ccsrc/utils/complex.h
View File

@ -63,12 +63,8 @@ struct alignas(sizeof(T) * 2) Complex {
}
#endif
template <typename U = T>
HOST_DEVICE explicit Complex(const std::enable_if_t<std::is_same<U, float>::value, Complex<double>> &other)
: real_(other.real()), imag_(other.imag()) {}
template <typename U = T>
HOST_DEVICE explicit Complex(const std::enable_if_t<std::is_same<U, double>::value, Complex<float>> &other)
: real_(other.real()), imag_(other.imag()) {}
HOST_DEVICE inline explicit Complex(const Complex<U> &other)
: real_(static_cast<T>(other.real())), imag_(static_cast<T>(other.imag())) {}
HOST_DEVICE inline explicit operator bool() const { return static_cast<bool>(real_) || static_cast<bool>(imag_); }
HOST_DEVICE inline explicit operator signed char() const { return static_cast<signed char>(real_); }
@ -87,7 +83,7 @@ struct alignas(sizeof(T) * 2) Complex {
inline explicit operator float16() const { return static_cast<float16>(real_); }
#endif
HOST_DEVICE inline constexpr Complex<T> &operator=(const T &real) {
HOST_DEVICE inline Complex<T> &operator=(const T &real) {
real_ = real;
imag_ = T();
return *this;