NAMD
|
#include "common.h"
#include "charm++.h"
#include "DeviceCUDA.h"
#include "WorkDistrib.h"
#include "CudaUtils.h"
#include <cuda_runtime.h>
#include <cuda.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_initialize () |
void | cuda_finalize () |
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 74 of file DeviceCUDA.C.
Referenced by DeviceCUDA::initialize().
#define MAX_NUM_RANKS 2048 |
Definition at line 71 of file DeviceCUDA.C.
Referenced by DeviceCUDA::initialize().
#define REGISTER_DEVICE_EVENTS | ( | DEV | ) |
void cuda_finalize | ( | ) |
Definition at line 30 of file DeviceCUDA.C.
Referenced by BackEnd::exit().
void cuda_getargs | ( | char ** | argv | ) |
Definition at line 55 of file DeviceCUDA.C.
References cuda_args, cuda_args_t::devicelist, cuda_args_t::devicesperreplica, cuda_args_t::ignoresharing, cuda_args_t::mergegrids, NAMD_die(), cuda_args_t::nomergegrids, cuda_args_t::nostreaming, and cuda_args_t::usedevicelist.
Referenced by all_init().
void cuda_initialize | ( | ) |
Definition at line 24 of file DeviceCUDA.C.
References deviceCUDA, and DeviceCUDA::initialize().
Referenced by WorkDistrib::peOrderingReady().
|
static |
Definition at line 53 of file DeviceCUDA.C.
Referenced by cuda_getargs(), and DeviceCUDA::initialize().
__thread DeviceCUDA* deviceCUDA |
Definition at line 22 of file DeviceCUDA.C.
Referenced by ComputeNonbondedCUDA::assignPatches(), CudaComputeNonbonded::assignPatches(), ComputeBondedCUDAKernel::bondedForce(), build_cuda_exclusions(), build_cuda_force_table(), CudaTileListKernel::buildTileLists(), ComputePmeMgr::chargeGridSubmitted(), ComputeNonbondedCUDA::ComputeNonbondedCUDA(), ComputeMgr::createComputes(), ComputeCUDAMgr::createCudaComputeNonbonded(), cuda_check_local_calc(), cuda_check_remote_calc(), cuda_check_remote_progress(), cuda_initialize(), ComputePme::doWork(), ComputeNonbondedCUDA::doWork(), ComputeNonbondedCUDA::finishWork(), CudaComputeGBISKernel::GBISphase1(), CudaComputeGBISKernel::GBISphase2(), CudaComputeGBISKernel::GBISphase3(), ComputeCUDAMgr::getCudaComputeNonbonded(), ComputeCUDAMgr::initialize(), ComputePmeMgr::initialize(), ComputePmeMgr::initialize_computes(), ComputePmeMgr::initialize_pencils(), ComputePmeCUDAMgr::initialize_pencils(), ComputePmeCUDAMgr::isPmeDevice(), CudaComputeNonbondedKernel::nonbondedForce(), ComputeNonbondedCUDA::recvYieldDevice(), CudaComputeNonbondedKernel::reduceVirialEnergy(), ComputeNonbondedCUDA::requirePatch(), ComputePmeCUDAMgr::setupPencils(), ComputePmeMgr::ungridCalc(), and ComputeCUDAMgr::update().
int deviceIDList[MAX_NUM_RANKS] |
Definition at line 72 of file DeviceCUDA.C.
Referenced by DeviceCUDA::getDeviceIDforPe(), and DeviceCUDA::initialize().
int masterPeList[MAX_NUM_DEVICES] |
Definition at line 75 of file DeviceCUDA.C.
Referenced by DeviceCUDA::getMasterPeForDeviceID(), and DeviceCUDA::initialize().