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

Update test bench for CUDA

Browse Source
This commit is contained in:
Gallo Alejandro 2022-08-12 18:28:20 +02:00
parent 4101c89907
commit 4651231d3b
1 changed files with 22 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
bench/test_main.cxx
View File

@ -45,6 +45,16 @@ int main(int argc, char** argv) {
checkpoint_percentage,
"Percentage for checkpoints");
#if defined(HAVE_CUDA)
size_t ooo_threads = 0, ooo_blocks = 0;
app.add_option("--ooo-blocks",
ooo_blocks,
"CUDA: Number of blocks per block for kernels going through ooo tensors");
app.add_option("--ooo-threads",
ooo_threads,
"CUDA: Number of threads per block for kernels going through ooo tensors");
#endif
CLI11_PARSE(app, argc, argv);
CTF::World world(argc, argv);
@ -154,15 +164,24 @@ int main(int argc, char** argv) {
.with_checkpointAtPercentage(checkpoint_percentage)
.with_checkpointPath(checkpoint_path)
.with_readCheckpointIfExists(!noCheckpoint)
#if defined(HAVE_CUDA)
.with_oooThreads(ooo_threads)
.with_oooBlocks(ooo_blocks)
#endif
;
try {
auto out = atrip::Atrip::run(in);
if (atrip::Atrip::rank == 0)
std::cout << "Energy: " << out.energy << std::endl;
} catch (const char* msg) {
if (atrip::Atrip::rank == 0)
std::cout << "Atrip throwed with msg:\n\t\t " << msg << "\n";
}
if (!in.deleteVppph)
delete Vppph;
if (atrip::Atrip::rank == 0)
std::cout << "Energy: " << out.energy << std::endl;
MPI_Finalize();
return 0;