代码同步,检查bug
parent
d1806f7427
commit
7a374c135b
|
@ -354,7 +354,7 @@ extern "C" void FreeCUDAHost(void* ptr) {
|
|||
}
|
||||
|
||||
// 多GPU内存分配函数
|
||||
void* mallocCUDADevice(size_t memsize, int device_id = 0)
|
||||
void* mallocCUDADevice(size_t memsize, int device_id )
|
||||
{
|
||||
void* ptr = nullptr;
|
||||
cudaError_t err;
|
||||
|
|
|
@ -77,7 +77,7 @@ extern "C" GPUBASELIBAPI void checkCudaError(cudaError_t err, const char* msg);
|
|||
// GPU 内存函数
|
||||
extern "C" GPUBASELIBAPI void* mallocCUDAHost(size_t memsize); // 主机内存声明
|
||||
extern "C" GPUBASELIBAPI void FreeCUDAHost(void* ptr);
|
||||
extern "C" GPUBASELIBAPI void* mallocCUDADevice(size_t memsize); // GPUÄÚ´æÉùÃ÷
|
||||
extern "C" GPUBASELIBAPI void* mallocCUDADevice(size_t memsize, int device_id=0); // GPUÄÚ´æÉùÃ÷
|
||||
extern "C" GPUBASELIBAPI void FreeCUDADevice(void* ptr);
|
||||
extern "C" GPUBASELIBAPI void HostToDevice(void* hostptr, void* deviceptr, size_t memsize);//GPU 内存数据转移 设备 -> GPU
|
||||
extern "C" GPUBASELIBAPI void DeviceToHost(void* hostptr, void* deviceptr, size_t memsize);//GPU 内存数据转移 GPU -> 设备
|
||||
|
|
Loading…
Reference in New Issue