代码同步,检查bug

pull/13/head
chenzenghui 2025-03-22 20:04:33 +08:00
parent d1806f7427
commit 7a374c135b
2 changed files with 2 additions and 2 deletions

View File

@ -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;

View File

@ -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 -> 设备