Skip to content
Snippets Groups Projects
Commit 924dbbd4 authored by Daniel Vonk's avatar Daniel Vonk Committed by Daniel Vonk
Browse files

Only run cuda device on rank 0

parent b56c42a9
No related branches found
No related tags found
2 merge requests!19Merge develop into main,!17Add CUDA backend
......@@ -156,10 +156,9 @@ ScatterDeviceFactory::create(boost::mpi::communicator &scatter_comm, Sample &sam
// all_comm for inter-partition communication, parition_comm for
// intra-partition communication
if (Params::Inst()->scattering.type == "self") {
if (Params::Inst()->device == sass::platform::Device::CUDA) {
if (scatter_comm.rank() == 0) {
Info::Inst()->write("Initializing CUDA Scatter Device, Vectors (self)");
}
// Only rank 0 should claim the CUDA device
if (Params::Inst()->device == sass::platform::Device::CUDA && scatter_comm.rank() == 0) {
Info::Inst()->write("Initializing CUDA Scatter Device, Vectors (self)");
p_ScatterDevice = new SelfVectorsCUDAScatterDevice(
all_comm, partition_comm, sample, thispartition_QIV, NAF, fileservice_endpoint,
monitorservice_endpoint);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment