diff --git a/Benchmark/cholesky_gemm/gemm.cpp b/Benchmark/cholesky_gemm/gemm.cpp index c12c65e121bfa9592934f4f4dbbc4c27288f4e80..5b6875ca20da64345bb908b8684d1e12ff805e89 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