1 #ifndef GLOBAL_GPU_MGR_H 2 #define GLOBAL_GPU_MGR_H 12 #include "GlobalGPUMgr.decl.h" 14 #if defined(NAMD_CUDA) || defined(NAMD_HIP) 41 cpu_set_t* current =
nullptr;
42 cpu_set_t* temp =
nullptr;
44 bool validCpu =
false;
63 return CkpvAccessOther(GlobalGPUMgr_instance, CmiRankOf(pe));
134 int deviceIndex = -1;
137 std::vector<int> deviceIndexToPeMap;
138 std::vector<int> peToDeviceIDMap;
139 std::vector<int> peToDeviceIndexMap;
static constexpr int kNvshmemOffset
void initializeBackends()
const std::vector< int > & getDeviceIndexToPeMap() const
RunWithAffinity(const bool valid, const int tempCpu)
static constexpr int kNcclOffset
static GlobalGPUMgr * ObjectOnPe(const int pe)
int getMasterPeForDeviceIndex(int device)
static GlobalGPUMgr * Object()
static constexpr int kCudaOffset