Function gpu_bindLocalGPUsToNodes

quest/src/gpu/gpu_config.cpp:370–391

void gpu_bindLocalGPUsToNodes() {
#if COMPILE_CUDA

    // distribute local MPI processes across local GPUs;
    int numLocalGpus = gpu_getNumberOfLocalGpus();
    int localGpuInd = comm_getRank() % numLocalGpus;
    CUDA_CHECK( cudaSetDevice(localGpuInd) );

    // note it is possible for multiple MPI processes
    // to bind to the same local GPU (as can be assessed
    // with gpu_doAnyMpiProcessesShareLocalGpu()), but
    // this will incur slowdowns due to context-switching
    // and has no benefit - is if further illegal when
    // using cuStateVec, as the caller will validate

    // indicate that other GPU-queries are now legal
    hasGpuBeenBound = true;

#else
    error_gpuQueriedButGpuNotCompiled();
#endif 
}


bool gpu_areAnyNodesBoundToSameGpu() {