Refactored Extras.cu / Extras.h for gpu build
This commit is contained in:
@@ -1,19 +1,19 @@
|
||||
// Basic cuda functions callable from C/C++ code
|
||||
#include <cuda.h>
|
||||
|
||||
extern "C" void dvc_AllocateDeviceMemory(void** address, size_t size){
|
||||
extern "C" void AllocateDeviceMemory(void** address, size_t size){
|
||||
cudaMalloc(address,size);
|
||||
}
|
||||
|
||||
extern "C" void dvc_CopyToDevice(void* dest, void* source, size_t size){
|
||||
extern "C" void CopyToDevice(void* dest, void* source, size_t size){
|
||||
cudaMemcpy(dest,source,size,cudaMemcpyHostToDevice);
|
||||
}
|
||||
|
||||
|
||||
extern "C" void dvc_CopyToHost(void* dest, void* source, size_t size){
|
||||
extern "C" void CopyToHost(void* dest, void* source, size_t size){
|
||||
cudaMemcpy(dest,source,size,cudaMemcpyDeviceToHost);
|
||||
}
|
||||
|
||||
extern "C" void dvc_Barrier(){
|
||||
extern "C" void DeviceBarrier(){
|
||||
cudaDeviceSynchronize();
|
||||
}
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
extern "C" void dvc_AllocateDeviceMemory(void** address, size_t size);
|
||||
extern "C" void AllocateDeviceMemory(void** address, size_t size);
|
||||
|
||||
extern "C" void dvc_CopyToDevice(void* dest, void* source, size_t size);
|
||||
extern "C" void CopyToDevice(void* dest, void* source, size_t size);
|
||||
|
||||
extern "C" void dvc_CopyToHost(void* dest, void* source, size_t size);
|
||||
extern "C" void CopyToHost(void* dest, void* source, size_t size);
|
||||
|
||||
extern "C" void dvc_Barrier();
|
||||
extern "C" void DeviceBarrier();
|
||||
|
||||
Reference in New Issue
Block a user