2014-03-17 10:14:46 -04:00
|
|
|
// Basic cuda functions callable from C/C++ code
|
|
|
|
|
#include <cuda.h>
|
|
|
|
|
|
2014-03-19 09:07:50 -04:00
|
|
|
extern "C" void AllocateDeviceMemory(void** address, size_t size){
|
2014-03-17 10:14:46 -04:00
|
|
|
cudaMalloc(address,size);
|
|
|
|
|
}
|
|
|
|
|
|
2014-03-19 09:07:50 -04:00
|
|
|
extern "C" void CopyToDevice(void* dest, void* source, size_t size){
|
2014-03-17 10:14:46 -04:00
|
|
|
cudaMemcpy(dest,source,size,cudaMemcpyHostToDevice);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2014-03-19 09:07:50 -04:00
|
|
|
extern "C" void CopyToHost(void* dest, void* source, size_t size){
|
2014-03-17 10:14:46 -04:00
|
|
|
cudaMemcpy(dest,source,size,cudaMemcpyDeviceToHost);
|
|
|
|
|
}
|
|
|
|
|
|
2014-03-19 09:07:50 -04:00
|
|
|
extern "C" void DeviceBarrier(){
|
2014-03-17 10:14:46 -04:00
|
|
|
cudaDeviceSynchronize();
|
|
|
|
|
}
|