From 95b7b14ca687b71d91e47eacf7d1d0e3b12eff7a Mon Sep 17 00:00:00 2001
From: Berenger Bramas <Berenger.Bramas@inria.fr>
Date: Wed, 30 Oct 2024 17:07:45 +0100
Subject: [PATCH] print nb cpus and cudas
---
Benchmark/cholesky_gemm/gemm.cpp | 3 +++
1 file changed, 3 insertions(+)
diff --git a/Benchmark/cholesky_gemm/gemm.cpp b/Benchmark/cholesky_gemm/gemm.cpp
index c12c65e..5b6875c 100644
--- a/Benchmark/cholesky_gemm/gemm.cpp
+++ b/Benchmark/cholesky_gemm/gemm.cpp
@@ -58,6 +58,9 @@ auto gemm(const int NbLoops, SpBlas::Block blocksC[], const SpBlas::Block blocks
scheduler = std::unique_ptr<SpAbstractScheduler>(new SpMultiPrioScheduler<MaxNbDevices,FavorLocality>(nbGpu*SpCudaUtils::GetDefaultNbStreams()));
}
SpComputeEngine ce(SpWorkerTeamBuilder::TeamOfCpuGpuWorkers(SpUtils::DefaultNumThreads(), nbGpu), std::move(scheduler));
+ std::cout << "Compute engine:" << std::endl;
+ std::cout << " - nb cpus: " << ce.getNbCpuWorkers() << std::endl;
+ std::cout << " - nb gpus: " << ce.getNbCudaWorkers() << std::endl;
#else
SpComputeEngine ce(SpWorkerTeamBuilder::TeamOfCpuWorkers());
#endif
--
GitLab