NAMD
|
#include <cuda_runtime.h>
#include <cuda.h>
#include "common.h"
#include "charm++.h"
#include "DeviceCUDA.h"
#include "WorkDistrib.h"
#include "CudaUtils.h"
#include "Node.h"
#include "SimParameters.h"
Go to the source code of this file.
Classes | |
struct | cuda_args_t |
Macros | |
#define | MAX_NUM_RANKS 2048 |
#define | MAX_NUM_DEVICES 256 |
#define | REGISTER_DEVICE_EVENTS(DEV) |
Functions | |
void | cuda_finalize () |
void | cuda_initialize () |
void | cuda_getargs (char **argv) |
Variables | |
__thread DeviceCUDA * | deviceCUDA |
static __thread cuda_args_t | cuda_args |
int | deviceIDList [MAX_NUM_RANKS] |
int | masterPeList [MAX_NUM_DEVICES] |
#define MAX_NUM_DEVICES 256 |
Definition at line 92 of file DeviceCUDA.C.
Referenced by DeviceCUDA::initialize().
#define MAX_NUM_RANKS 2048 |
Definition at line 89 of file DeviceCUDA.C.
Referenced by DeviceCUDA::initialize().
#define REGISTER_DEVICE_EVENTS | ( | DEV | ) |
void cuda_finalize | ( | ) |
void cuda_getargs | ( | char ** | argv | ) |
Definition at line 67 of file DeviceCUDA.C.
References cuda_args, cuda_args_t::devicelist, cuda_args_t::devicesperreplica, cuda_args_t::globaldevice, cuda_args_t::ignoresharing, cuda_args_t::mergegrids, NAMD_die(), cuda_args_t::nomergegrids, cuda_args_t::nostreaming, cuda_args_t::pmedevice, cuda_args_t::pmePes, and cuda_args_t::usedevicelist.
Referenced by all_init().
void cuda_initialize | ( | ) |
Definition at line 27 of file DeviceCUDA.C.
References cuda_finalize(), deviceCUDA, and DeviceCUDA::initialize().
Referenced by WorkDistrib::peOrderingReady().
|
static |
Definition at line 65 of file DeviceCUDA.C.
Referenced by cuda_getargs(), and DeviceCUDA::initialize().
__thread DeviceCUDA* deviceCUDA |
Definition at line 23 of file DeviceCUDA.C.
Referenced by CudaComputeNonbonded::assignPatches(), CudaPmeOneDevice::checkPatchLevelSimParamCompatibility(), ComputeMgr::createComputes(), ComputeCUDAMgr::createCudaComputeNonbonded(), ComputeCUDAMgr::createCudaGlobalMaster(), ComputeCUDAMgr::createCudaPmeOneDevice(), cuda_initialize(), ComputePme::doWork(), ComputeCUDAMgr::getCudaComputeNonbonded(), globalAtomIDToSOAID(), ComputeCUDAMgr::initialize(), CudaComputeNonbonded::initialize(), ComputePmeMgr::initialize(), ComputePmeMgr::initialize_computes(), ComputePmeMgr::initialize_pencils(), ComputePmeCUDAMgr::initialize_pencils(), ComputePmeCUDAMgr::isPmeDevice(), ComputeMgr::recvCudaGlobalMasterCreateMsg(), ComputeMgr::recvCudaGlobalMasterRemoveMsg(), ComputeMgr::recvCudaGlobalMasterUpdateMsg(), Sequencer::Sequencer(), ComputePmeCUDAMgr::setupPencils(), ComputePmeMgr::ungridCalc(), and ComputeCUDAMgr::update().
int deviceIDList[MAX_NUM_RANKS] |
Definition at line 90 of file DeviceCUDA.C.
Referenced by DeviceCUDA::getDeviceIDforPe(), and DeviceCUDA::initialize().
int masterPeList[MAX_NUM_DEVICES] |
Definition at line 93 of file DeviceCUDA.C.
Referenced by DeviceCUDA::getMasterPeForDeviceID(), and DeviceCUDA::initialize().