portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-09 21:50:45 +01:00
Code Releases Activity

Views must be closed

Browse Source
This commit is contained in:
Peter Boyle 2020-06-03 09:10:29 -04:00
parent 32237895bd
commit 1c9f20b15e
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
Grid/algorithms/iterative/ConjugateGradient.h
View File

@ -140,15 +140,17 @@ public:
b = cp / c; b = cp / c;
LinearCombTimer.Start(); LinearCombTimer.Start();
auto psi_v = psi.View(AcceleratorWrite); {
auto p_v = p.View(AcceleratorWrite); auto psi_v = psi.View(AcceleratorWrite);
auto r_v = r.View(AcceleratorWrite); auto p_v = p.View(AcceleratorWrite);
accelerator_for(ss,p_v.size(), Field::vector_object::Nsimd(),{ auto r_v = r.View(AcceleratorWrite);
coalescedWrite(psi_v[ss], a * p_v(ss) + psi_v(ss)); accelerator_for(ss,p_v.size(), Field::vector_object::Nsimd(),{
coalescedWrite(p_v[ss] , b * p_v(ss) + r_v (ss)); coalescedWrite(psi_v[ss], a * p_v(ss) + psi_v(ss));
}); coalescedWrite(p_v[ss] , b * p_v(ss) + r_v (ss));
LinearCombTimer.Stop(); });
LinalgTimer.Stop(); LinearCombTimer.Stop();
LinalgTimer.Stop();
}
std::cout << GridLogIterative << "ConjugateGradient: Iteration " << k std::cout << GridLogIterative << "ConjugateGradient: Iteration " << k
<< " residual " << sqrt(cp/ssq) << " target " << Tolerance << std::endl; << " residual " << sqrt(cp/ssq) << " target " << Tolerance << std::endl;