saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove std::complex from c10::Half (#39833)

Browse source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39833

Reviewed By: mrshenli

Differential Revision: D22644987

Pulled By: anjali411

fbshipit-source-id: 5ae5db10b12d410560eca43234efa04b711a639c
This commit is contained in:
Xiang Gao 2020-08-18 15:19:48 -07:00 committed by Facebook GitHub Bot
parent 6c99d5611d
commit d5bc2a8058
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
c10/util/Half.h
View file

@ -372,9 +372,9 @@ struct alignas(4) complex<Half> {
Half imag() const {
return imag_;
}
inline complex(std::complex<float> value)
inline complex(c10::complex<float> value)
: real_(value.real()), imag_(value.imag()) {}
inline operator std::complex<float>() const {
inline operator c10::complex<float>() const {
return {real_, imag_};
}
};