Skip to content

Commit 7992040

Browse files
committed
1. update Parallel_LRI::get_list()
1 parent 553c91c commit 7992040

File tree

4 files changed

+12
-14
lines changed

4 files changed

+12
-14
lines changed

include/RI/parallel/Parallel_LRI.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,9 +40,9 @@ class Parallel_LRI
4040
const std::map<TA,std::map<TAC,Tensor<Tdata>>> &Ds) const =0;
4141

4242
virtual const std::vector<TA >& get_list_Aa01() const =0;
43-
virtual const std::vector<TAC>& get_list_Aa2 (const TA &Aa01) const =0;
44-
virtual const std::vector<TAC>& get_list_Ab01(const TA &Aa01, const TAC &Aa2) const =0;
45-
virtual const std::vector<TAC>& get_list_Ab2 (const TA &Aa01, const TAC &Aa2, const TAC &Ab01) const =0;
43+
virtual const std::vector<TAC>& get_list_Aa2 () const =0;
44+
virtual const std::vector<TAC>& get_list_Ab01() const =0;
45+
virtual const std::vector<TAC>& get_list_Ab2 () const =0;
4646
};
4747

4848
}

include/RI/parallel/Parallel_LRI_Equally.h

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,10 @@ class Parallel_LRI_Equally: public Parallel_LRI<TA,Tcell,Ndim,Tdata>
2828
const Label::ab &label,
2929
const std::map<TA,std::map<TAC,Tensor<Tdata>>> &Ds) const override;
3030

31-
const std::vector<TA >& get_list_Aa01() const override { return this->list_Aa01; }
32-
const std::vector<TAC>& get_list_Aa2 (const TA &Aa01) const override { return this->list_Aa2; }
33-
const std::vector<TAC>& get_list_Ab01(const TA &Aa01, const TAC &Aa2) const override { return this->list_Ab01; }
34-
const std::vector<TAC>& get_list_Ab2 (const TA &Aa01, const TAC &Aa2, const TAC &Ab01) const override { return this->list_Ab2; }
31+
const std::vector<TA >& get_list_Aa01() const override { return this->list_Aa01; }
32+
const std::vector<TAC>& get_list_Aa2 () const override { return this->list_Aa2; }
33+
const std::vector<TAC>& get_list_Ab01() const override { return this->list_Ab01; }
34+
const std::vector<TAC>& get_list_Ab2 () const override { return this->list_Ab2; }
3535

3636
// private:
3737
public:

include/RI/ri/LRI-cal.hpp

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,6 @@ void LRI<TA,Tcell,Ndim,Tdata>::cal(
2424
{
2525
using namespace Array_Operator;
2626

27-
dynamic_cast<Parallel_LRI_Equally_Filter<TA,Tcell,Ndim,Tdata>*>(this->parallel.get())->filter_Ab2(this->Ds_ab[Label::ab::b]);
28-
2927
const bool flag_D_b_transpose = [&labels]() -> bool
3028
{
3129
for(const Label::ab_ab &label : labels)
@@ -62,13 +60,13 @@ void LRI<TA,Tcell,Ndim,Tdata>::cal(
6260

6361
for(const TA &Aa01 : this->parallel->get_list_Aa01())
6462
{
65-
for(const TAC &Aa2 : this->parallel->get_list_Aa2(Aa01))
63+
for(const TAC &Aa2 : this->parallel->get_list_Aa2())
6664
{
6765
const Tensor<Tdata> D_a = tools.get_Ds_ab(Label::ab::a, Aa01, Aa2);
6866
if(D_a.empty()) continue;
6967
const Tensor<Tdata> D_a_transpose = LRI_Cal_Aux::tensor3_transpose(D_a);
7068

71-
const std::vector<TAC> &list_Ab01 = this->parallel->get_list_Ab01(Aa01, Aa2);
69+
const std::vector<TAC> &list_Ab01 = this->parallel->get_list_Ab01();
7270
#pragma omp for schedule(dynamic) nowait
7371
for(std::size_t ib01=0; ib01<list_Ab01.size(); ++ib01)
7472
{
@@ -81,7 +79,7 @@ void LRI<TA,Tcell,Ndim,Tdata>::cal(
8179
Ds_b01.reserve(Label::array_ab_ab.size());
8280
Ds_b01_csm.reserve(Label::array_ab_ab.size());
8381

84-
for(const TAC &Ab2 : this->parallel->get_list_Ab2(Aa01, Aa2, Ab01))
82+
for(const TAC &Ab2 : this->parallel->get_list_Ab2())
8583
{
8684
const Tensor<Tdata> D_b = tools.get_Ds_ab(Label::ab::b, Ab01, Ab2);
8785
if(D_b.empty()) continue;

include/RI/ri/LRI.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
#include "../ri/Label.h"
1010
#include "../global/Tensor.h"
1111
#include "CS_Matrix.h"
12-
#include "../parallel/Parallel_LRI_Equally_Filter.h"
12+
#include "../parallel/Parallel_LRI_Equally.h"
1313
#include "RI_Tools.h"
1414
#include "../global/Global_Func-2.h"
1515
#include "Save_Load.h"
@@ -59,7 +59,7 @@ class LRI
5959

6060
public:
6161
std::shared_ptr<Parallel_LRI<TA,Tcell,Ndim,Tdata>>
62-
parallel = std::make_shared<Parallel_LRI_Equally_Filter<TA,Tcell,Ndim,Tdata>>();
62+
parallel = std::make_shared<Parallel_LRI_Equally<TA,Tcell,Ndim,Tdata>>();
6363
std::unordered_map< Label::ab, RI_Tools::T_filter_func<Tdata> >
6464
filter_funcs;
6565
std::vector<std::function<Tdata (const Label::ab_ab &label, const TA &Aa01, const TAC &Aa2, const TAC &Ab01, const TAC &Ab2)>>

0 commit comments

Comments
 (0)