LCPQ/dft_tools
3
0
Fork 0
mirror of https://github.com/triqs/dft_tools synced 2024-11-01 03:33:50 +01:00
Code Issues Releases Wiki Activity

Fix in tail-matrix multiplication

Browse Source 
Set order_max to correspond to the mask
This commit is contained in:
Priyanka Seth 2014-10-07 13:48:47 +02:00
parent 9be131e8e1
commit 6a38d6747c
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
triqs/gfs/local/tail.hpp
View File

@ -337,6 +337,7 @@ namespace triqs { namespace gfs { namespace local {
template <typename T1, typename T2> template <typename T1, typename T2>
TYPE_ENABLE_IF(tail, mpl::and_<tqa::ImmutableMatrix<T1>, LocalTail<T2>>) operator*(T1 const &a, T2 const &b) { TYPE_ENABLE_IF(tail, mpl::and_<tqa::ImmutableMatrix<T1>, LocalTail<T2>>) operator*(T1 const &a, T2 const &b) {
auto res = tail{first_dim(a), b.shape()[1], b.size(), b.order_min()}; auto res = tail{first_dim(a), b.shape()[1], b.size(), b.order_min()};
res.mask_view() = b.order_max();
for (int n = res.order_min(); n <= res.order_max(); ++n) res(n) = a * b(n); for (int n = res.order_min(); n <= res.order_max(); ++n) res(n) = a * b(n);
return res; return res;
} }
@ -344,6 +345,7 @@ namespace triqs { namespace gfs { namespace local {
template <typename T1, typename T2> template <typename T1, typename T2>
TYPE_ENABLE_IF(tail, mpl::and_<LocalTail<T1>, tqa::ImmutableMatrix<T2>>) operator*(T1 const &a, T2 const &b) { TYPE_ENABLE_IF(tail, mpl::and_<LocalTail<T1>, tqa::ImmutableMatrix<T2>>) operator*(T1 const &a, T2 const &b) {
auto res = tail{a.shape()[0], second_dim(b), a.size(), a.order_min()}; auto res = tail{a.shape()[0], second_dim(b), a.size(), a.order_min()};
res.mask_view() = a.order_max();
for (int n = res.order_min(); n <= res.order_max(); ++n) res(n) = a(n) * b; for (int n = res.order_min(); n <= res.order_max(); ++n) res(n) = a(n) * b;
return res; return res;
} }