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

Fix bench for deleting Vppph

Browse Source
This commit is contained in:
Alejandro Gallo 2022-04-05 17:32:21 +02:00
parent fed19ff52c
commit 78d67070c0
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
bench/test_main.cxx
View File

@ -97,9 +97,12 @@ int main(int argc, char** argv) {
, Tpphh(4, vvoo.data(), symmetries.data(), world) , Tpphh(4, vvoo.data(), symmetries.data(), world)
, Vpphh(4, vvoo.data(), symmetries.data(), world) , Vpphh(4, vvoo.data(), symmetries.data(), world)
, Vhhhp(4, ooov.data(), symmetries.data(), world) , Vhhhp(4, ooov.data(), symmetries.data(), world)
, Vppph(4, vvvo.data(), symmetries.data(), world)
; ;
// initialize deletable tensors in heap
auto Vppph
= new CTF::Tensor<double>(4, vvvo.data(), symmetries.data(), world);
_print_size(Vabci, no*nv*nv*nv) _print_size(Vabci, no*nv*nv*nv)
_print_size(Vabij, no*no*nv*nv) _print_size(Vabij, no*no*nv*nv)
_print_size(Vijka, no*no*no*nv) _print_size(Vijka, no*no*no*nv)
@ -110,7 +113,7 @@ int main(int argc, char** argv) {
Tph.fill_random(0, 1); Tph.fill_random(0, 1);
Vpphh.fill_random(0, 1); Vpphh.fill_random(0, 1);
Vhhhp.fill_random(0, 1); Vhhhp.fill_random(0, 1);
Vppph.fill_random(0, 1); Vppph->fill_random(0, 1);
atrip::Atrip::init(); atrip::Atrip::init();
const auto in const auto in
@ -122,7 +125,7 @@ int main(int argc, char** argv) {
.with_Tabij(&Tpphh) .with_Tabij(&Tpphh)
.with_Vabij(&Vpphh) .with_Vabij(&Vpphh)
.with_Vijka(&Vhhhp) .with_Vijka(&Vhhhp)
.with_Vabci(&Vppph) .with_Vabci(Vppph)
// some options // some options
.with_deleteVppph(!keepVppph) .with_deleteVppph(!keepVppph)
.with_barrier(barrier) .with_barrier(barrier)
@ -135,6 +138,9 @@ int main(int argc, char** argv) {
auto out = atrip::Atrip::run(in); auto out = atrip::Atrip::run(in);
if (!in.deleteVppph)
delete Vppph;
if (atrip::Atrip::rank == 0) if (atrip::Atrip::rank == 0)
std::cout << "Energy: " << out.energy << std::endl; std::cout << "Energy: " << out.energy << std::endl;