gallo/atrip
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add correct conjugate templated function

Browse Source
This commit is contained in:
Alejandro Gallo 2022-02-09 19:35:00 +01:00
parent c2b1c78c67
commit e89bd8f150
1 changed files with 21 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
atrip.org
View File

@ -21,6 +21,9 @@ The algorithm uses two main data types, the =Slice= and the
namespace atrip { namespace atrip {
template <typename FF> FF maybeConjugate(const FF a) { return a; }
template <> Complex maybeConjugate(const Complex a) { return std::conj(a); }
namespace traits { namespace traits {
template <typename FF> bool isComplex() { return false; }; template <typename FF> bool isComplex() { return false; };
template <> bool isComplex<Complex>() { return true; }; template <> bool isComplex<Complex>() { return true; };
@ -1521,12 +1524,12 @@ namespace atrip {
, X(Zijk_[j + No*k + No*No*i]) , X(Zijk_[j + No*k + No*No*i])
, Y(Zijk_[k + No*i + No*No*j]) , Y(Zijk_[k + No*i + No*No*j])
, Z(Zijk_[k + No*j + No*No*i]) , Z(Zijk_[k + No*j + No*No*i])
, A(std::conj(Tijk_[i + No*j + No*No*k])) , A(maybeConjugate<F>(Tijk_[i + No*j + No*No*k]))
, B(std::conj(Tijk_[i + No*k + No*No*j])) , B(maybeConjugate<F>(Tijk_[i + No*k + No*No*j]))
, C(std::conj(Tijk_[j + No*i + No*No*k])) , C(maybeConjugate<F>(Tijk_[j + No*i + No*No*k]))
, D(std::conj(Tijk_[j + No*k + No*No*i])) , D(maybeConjugate<F>(Tijk_[j + No*k + No*No*i]))
, E(std::conj(Tijk_[k + No*i + No*No*j])) , E(maybeConjugate<F>(Tijk_[k + No*i + No*No*j]))
, F(std::conj(Tijk_[k + No*j + No*No*i])) , F(maybeConjugate<F>(Tijk_[k + No*j + No*No*i]))
, value , value
= 3.0 * ( A * U = 3.0 * ( A * U
+ B * V + B * V
@ -1583,9 +1586,9 @@ namespace atrip {
, U(Zijk_[i + No*j + No*No*k]) , U(Zijk_[i + No*j + No*No*k])
, V(Zijk_[j + No*k + No*No*i]) , V(Zijk_[j + No*k + No*No*i])
, W(Zijk_[k + No*i + No*No*j]) , W(Zijk_[k + No*i + No*No*j])
, A(std::conj(Tijk_[i + No*j + No*No*k])) , A(maybeConjugate<F>(Tijk_[i + No*j + No*No*k]))
, B(std::conj(Tijk_[j + No*k + No*No*i])) , B(maybeConjugate<F>(Tijk_[j + No*k + No*No*i]))
, C(std::conj(Tijk_[k + No*i + No*No*j])) , C(maybeConjugate<F>(Tijk_[k + No*i + No*No*j]))
, value , value
= F(3.0) * ( A * U = F(3.0) * ( A * U
+ B * V + B * V
@ -1701,14 +1704,9 @@ namespace atrip {
, _t_buffer.data() \ , _t_buffer.data() \
, (int const*)&NoNo \ , (int const*)&NoNo \
); );
#define MAYBE_CONJ(_conj, _buffer) \ #define MAYBE_CONJ(_conj, _buffer) \
if (traits::isComplex<F>()) { \ for (size_t __i = 0; __i < NoNoNo; ++__i) \
for (size_t __i = 0; __i < NoNoNo; ++__i) \ _conj[__i] = maybeConjugate<F>(_buffer[__i]); \
_conj[__i] = std::conj(_buffer[__i]); \
} else { \
for (size_t __i = 0; __i < NoNoNo; ++__i) \
_conj[__i] = _buffer[__i]; \
}
const size_t NoNoNo = No*NoNo; const size_t NoNoNo = No*NoNo;
std::vector<F> _t_buffer; std::vector<F> _t_buffer;
@ -2259,11 +2257,12 @@ Atrip::Output Atrip::run(Atrip::Input<F> const& in) {
const double doublesFlops const double doublesFlops
= double(No) = double(No)
,* double(No) * double(No)
,* double(No) * double(No)
,* (double(No) + double(Nv)) * (double(No) + double(Nv))
,* 2 * 1
,* 6 * (traits::isComplex<F>() ? 2.0 : 1.0)
* 6
/ 1e9 / 1e9
; ;