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

Add support for reading tensors from file in atrip bench

Browse Source
This commit is contained in:
Alejandro Gallo 2022-12-06 21:20:03 +01:00
parent 9087e3af19
commit 1d96800d45
2 changed files with 74 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.org
View File

@ -117,13 +117,13 @@ Notice that you can give a path for ctf for all of them by doing
** Main benchmark ** Main benchmark
The main benchmark gets built in =bench/main= and is used to run an The main benchmark gets built in =bench/atrip= and is used to run an
atrip run with random tensors. atrip run with random tensors.
A common run of this script will be the following A common run of this script will be the following
#+begin_src sh #+begin_src sh
bench/main \ bench/atrip \
--no 100 \ --no 100 \
--nv 1000 \ --nv 1000 \
--mod 1 \ --mod 1 \

102
bench/main.cxx
View File

@ -5,13 +5,15 @@
#include <CLI11.hpp> #include <CLI11.hpp>
#define _print_size(what, size) \ #define _print_size(what, size) \
if (rank == 0) { \ do { \
std::cout << #what \ if (rank == 0) { \
<< " => " \ std::cout << #what \
<< (double)size * elem_to_gb \ << " => " \
<< "GB" \ << (double)size * elem_to_gb \
<< std::endl; \ << "GB" \
} << std::endl; \
} \
} while (0)
int main(int argc, char** argv) { int main(int argc, char** argv) {
MPI_Init(&argc, &argv); MPI_Init(&argc, &argv);
@ -48,6 +50,19 @@ int main(int argc, char** argv) {
checkpoint_percentage, checkpoint_percentage,
"Percentage for checkpoints"); "Percentage for checkpoints");
// Optional tensor files
std::string
ei_path, ea_path,
Tph_path, Tpphh_path,
Vpphh_path, Vhhhp_path, Vppph_path;
app.add_option("--ei", ei_path, "Path for ei");
app.add_option("--ea", ea_path, "Path for ea");
app.add_option("--Tpphh", Tpphh_path, "Path for Tpphh");
app.add_option("--Tph", Tph_path, "Path for Tph");
app.add_option("--Vpphh", Vpphh_path, "Path for Vpphh");
app.add_option("--Vhhhp", Vhhhp_path, "Path for Vhhhp");
app.add_option("--Vppph", Vppph_path, "Path for Vppph");
#if defined(HAVE_CUDA) #if defined(HAVE_CUDA)
size_t ooo_threads = 0, ooo_blocks = 0; size_t ooo_threads = 0, ooo_blocks = 0;
app.add_option("--ooo-blocks", app.add_option("--ooo-blocks",
@ -151,37 +166,64 @@ int main(int argc, char** argv) {
} }
std::vector<int> symmetries(4, NS) std::vector<int>
, vo({nv, no}) symmetries(4, NS),
, vvoo({nv, nv, no, no}) vo({nv, no}),
, ooov({no, no, no, nv}) vvoo({nv, nv, no, no}),
, vvvo({nv, nv, nv, no}) ooov({no, no, no, nv}),
; vvvo({nv, nv, nv, no});
CTF::Tensor<double> CTF::Tensor<double>
ei(1, ooov.data(), symmetries.data(), world) ei(1, ooov.data(), symmetries.data(), world),
, ea(1, vo.data(), symmetries.data(), world) ea(1, vo.data(), symmetries.data(), world),
, Tph(2, vo.data(), symmetries.data(), world) Tph(2, vo.data(), symmetries.data(), world),
, 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);
;
// initialize deletable tensors in heap // initialize deletable tensors in heap
auto Vppph auto Vppph
= new CTF::Tensor<double>(4, vvvo.data(), symmetries.data(), world); = 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);
ei.fill_random(-40.0, -2); if (ei_path.size()) {
ea.fill_random(2, 50); ei.read_dense_from_file(ei_path.c_str());
Tpphh.fill_random(0, 1); } else {
Tph.fill_random(0, 1); ei.fill_random(-40.0, -2);
Vpphh.fill_random(0, 1); }
Vhhhp.fill_random(0, 1); if (ea_path.size()) {
Vppph->fill_random(0, 1); ea.read_dense_from_file(ea_path.c_str());
} else {
ea.fill_random(2, 50);
}
if (Tpphh_path.size()) {
Tpphh.read_dense_from_file(Tpphh_path.c_str());
} else {
Tpphh.fill_random(0, 1);
}
if (Tph_path.size()) {
Tph.read_dense_from_file(Tph_path.c_str());
} else {
Tph.fill_random(0, 1);
}
if (Vpphh_path.size()) {
Vpphh.read_dense_from_file(Vpphh_path.c_str());
} else {
Vpphh.fill_random(0, 1);
}
if (Vhhhp_path.size()) {
Vhhhp.read_dense_from_file(Vhhhp_path.c_str());
} else {
Vhhhp.fill_random(0, 1);
}
if (Vppph_path.size()) {
Vppph->read_dense_from_file(Vppph_path.c_str());
} else {
Vppph->fill_random(0, 1);
}
atrip::Atrip::init(MPI_COMM_WORLD); atrip::Atrip::init(MPI_COMM_WORLD);
const auto in const auto in