Skip to content
Snippets Groups Projects
Unverified Commit 4650565d authored by Recolic Keghart's avatar Recolic Keghart
Browse files

sync

parent 7e33fd4d
No related branches found
No related tags found
No related merge requests found
...@@ -125,11 +125,11 @@ namespace rlib { ...@@ -125,11 +125,11 @@ namespace rlib {
__device__ bool dev_apply_new_cap() { __device__ bool dev_apply_new_cap() {
void *new_mem = nullptr; void *new_mem = nullptr;
(cudaMalloc(&new_mem, cap)); rlib::dev_cuda_assert(cudaMalloc(&new_mem, cap));
if(new_mem == nullptr) if(new_mem == nullptr)
return false; return false;
if(mem != nullptr) { if(mem != nullptr) {
(memcpy(new_mem, mem, m_size * sizeof(T))); rlib::dev_cuda_assert(memcpy(new_mem, mem, m_size * sizeof(T)));
// hope it success....... // hope it success.......
//(cudaFree(mem)); // let it leak!!! It doesn't matter now. //(cudaFree(mem)); // let it leak!!! It doesn't matter now.
} }
......
...@@ -6,6 +6,13 @@ namespace rlib { ...@@ -6,6 +6,13 @@ namespace rlib {
if(err != cudaError::cudaSuccess) if(err != cudaError::cudaSuccess)
throw std::runtime_error("CUDA runtime error: err code is " + std::to_string(err) + ", please refer to https://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART__TYPES.html#group__CUDART__TYPES_1g3f51e3575c2178246db0a94a430e0038"); throw std::runtime_error("CUDA runtime error: err code is " + std::to_string(err) + ", please refer to https://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART__TYPES.html#group__CUDART__TYPES_1g3f51e3575c2178246db0a94a430e0038");
} }
__device__ bool dev_cuda_assert(cudaError_t err) {
if(err != cudaError::cudaSuccess) {
printf("FUCK! CUDA runtime error: %d", err);
return false;
}
return true;
}
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment