10#ifndef CCSTRUCTURES_H_
11#define CCSTRUCTURES_H_
77 ss <<
"memory usage of process "<< world.
rank()<<
" on "<< hostname<<
": "<< mem/1024/1024<<
"MB";
78 std::string msg=ss.str();
79 auto memusage=world.
gop.
concat0(std::vector<std::string>(1,msg));
80 std::sort(memusage.begin(),memusage.end());
81 if (world.
rank()==0)
for (
const auto& msg : memusage)
print(msg);
101 output(
const std::string& msg)
const;
104 section(
const std::string& msg)
const;
110 warning(
const std::string& msg)
const;
118 using madness::operators::operator<<;
156 info(
const bool debug =
true,
const double norm = 12345.6789);
197 void print(
const std::pair<double, double>& times)
const {
199 std::cout << std::setfill(
' ') << std::scientific << std::setprecision(2)
200 <<
"Timer: " << times.first <<
" (Wall), " << times.second <<
" (CPU)" <<
", (" +
operation +
")"
208 std::vector<std::pair<int, int>>
map;
213 std::vector<std::pair<int, int>>
map;
214 for (
int i=nfreeze; i<nocc; ++i) {
215 for (
int j=i; j<nocc; ++j) {
216 map.push_back(std::make_pair(i,j));
223 std::vector<std::pair<int, int>>
map;
224 for (
int i=nfreeze; i<nocc; ++i) {
225 for (
int j=nfreeze; j<nocc; ++j) {
226 map.push_back(std::make_pair(i,j));
232 void print(
const std::string msg=
"PairVectorMap")
const {
235 for (
size_t i=0; i<
map.size(); ++i) {
254 template<
typename R,
typename opT>
257 for (
auto&
p :
arg.allpairs) {
259 int j=
p.first.second;
267 for (
int i=0; i<argument.size(); ++i) {
268 pairs.
insert(map.
map[i].first,map.
map[i].second,argument[i]);
274 std::vector<T> vector;
275 for (
size_t i=0; i<argument.
allpairs.size(); ++i) {
276 vector.push_back(argument(map.
map[i].first,map.
map[i].second));
283 return allpairs.at(std::make_pair(i, j));
291 return allpairs[std::make_pair(i, j)];
297 std::pair<int, int> key = std::make_pair(i, j);
298 allpairs.insert(std::make_pair(key, pair));
307 if (
allpairs.size() == 0)
return true;
313template <
typename T, std::
size_t NDIM>
320template<
typename T, std::
size_t NDIM>
324 for (
const auto& tmp : im.allpairs) {
325 size += get_size<T, NDIM>(tmp.second);
341 for (
size_t i = 0; i <
v.size(); i++) {
343 functions.insert(std::make_pair(i, tmp));
348 for (
size_t i = 0; i <
v.size(); i++) {
355 for (
size_t i = 0; i <
v.size(); i++) {
357 functions.insert(std::make_pair(i, tmp));
365 for (
size_t i = 0; i <
v.size(); i++) {
367 functions.insert(std::make_pair(freeze + i, tmp));
373 for (
auto x:
v)
functions.insert(std::make_pair(x.i, x));
386 if (
this == &other)
return *
this;
403 tmp.
functions.insert(std::make_pair(x.first,
copy(x.second)));
409 for (
auto& x :
functions) x.second.function.reconstruct();
437 template<
typename Archive>
439 typedef std::vector<std::pair<std::size_t, CCFunction<double,3>>> CC_functionvec;
442 return CC_functionvec(map.begin(), map.end());
444 auto vector2map = [] (
const CC_functionvec&
vec) {
449 if (ar.is_input_archive) {
452 CC_functionvec tmp(
size);
454 for (
auto& t : tmp) ar & t.first & t.second;
456 }
else if (ar.is_output_archive) {
459 for (
auto& t : tmp) ar & t.first & t.second;
464 hashT hashval = std::hash<FuncType>{}(
type);
516 for (
size_t i = 0; i <
v.size(); i++) {
518 functions.insert(std::make_pair(freeze + i, tmp));
525 for (
auto x:
functions) tmp.push_back(x.second.function);
536 print_size(
const std::string& msg =
"!?not assigned!?")
const;
541 const size_t freeze =
functions.cbegin()->first;
548 ktmp.second.function.scale(factor);
556 void plot(
const std::string& msg =
"")
const {
558 ktmp.second.plot(msg);
570template<
typename T=
double, std::
size_t NDIM=3>
593 template<
typename archiveT>
622 param.freeze=
a.parameters.freeze;
626 friend std::shared_ptr<CCConvolutionOperator>
combine(
const std::shared_ptr<CCConvolutionOperator>&
a,
627 const std::shared_ptr<CCConvolutionOperator>&
b) {
628 if (
a and (not
b))
return a;
629 if ((not
a) and
b)
return b;
630 if ((not
a) and (not
b))
return nullptr;
638 if (
op)
return ((*
op)(
f)).truncate();
647 std::vector<Function<T, NDIM>> result;
648 if constexpr (
NDIM == 3) {
653 result.push_back(tmpi);
671 return apply<T,T,NDIM,NDIM>(
world, (*
op),
f);
698 std::stringstream ss;
726 for (
const auto& tmp:
imH.allpairs)
727 tmp.second.print_size(
"<H" + std::to_string(tmp.first.first) +
"|" +
name() +
"|H" +
728 std::to_string(tmp.first.second) +
"> intermediate");
730 for (
const auto& tmp:
imP.allpairs)
731 tmp.second.print_size(
"<H" + std::to_string(tmp.first.first) +
"|" +
name() +
"|P" +
732 std::to_string(tmp.first.second) +
"> intermediate");
734 for (
const auto& tmp:
imR.allpairs)
735 tmp.second.print_size(
"<H" + std::to_string(tmp.first.first) +
"|" +
name() +
"|R" +
736 std::to_string(tmp.first.second) +
"> intermediate");
742 factory.set_info(
op->info);
750 std::shared_ptr<SeparatedConvolution<T,NDIM>>
get_op()
const {
return op;};
761 std::shared_ptr<SeparatedConvolution<T,NDIM>>
op;
768 void error(
const std::string& msg)
const {
770 std::cout <<
"\n\n!!!!ERROR in CCConvolutionOperator " <<
name() <<
": " << msg
771 <<
"!!!!!\n\n" << std::endl;
777template<
typename T, std::
size_t NDIM>
874 for (
const auto&
f :
v) {
910 void output(
const std::string& msg)
const {
912 std::cout <<
"Intermediate Potential Manager: " << msg <<
"\n";
928 std::vector<Function<double,3>>
U1;
965 records+=cloud.
store(world,
U2);
966 records+=cloud.
store(world,
U1);
985 U1=cloud.
forward_load<std::vector<Function<double,3>>>(world,recordlist);
1017 std::vector<unsigned char>
v;
1035 std::vector<unsigned char>
v=cloud.
forward_load<std::vector<unsigned char>>(
world,recordlist);
1037 bool function_is_assigned =
false, constant_part_is_initialized=
false;
1038 arin &
type &
ctype &
i &
j &
bsh_eps & function_is_assigned & constant_part_is_initialized;
1075 template<
typename Archive>
1078 bool fexist = (f_size > 0) && (
functions[0].get_function().is_initialized());
1081 if constexpr (Archive::is_input_archive) {
1089 if (fexist) ar &
functions[0].get_function();
1098 if (
f.is_assigned() and
f.is_pure())
f.get_function().
reconstruct();
1103 std::string fname=this->
name();
1104 if (verbose and
world.
rank()==0)
print(
"loading pair from file", fname);
1116 std::string fname =this->
name();
1117 if (verbose and
world.
rank()==0)
print(
"loading pair from file", fname);
1124 hashT hash_i = std::hash<std::size_t>{}(
i);
1146 std::string
name =
"???";
1222 template <
typename T,
size_t NDIM>
1231 if (do_print)
f.print_size(
name);
1232 if (
f.is_compressed()) {
1233 if (
world.
rank()==0 and do_print)
print(
"function is compressed -- reconstructing");
1235 if (do_print)
f.print_size(
name+
" reconstructed");
1261 template<
typename T, std::
size_t NDIM>
1267 if (vp.empty())
return;
1268 for (
const auto&
p : vp) {
1275 template<
typename... Args>
1276 void add(
const Args&... args) {
1282 world.
gop.
sum(size_global);
1300 const std::string
policy)
const override {
1302 for (
size_t i = 0; i < vsize1; i++) {
1304 p.push_back(std::make_pair(
batch,1.0));
1321 std::size_t n = std::get<0>(argtuple).size();
1322 resultT result = zero_functions_auto_tree_state<double, 6>(world, n);
1330 resultT operator() (
const std::vector<CCPair>& pair,
const Info& info,
const std::vector<std::string>& argument)
const;
1346 const std::string
policy)
const override {
1348 for (
size_t i = 0; i < vsize1; i++) {
1350 p.push_back(std::make_pair(
batch,1.0));
1359 name=
"ConstantPart";
1365 typedef std::tuple<const std::vector<CCPair>&,
1366 const std::vector<Function<double,3>>&,
const std::vector<Function<double,3>>&,
1372 std::size_t n = std::get<0>(argtuple).size();
1373 resultT result = zero_functions_auto_tree_state<double, 6>(world, n);
1379 const Info& info)
const;
1391 const std::string
policy)
const override {
1393 for (
size_t i = 0; i < vsize1; i++) {
1395 p.push_back(std::make_pair(
batch, 1.0));
1403 name=
"MP2UpdatePair";
1410 typedef std::tuple<const std::vector<CCPair>&,
const std::vector<real_function_6d>&,
1416 std::size_t n = std::get<0>(argtuple).size();
1417 resultT result = zero_functions_auto_tree_state<double, 6>(world, n);
1425 resultT operator() (
const std::vector<CCPair>& pair,
const std::vector<real_function_6d>& mp2_coupling,
1437 const std::string
policy)
const override {
1439 for (
size_t i = 0; i < vsize1; i++) {
1441 p.push_back(std::make_pair(
batch,1.0));
1453 const std::vector<CCPair>&,
1454 const std::vector<real_function_6d>&,
1464 std::size_t n = std::get<0>(argtuple).size();
1465 resultT result = zero_functions_auto_tree_state<double, 6>(world, n);
1481 const std::vector<real_function_6d>& local_coupling,
1485 const std::size_t&
maxiter)
const;
1493 name=
"SinglesPotentialEx";
1499 const std::vector<int>&,
1501 const std::vector<CCPair>&,
1503 const std::vector<CCPair>&,
1508 using resultT = std::tuple<std::vector<real_function_3d>,std::vector<real_function_3d>>;
1511 std::size_t n = std::get<0>(argtuple).size();
1512 std::vector<real_function_3d> result = zero_functions_auto_tree_state<double, 3>(world, n);
1513 std::vector<real_function_3d> intermediate = zero_functions_auto_tree_state<double, 3>(world, n);
1515 return std::make_tuple(result,intermediate);
1520 const std::vector<CCPair>& doubles_gs,
1522 const std::vector<CCPair>& doubles_ex,
1531 name=
"SinglesPotentialGs";
1535 const std::vector<int>&,
1537 const std::vector<CCPair>&,
1543 using resultT = std::tuple<std::vector<real_function_3d>,std::vector<real_function_3d>>;
1547 std::size_t n = std::get<0>(argtuple).size();
1548 std::vector<real_function_3d> result = zero_functions_auto_tree_state<double, 3>(world, n);
1549 std::vector<real_function_3d> intermediate = zero_functions_auto_tree_state<double, 3>(world, n);
1551 return std::make_tuple(result,intermediate);
1556 const std::vector<CCPair>& doubles_gs,
1566 name=
"CorrelationEnergy";
1570 const std::vector<CCPair>&,
1576 typedef std::vector<ScalarResult<double>>
resultT;
1581 std::size_t n = std::get<0>(argtuple).size();
1582 return scalar_result_vector<double>(world,n);
1587 const Info& info)
const;
double potential(const coord_3d &r)
Definition 3dharmonic.cc:132
Definition macrotaskpartitioner.h:55
a batch consists of a 2D-input batch and a 1D-output batch: K-batch <- (I-batch, J-batch)
Definition macrotaskpartitioner.h:124
Definition CCStructures.h:571
intermediateT< T, NDIM > imH
Definition CCStructures.h:762
CCConvolutionOperator(const CCConvolutionOperator &other)=default
friend hashT hash_value(CCConvolutionOperator< T, NDIM > &op)
Definition CCStructures.h:710
static std::shared_ptr< CCConvolutionOperator > CCConvolutionOperatorPtr(World &world, const OpType type, Parameters param)
Definition CCStructures.h:610
Function< T, NDIM > operator()(const Function< T, NDIM > &f) const
Definition CCStructures.h:637
SeparatedConvolution< T, NDIM > * init_op(const OpType &type, const Parameters ¶meters) const
Definition CCStructures.cc:370
size_t info() const
prints out information (operatorname, number of stored intermediates ...)
Definition CCStructures.cc:352
std::shared_ptr< SeparatedConvolution< T, NDIM > > op
Definition CCStructures.h:761
std::vector< Function< T, NDIM > > operator()(const std::vector< Function< T, NDIM > > &f) const
Definition CCStructures.h:669
OpType type() const
Definition CCStructures.h:746
void clear_intermediates(const FuncType &type)
Definition CCStructures.cc:330
World & world
the world
Definition CCStructures.h:754
void sanity() const
sanity check .. doens not do so much
Definition CCStructures.h:721
std::string name() const
Definition CCStructures.h:697
friend std::shared_ptr< CCConvolutionOperator > combine(const std::shared_ptr< CCConvolutionOperator > &a, const std::shared_ptr< CCConvolutionOperator > &b)
Definition CCStructures.h:626
void update_elements(const CC_vecfunction &bra, const CC_vecfunction &ket)
Definition CCStructures.cc:298
intermediateT< T, NDIM > imP
Definition CCStructures.h:763
void error(const std::string &msg) const
Definition CCStructures.h:768
const Parameters parameters
Definition CCStructures.h:748
CCConvolutionOperator(World &world, const OpType type, Parameters param)
Definition CCStructures.h:603
std::vector< Function< T, NDIM > > operator()(const CC_vecfunction &bra, const CCFunction< T, NDIM > &ket) const
Definition CCStructures.h:645
std::shared_ptr< SeparatedConvolution< T, NDIM > > get_op() const
Definition CCStructures.h:750
void print_intermediate(const FuncType type) const
Definition CCStructures.h:724
intermediateT< T, NDIM > imR
Definition CCStructures.h:764
TwoElectronFactory< T, 2 *NDIM > get_kernel() const
create a TwoElectronFactory with the operatorkernel
Definition CCStructures.h:740
friend CCConvolutionOperator combine(const CCConvolutionOperator &a, const CCConvolutionOperator &b)
Definition CCStructures.h:616
structure for a CC Function 3D which holds an index and a type
Definition ccpairfunction.h:45
Function< T, NDIM > function
Definition ccpairfunction.h:70
size_t i
Definition ccpairfunction.h:78
build an MP2 or CC2 or LRCC2 etc pair, possibly including the lo-rank parts
Definition CCStructures.h:1162
CCPairBuilder & set_ex_singles(const CC_vecfunction &ex)
provide excited state singles, needed for CC2 and LRCC2 wave function ansatz
Definition CCStructures.h:1173
CCPairBuilder(World &world, const Info &info)
Definition CCStructures.h:1164
CC_vecfunction gs_singles
Definition CCStructures.h:1249
const Info & info
Definition CCStructures.h:1248
CCPair make_bare_pair(const int i, const int j) const
make a CCPair without the 6d function and some bookkeeping information
Definition CCStructures.cc:506
CCPairBuilder & set_gs_singles(const CC_vecfunction &gs)
provide ground-state singles, needed for CC2 and LRCC2 wave function ansatz
Definition CCStructures.h:1167
static CCState cc_state(const CalcType &type)
Definition CCStructures.h:1192
CCPair make_bare_pair_from_file(const int i, const int j) const
make a CCPair with the 6d function only and some bookkeeping information
Definition CCStructures.cc:529
World & world
Definition CCStructures.h:1247
Function< T, NDIM > load_function(const std::string name, bool do_print) const
Function to load a function from disk.
Definition CCStructures.h:1223
CCPair make_pair(const int i, const int j, const std::vector< CCPairFunction< double, 6 > > &u) const
make a CCPair
Definition CCStructures.cc:520
CalcType ctype
Definition CCStructures.h:1250
CC_vecfunction ex_singles
Definition CCStructures.h:1249
CCPairBuilder & set_ctype(const CalcType &type)
Definition CCStructures.h:1178
CCPair complete_pair_with_low_rank_parts(const CCPair &pair) const
complete the given pair with the low-rank parts
Definition CCStructures.cc:545
Pairs< CCPair > make_all_bare_pairs() const
make a CCPair with the 6d function only and some bookkeeping information
Definition CCStructures.h:1201
a 6D function, either in full or low rank form, possibly including an 2-particle function
Definition ccpairfunction.h:373
Definition CCStructures.h:991
void reconstruct() const
reconstruct constant part and all functions
Definition CCStructures.h:1095
hashT hash() const
Definition CCStructures.h:1123
void info() const
Definition CCStructures.cc:84
std::string basename() const
return the base name like "MP2_pair_u" or "CC2_pair_x"
Definition CCStructures.h:1145
CCPair(const CCPair &other)
Definition CCStructures.h:1002
World & world() const
return the world of the function
Definition CCStructures.h:1047
void store_pair(World &world, const bool verbose=false)
Definition CCStructures.h:1115
Recordlist< Cloud::keyT > cloud_store(World &world, Cloud &cloud) const
customized function to store this to the cloud
Definition CCStructures.h:1015
bool function_exists() const
check if the pair has a function assigned
Definition CCStructures.h:1053
std::string name() const
Definition CCStructures.h:1152
CCPair(const size_t ii, const size_t jj, const CCState t, const CalcType c, const std::vector< CCPairFunction< double, 6 > > &f)
Definition CCStructures.h:998
size_t i
Definition CCStructures.h:1008
bool load_pair(World &world, const bool verbose=false)
Definition CCStructures.h:1102
real_function_6d function() const
gives back the pure 6D part of the pair function
Definition CCStructures.h:1058
real_function_6d constant_part
the constant part
Definition CCStructures.h:1136
std::vector< CCPairFunction< double, 6 > > functions
the functions which belong to the pair
Definition CCStructures.h:1133
CCPair(const size_t ii, const size_t jj, const CCState t, const CalcType c)
Definition CCStructures.h:995
void update_u(const real_function_6d &u)
updates the pure 6D part of the pair function
Definition CCStructures.h:1065
size_t j
Definition CCStructures.h:1009
CCState type
Definition CCStructures.h:1006
double bsh_eps
Definition CCStructures.h:1142
void serialize(const Archive &ar)
Definition CCStructures.h:1076
CalcType ctype
Definition CCStructures.h:1007
void cloud_load(World &world, const Cloud &cloud, Recordlist< Cloud::keyT > &recordlist)
customized function to load this from the cloud
Definition CCStructures.h:1033
cloud class
Definition cloud.h:147
recordlistT store(madness::World &world, const T &source)
Definition cloud.h:316
T forward_load(madness::World &world, recordlistT &recordlist) const
load a single object from the cloud, recordlist is consumed while loading elements
Definition cloud.h:298
FunctionDefaults holds default paramaters as static class members.
Definition funcdefaults.h:100
static const double & get_thresh()
Returns the default threshold.
Definition funcdefaults.h:176
A multiresolution adaptive numerical function.
Definition mra.h:139
World & world() const
Returns the world.
Definition mra.h:688
void set_thresh(double value, bool fence=true)
Sets the value of the truncation threshold. Optional global fence.
Definition mra.h:617
const std::shared_ptr< FunctionImpl< T, NDIM > > & get_impl() const
Returns a shared-pointer to the implementation.
Definition mra.h:654
const Function< T, NDIM > & reconstruct(bool fence=true) const
Reconstructs the function, transforming into scaling function basis. Possible non-blocking comm.
Definition mra.h:817
void clear(bool fence=true)
Clears the function as if constructed uninitialized. Optional fence.
Definition mra.h:889
bool is_initialized() const
Returns true if the function is initialized.
Definition mra.h:167
Definition CCStructures.h:1562
resultT operator()(const std::vector< CCPair > &pairs, const CC_vecfunction &singles_gs, const Info &info) const
Definition CCStructures.cc:748
std::tuple< const std::vector< CCPair > &, const CC_vecfunction &, const Info & > argtupleT
Definition CCStructures.h:1573
MacroTaskComputeCorrelationEnergy()
Definition CCStructures.h:1565
resultT allocator(World &world, const argtupleT &argtuple) const
allocate the result and set the name of this task
Definition CCStructures.h:1580
std::vector< ScalarResult< double > > resultT
first vector is the potential, second is an intermediate (if applicable, e.g. for s2b and s2c potenti...
Definition CCStructures.h:1576
std::string basename
Definition CCStructures.h:1564
Definition CCStructures.h:1341
partitionT do_partitioning(const std::size_t &vsize1, const std::size_t &vsize2, const std::string policy) const override
override this if you want your own partitioning
Definition CCStructures.h:1345
ConstantPartPartitioner()
Definition CCStructures.h:1343
Definition CCStructures.h:1339
std::vector< real_function_6d > resultT
Definition CCStructures.h:1369
MacroTaskConstantPart()
Definition CCStructures.h:1357
std::tuple< const std::vector< CCPair > &, const std::vector< Function< double, 3 > > &, const std::vector< Function< double, 3 > > &, const madness::Info & > argtupleT
Definition CCStructures.h:1367
resultT allocator(World &world, const argtupleT &argtuple) const
Definition CCStructures.h:1371
resultT operator()(const std::vector< CCPair > &pair, const std::vector< Function< double, 3 > > &gs_singles, const std::vector< Function< double, 3 > > &ex_singles, const Info &info) const
Definition CCStructures.cc:616
Definition CCStructures.h:1432
IteratePairPartitioner()=default
partitionT do_partitioning(const std::size_t &vsize1, const std::size_t &vsize2, const std::string policy) const override
override this if you want your own partitioning
Definition CCStructures.h:1436
Definition CCStructures.h:1430
std::tuple< const std::vector< CCPair > &, const std::vector< real_function_6d > &, const CC_vecfunction &, const CC_vecfunction &, const Info &, const std::size_t & > argtupleT
Definition CCStructures.h:1459
resultT operator()(const std::vector< CCPair > &pair, const std::vector< real_function_6d > &local_coupling, const CC_vecfunction &gs_singles, const CC_vecfunction &ex_singles, const Info &info, const std::size_t &maxiter) const
iterate a given pair of the MP2, CC2 or LRCC2 calculation
Definition CCStructures.cc:653
MacroTaskIteratePair()
Definition CCStructures.h:1447
resultT allocator(World &world, const argtupleT &argtuple) const
Definition CCStructures.h:1463
std::vector< real_function_6d > resultT
Definition CCStructures.h:1461
Definition CCStructures.h:1295
ConstantPartPartitioner()
Definition CCStructures.h:1297
partitionT do_partitioning(const std::size_t &vsize1, const std::size_t &vsize2, const std::string policy) const override
override this if you want your own partitioning
Definition CCStructures.h:1299
Definition CCStructures.h:1293
resultT operator()(const std::vector< CCPair > &pair, const Info &info, const std::vector< std::string > &argument) const
Definition CCStructures.cc:604
MacroTaskMp2ConstantPart()
Definition CCStructures.h:1311
std::tuple< const std::vector< CCPair > &, const madness::Info &, const std::vector< std::string > & > argtupleT
Definition CCStructures.h:1316
std::vector< real_function_6d > resultT
Definition CCStructures.h:1318
resultT allocator(World &world, const argtupleT &argtuple) const
Definition CCStructures.h:1320
Definition CCStructures.h:1384
UpdatePairPartitioner()
Definition CCStructures.h:1386
partitionT do_partitioning(const std::size_t &vsize1, const std::size_t &vsize2, const std::string policy) const override
override this if you want your own partitioning
Definition CCStructures.h:1390
Definition CCStructures.h:1382
MacroTaskMp2UpdatePair()
Definition CCStructures.h:1401
resultT allocator(World &world, const argtupleT &argtuple) const
Definition CCStructures.h:1415
std::vector< real_function_6d > resultT
Definition CCStructures.h:1413
std::tuple< const std::vector< CCPair > &, const std::vector< real_function_6d > &, const std::vector< madness::Vector< double, 3 > > &, const Info & > argtupleT
Definition CCStructures.h:1411
resultT operator()(const std::vector< CCPair > &pair, const std::vector< real_function_6d > &mp2_coupling, const std::vector< madness::Vector< double, 3 > > &all_coords_vec, const Info &info) const
Definition CCStructures.cc:635
Definition macrotaskq.h:978
Batch batch
Definition macrotaskq.h:980
std::shared_ptr< MacroTaskPartitioner > partitioner
Definition macrotaskq.h:982
std::string name
Definition macrotaskq.h:981
partition one (two) vectors into 1D (2D) batches.
Definition macrotaskpartitioner.h:182
std::string policy
how to partition the batches
Definition macrotaskpartitioner.h:190
std::list< std::pair< Batch, double > > partitionT
Definition macrotaskpartitioner.h:186
MacroTaskPartitioner & set_dimension(const std::size_t &n)
Definition macrotaskpartitioner.h:205
Definition CCStructures.h:1489
std::tuple< std::vector< real_function_3d >, std::vector< real_function_3d > > resultT
Definition CCStructures.h:1508
resultT allocator(World &world, const argtupleT &argtuple) const
Definition CCStructures.h:1510
std::tuple< const std::vector< int > &, const CC_vecfunction &, const std::vector< CCPair > &, const CC_vecfunction &, const std::vector< CCPair > &, const int &, const Info & > argtupleT
Definition CCStructures.h:1506
std::string basename
Definition CCStructures.h:1491
MacroTaskSinglesPotentialEx()
Definition CCStructures.h:1492
resultT operator()(const std::vector< int > &result_index, const CC_vecfunction &singles_gs, const std::vector< CCPair > &doubles_gs, const CC_vecfunction &singles_ex, const std::vector< CCPair > &doubles_ex, const int &name, const Info &info)
convenience function
Definition CCStructures.cc:674
Definition CCStructures.h:1527
std::string basename
Definition CCStructures.h:1529
std::tuple< const std::vector< int > &, const CC_vecfunction &, const std::vector< CCPair > &, const int &, const Info & > argtupleT
Definition CCStructures.h:1540
std::tuple< std::vector< real_function_3d >, std::vector< real_function_3d > > resultT
first vector is the potential, second is an intermediate (if applicable, e.g. for s2b and s2c potenti...
Definition CCStructures.h:1543
resultT operator()(const std::vector< int > &result_index, const CC_vecfunction &singles_gs, const std::vector< CCPair > &doubles_gs, const int &name, const Info &info)
Definition CCStructures.cc:716
MacroTaskSinglesPotentialGs()
Definition CCStructures.h:1530
resultT allocator(World &world, const argtupleT &argtuple) const
allocate the result and set the name of this task
Definition CCStructures.h:1546
Convolutions in separated form (including Gaussian)
Definition operator.h:139
static OperatorInfo combine_OT(const SeparatedConvolution< Q, NDIM > &left, const SeparatedConvolution< Q, NDIM > &right)
return operator type and other info of the combined operator (e.g. fg = f(1,2)* g(1,...
Definition operator.h:1706
A tensor is a multidimensional array.
Definition tensor.h:317
factory for generating TwoElectronInterfaces
Definition function_factory.h:443
A simple, fixed dimension vector.
Definition vector.h:64
void fence(bool debug=false)
Synchronizes all processes in communicator AND globally ensures no pending AM or tasks.
Definition worldgop.cc:161
std::vector< T > concat0(const std::vector< T > &v, size_t bufsz=1024 *1024)
Concatenate an STL vector of serializable stuff onto node 0.
Definition worldgop.h:955
void sum(T *buf, size_t nelem)
Inplace global sum while still processing AM & tasks.
Definition worldgop.h:872
A parallel world class.
Definition world.h:132
ProcessID rank() const
Returns the process rank in this World (same as MPI_Comm_rank()).
Definition world.h:320
WorldGopInterface & gop
Global operations.
Definition world.h:207
static std::enable_if_t< std::is_same< X, BinaryFstreamInputArchive >::value||std::is_same< X, BinaryFstreamOutputArchive >::value, bool > exists(World &world, const char *filename)
Returns true if the named, unopened archive exists on disk with read access.
Definition parallel_archive.h:224
An archive for storing local or parallel data wrapping a BinaryFstreamOutputArchive.
Definition parallel_archive.h:321
Objects that implement their own parallel archive interface should derive from this class.
Definition parallel_archive.h:58
Wraps an archive around an STL vector for output.
Definition vector_archive.h:55
double(* f1)(const coord_3d &)
Definition derivatives.cc:55
char * p(char *buf, const char *name, int k, int initial_level, double thresh, int order)
Definition derivatives.cc:72
static bool debug
Definition dirac-hatom.cc:16
auto T(World &world, response_space &f) -> response_space
Definition global_functions.cc:28
Tensor< typename Tensor< T >::scalar_type > arg(const Tensor< T > &t)
Return a new tensor holding the argument of each element of t (complex types only)
Definition tensor.h:2503
const int maxiter
Definition gygi_soltion.cc:68
static const double v
Definition hatom_sf_dirac.cc:20
static double u(double r, double c)
Definition he.cc:20
Tensor< double > op(const Tensor< double > &x)
Definition kain.cc:508
Declares the macrotaskq and MacroTaskBase classes.
#define MADNESS_CHECK(condition)
Check a condition — even in a release build the condition is always evaluated so it can have side eff...
Definition madness_exception.h:182
#define MADNESS_EXCEPTION(msg, value)
Macro for throwing a MADNESS exception.
Definition madness_exception.h:119
#define MADNESS_CHECK_THROW(condition, msg)
Check a condition — even in a release build the condition is always evaluated so it can have side eff...
Definition madness_exception.h:207
Main include file for MADNESS and defines Function interface.
Namespace for all elements and tools of MADNESS.
Definition DFParameters.h:10
long get_memory_usage()
Definition unique_filename.cc:29
static const char * filename
Definition legendre.cc:96
CalcType
Calculation Types used by CC2.
Definition CCParameters.h:14
@ CT_CC2
Definition CCParameters.h:15
@ CT_LRCC2
Definition CCParameters.h:15
@ CT_MP3
Definition CCParameters.h:15
@ CT_UNDEFINED
Definition CCParameters.h:15
@ CT_CISPD
Definition CCParameters.h:15
@ CT_MP2
Definition CCParameters.h:15
@ CT_ADC2
Definition CCParameters.h:15
static std::string stringify(T arg)
Definition funcplot.h:1039
static double cpu_time()
Returns the cpu time in seconds relative to an arbitrary origin.
Definition timers.h:127
void print_memory_usage(const World &world)
check memory usage using getrusage
Definition CCStructures.h:73
Function< TENSOR_RESULT_TYPE(Q, T), NDIM > mul(const Q alpha, const Function< T, NDIM > &f, bool fence=true)
Returns new function equal to alpha*f(x) with optional fence.
Definition mra.h:1765
void truncate(World &world, response_space &v, double tol, bool fence)
Definition basic_operators.cc:31
CCState
Type of Pairs used by CC_Pair2 class.
Definition CCStructures.h:27
@ CCSTATE_UNDEFINED
Definition CCStructures.h:28
@ GROUND_STATE
Definition CCStructures.h:28
@ EXCITED_STATE
Definition CCStructures.h:28
FuncType
Definition ccpairfunction.h:26
@ RESPONSE
Definition ccpairfunction.h:26
@ HOLE
Definition ccpairfunction.h:26
@ UNDEFINED
Definition ccpairfunction.h:26
@ PARTICLE
Definition ccpairfunction.h:26
@ reconstructed
s coeffs at the leaves only
Definition funcdefaults.h:60
CalcType assign_calctype(const std::string name)
Assigns enum to string.
Definition CCStructures.cc:396
const std::vector< Function< T, NDIM > > & reconstruct(const std::vector< Function< T, NDIM > > &v)
reconstruct a vector of functions
Definition vmra.h:162
std::shared_ptr< CCConvolutionOperator< T, NDIM > > CCConvolutionOperatorPtr(World &world, const OpType type, typename CCConvolutionOperator< T, NDIM >::Parameters param)
Definition CCStructures.h:778
void hash_combine(hashT &seed, const T &v)
Combine hash values.
Definition worldhash.h:260
Pairs< Function< T, NDIM > > intermediateT
f12 and g12 intermediates of the form <f1|op|f2> (with op=f12 or op=g12) will be saved using the pair...
Definition CCStructures.h:314
double size_of(const intermediateT< T, NDIM > &im)
Returns the size of an intermediate.
Definition CCStructures.h:322
std::vector< real_function_3d > vector_real_function_3d
Definition functypedefs.h:94
std::shared_ptr< FunctionFunctorInterface< double, 3 > > func(new opT(g))
OpType
operator types
Definition operatorinfo.h:11
void print(const T &t, const Ts &... ts)
Print items to std::cout (items separated by spaces) and terminate with a new line.
Definition print.h:225
response_space apply(World &world, std::vector< std::vector< std::shared_ptr< real_convolution_3d > > > &op, response_space &f)
Definition basic_operators.cc:43
std::string get_hostname()
Definition unique_filename.cc:40
NDIM & f
Definition mra.h:2481
std::size_t hashT
The hash value type.
Definition worldhash.h:145
double wall_time()
Returns the wall time in seconds relative to an arbitrary origin.
Definition timers.cc:48
std::string type(const PairType &n)
Definition PNOParameters.h:18
constexpr Vector< T, sizeof...(Ts)+1 > vec(T t, Ts... ts)
Factory function for creating a madness::Vector.
Definition vector.h:750
PotentialType
CC2 Singles Potentials.
Definition CCStructures.h:32
@ POT_singles_
Definition CCStructures.h:49
@ POT_s2b_
Definition CCStructures.h:41
@ POT_F3D_
Definition CCStructures.h:34
@ POT_s5b_
Definition CCStructures.h:39
@ POT_s4c_
Definition CCStructures.h:45
@ POT_s5c_
Definition CCStructures.h:40
@ POT_s6_
Definition CCStructures.h:46
@ POT_s4b_
Definition CCStructures.h:44
@ POT_ccs_
Definition CCStructures.h:47
@ POT_s3c_
Definition CCStructures.h:37
@ POT_s4a_
Definition CCStructures.h:43
@ POT_s5a_
Definition CCStructures.h:38
@ POT_s3b_
Definition CCStructures.h:36
@ POT_s2c_
Definition CCStructures.h:42
@ POT_cis_
Definition CCStructures.h:48
@ POT_UNDEFINED
Definition CCStructures.h:33
@ POT_s3a_
Definition CCStructures.h:35
std::string assign_name(const CCState &input)
Assigns strings to enums for formated output.
Definition CCStructures.cc:379
std::string name(const FuncType &type, const int ex=-1)
Definition ccpairfunction.h:28
madness::hashT hash_value(const std::array< T, N > &a)
Hash std::array with madness hash.
Definition array_addons.h:78
void save(const Function< T, NDIM > &f, const std::string name)
Definition mra.h:2835
double get_size(World &world, const std::vector< Function< T, NDIM > > &v)
Definition vmra.h:1851
double get_size_local(World &world, const std::vector< Function< T, NDIM > > &v)
return the size of a vector of functions for each rank
Definition vmra.h:1832
static const double b
Definition nonlinschro.cc:119
static const double a
Definition nonlinschro.cc:118
static const double c
Definition relops.cc:10
parameter class
Definition CCStructures.h:575
int freeze
Definition CCStructures.h:590
Parameters()
Definition CCStructures.h:576
void serialize(archiveT &ar)
f12 exponent
Definition CCStructures.h:594
double lo
Definition CCStructures.h:589
double gamma
Definition CCStructures.h:591
Parameters(const CCParameters ¶m)
Definition CCStructures.h:585
double thresh_op
Definition CCStructures.h:588
Parameters(const Parameters &other)
Definition CCStructures.h:578
Definition CCStructures.h:86
void print_warnings() const
Definition CCStructures.h:112
void warning(const std::string &msg) const
Definition CCStructures.cc:45
void debug_output(const std::string &msg) const
Definition CCStructures.h:96
bool debug
Definition CCStructures.h:92
void subsection(const std::string &msg) const
Definition CCStructures.cc:34
CCMessenger operator<<(const T &t) const
Definition CCStructures.h:117
World & world
Definition CCStructures.h:89
bool scientific
Definition CCStructures.h:91
std::vector< std::string > warnings
collect all warnings that occur to print out at the end of the job
Definition CCStructures.h:124
std::ostream & os
output stream
Definition CCStructures.h:126
void section(const std::string &msg) const
Definition CCStructures.cc:23
void output(const std::string &msg) const
Definition CCStructures.cc:14
CCMessenger(World &world)
Definition CCStructures.h:87
size_t output_prec
Definition CCStructures.h:90
void operator()(const std::string &msg) const
Definition CCStructures.h:94
Definition CCParameters.h:20
long freeze() const
Definition CCParameters.h:181
bool debug() const
Definition CCParameters.h:165
print accumulated size of all functions
Definition CCStructures.h:1256
void print(World &world, const std::string msg="") const
Definition CCStructures.h:1280
double size_local
Definition CCStructures.h:1257
void add_helper(const std::vector< CCPair > &vp)
Definition CCStructures.h:1266
void add(const Args &... args)
variadic template parameters to add the size of all functions and pairs
Definition CCStructures.h:1276
void add_helper(const std::vector< Function< T, NDIM > > &v)
Definition CCStructures.h:1262
Timer Structure.
Definition CCStructures.h:131
std::pair< double, double > current_time(bool printout=false)
Definition CCStructures.h:184
CCTimer(World &world, std::string msg)
Definition CCStructures.h:135
void print() const
Definition CCStructures.h:193
std::string operation
Definition CCStructures.h:142
double start_cpu
Definition CCStructures.h:141
double time_wall
Definition CCStructures.h:145
double get_wall_time_diff() const
Definition CCStructures.h:180
void update_time()
Definition CCStructures.h:148
CCTimer stop()
Definition CCStructures.h:164
CCTimer start()
Definition CCStructures.h:158
void print()
Definition CCStructures.h:189
double time_cpu
Definition CCStructures.h:146
double start_wall
Definition CCStructures.h:140
double get_cpu_time_diff() const
Definition CCStructures.h:182
void print(const std::pair< double, double > ×) const
Definition CCStructures.h:197
double reset()
Definition CCStructures.h:172
World & world
Definition CCStructures.h:139
double end_wall
Definition CCStructures.h:143
double end_cpu
Definition CCStructures.h:144
A helper structure which holds a map of functions.
Definition CCStructures.h:334
std::string irrep
Definition CCStructures.h:477
hashT hash() const
Definition CCStructures.h:463
CC_vecfunction(const vector_real_function_3d &v, const FuncType &type, const size_t &freeze)
Definition CCStructures.h:361
std::map< std::size_t, CCFunction< double, 3 > > CC_functionmap
Definition CCStructures.h:470
CC_vecfunction(const CC_vecfunction &other)
copy ctor (shallow)
Definition CCStructures.h:377
const CCFunction< double, 3 > & operator()(const CCFunction< double, 3 > &i) const
getter
Definition CCStructures.h:489
void insert(const size_t &i, const CCFunction< double, 3 > &f)
setter
Definition CCStructures.h:509
CC_vecfunction(const vector_real_function_3d &v, const FuncType &type)
Definition CCStructures.h:353
static CC_vecfunction load_restartdata(World &world, std::string filename)
Definition CCStructures.h:425
const CCFunction< double, 3 > & operator()(const size_t &i) const
getter
Definition CCStructures.h:494
CC_vecfunction operator*(const double &fac) const
scalar multiplication
Definition CCStructures.h:539
double delta
Definition CCStructures.h:476
CCFunction< double, 3 > & operator()(const size_t &i)
getter
Definition CCStructures.h:504
void print_size(const std::string &msg="!?not assigned!?") const
Print the memory of which is used by all the functions in the map.
Definition CCStructures.cc:69
CC_vecfunction(const std::vector< CCFunction< double, 3 > > &v)
Definition CCStructures.h:347
bool operator<(const CC_vecfunction &b) const
operator needed for sort operation (sorted by omega values)
Definition CCStructures.h:553
CC_vecfunction(const FuncType type_)
Definition CCStructures.h:338
void serialize(const Archive &ar)
Definition CCStructures.h:438
CC_vecfunction(const std::vector< CCFunction< double, 3 > > &v, const FuncType type_)
Definition CCStructures.h:371
CC_vecfunction(const vector_real_function_3d &v)
Definition CCStructures.h:340
CC_vecfunction()
Definition CCStructures.h:336
void set_functions(const vector_real_function_3d &v, const FuncType &type, const size_t &freeze)
setter
Definition CCStructures.h:514
CCFunction< double, 3 > & operator()(const CCFunction< double, 3 > &i)
getter
Definition CCStructures.h:499
void scale(const double &factor)
scaling (inplace)
Definition CCStructures.h:546
void plot(const std::string &msg="") const
Definition CCStructures.h:556
FuncType type
Definition CCStructures.h:473
double current_error
excitation energy
Definition CCStructures.h:475
friend CC_vecfunction copy(const CC_vecfunction &other)
returns a deep copy (void shallow copy errors)
Definition CCStructures.h:399
size_t size() const
Get the size vector (number of functions in the map)
Definition CCStructures.h:530
CC_vecfunction & operator=(const CC_vecfunction &other)
assignment operator, shallow wrt the functions
Definition CCStructures.h:385
double omega
Definition CCStructures.h:474
bool is_converged(const double econv, const double dconv) const
Definition CCStructures.h:484
vector_real_function_3d get_vecfunction() const
Returns all the functions of the map as vector.
Definition CCStructures.h:523
void save_restartdata(World &world, std::string filename) const
Definition CCStructures.h:432
CC_functionmap functions
Definition CCStructures.h:471
std::string name(const int ex) const
excitation irrep (direct product of x function and corresponding orbital)
Definition CCStructures.h:480
void reconstruct() const
Definition CCStructures.h:408
POD holding some basic functions and some intermediates for the CC2 calculation.
Definition CCStructures.h:919
void reconstruct() const
Definition CCStructures.h:942
std::vector< Function< double, 3 > > mo_ket
Definition CCStructures.h:920
Recordlist< Cloud::keyT > cloud_store(World &world, Cloud &cloud) const
customized function to store this to the cloud
Definition CCStructures.h:955
Tensor< double > fock
Definition CCStructures.h:925
Function< double, 3 > R
Definition CCStructures.h:927
std::vector< Function< double, 3 > > U1
Definition CCStructures.h:928
CCIntermediatePotentials intermediate_potentials
Definition CCStructures.h:926
Function< double, 3 > U2
Definition CCStructures.h:927
std::vector< madness::Vector< double, 3 > > molecular_coordinates
Definition CCStructures.h:922
std::vector< double > orbital_energies
Definition CCStructures.h:924
std::vector< Function< double, 3 > > mo_bra
Definition CCStructures.h:921
CCParameters parameters
Definition CCStructures.h:923
vector_real_function_3d get_active_mo_bra() const
Definition CCStructures.h:936
void cloud_load(World &world, const Cloud &cloud, Recordlist< Cloud::keyT > &recordlist)
customized function to load this from the cloud
Definition CCStructures.h:974
Function< double, 3 > R_square
Definition CCStructures.h:927
vector_real_function_3d get_active_mo_ket() const
Definition CCStructures.h:930
Definition CCStructures.h:206
static PairVectorMap triangular_map(const int nfreeze, const int nocc)
Definition CCStructures.h:212
void print(const std::string msg="PairVectorMap") const
Definition CCStructures.h:232
std::vector< std::pair< int, int > > map
maps pair index (i,j) to vector index k
Definition CCStructures.h:208
static PairVectorMap quadratic_map(const int nfreeze, const int nocc)
Definition CCStructures.h:222
PairVectorMap(const std::vector< std::pair< int, int > > map1)
Definition CCStructures.h:210
Definition CCStructures.h:246
void swap(Pairs< T > &other)
swap the contant of the pairmap
Definition CCStructures.h:302
pairmapT allpairs
Definition CCStructures.h:249
static Pairs vector2pairs(const std::vector< T > &argument, const PairVectorMap map)
Definition CCStructures.h:265
Pairs< R > convert(const Pairs< T > arg, const opT op) const
convert Pairs<T> to another type
Definition CCStructures.h:255
std::map< std::pair< int, int >, T > pairmapT
Definition CCStructures.h:248
bool empty() const
Definition CCStructures.h:306
T & operator()(int i, int j)
getter
Definition CCStructures.h:289
void insert(int i, int j, const T &pair)
Definition CCStructures.h:296
const T & operator()(int i, int j) const
getter
Definition CCStructures.h:282
static std::vector< T > pairs2vector(const Pairs< T > &argument, const PairVectorMap map)
Definition CCStructures.h:273
Definition lowrankfunction.h:336
static double current_time
Definition tdse1d.cc:160
InputParameters param
Definition tdse.cc:203
double norm(const T i1)
Definition test_cloud.cc:72
constexpr std::size_t NDIM
Definition testgconv.cc:54
double h(const coord_1d &r)
Definition testgconv.cc:175