aboutsummaryrefslogtreecommitdiff
path: root/flang-rt/lib/cuda/allocator.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'flang-rt/lib/cuda/allocator.cpp')
-rw-r--r--flang-rt/lib/cuda/allocator.cpp21
1 files changed, 9 insertions, 12 deletions
diff --git a/flang-rt/lib/cuda/allocator.cpp b/flang-rt/lib/cuda/allocator.cpp
index 05d97a5..5436051 100644
--- a/flang-rt/lib/cuda/allocator.cpp
+++ b/flang-rt/lib/cuda/allocator.cpp
@@ -138,23 +138,21 @@ void RTDEF(CUFRegisterAllocator)() {
void *CUFAllocPinned(
std::size_t sizeInBytes, [[maybe_unused]] std::int64_t *asyncObject) {
void *p;
- CUDA_REPORT_IF_ERROR(cudaMallocHost((void **)&p, sizeInBytes));
+ cudaMallocHost((void **)&p, sizeInBytes);
return p;
}
-void CUFFreePinned(void *p) { CUDA_REPORT_IF_ERROR(cudaFreeHost(p)); }
+void CUFFreePinned(void *p) { cudaFreeHost(p); }
void *CUFAllocDevice(std::size_t sizeInBytes, std::int64_t *asyncObject) {
void *p;
if (Fortran::runtime::executionEnvironment.cudaDeviceIsManaged) {
- CUDA_REPORT_IF_ERROR(
- cudaMallocManaged((void **)&p, sizeInBytes, cudaMemAttachGlobal));
+ cudaMallocManaged((void **)&p, sizeInBytes, cudaMemAttachGlobal);
} else {
if (asyncObject == nullptr) {
- CUDA_REPORT_IF_ERROR(cudaMalloc(&p, sizeInBytes));
+ cudaMalloc(&p, sizeInBytes);
} else {
- CUDA_REPORT_IF_ERROR(
- cudaMallocAsync(&p, sizeInBytes, (cudaStream_t)*asyncObject));
+ cudaMallocAsync(&p, sizeInBytes, (cudaStream_t)*asyncObject);
insertAllocation(p, sizeInBytes, (cudaStream_t)*asyncObject);
}
}
@@ -167,21 +165,20 @@ void CUFFreeDevice(void *p) {
if (pos >= 0) {
cudaStream_t stream = deviceAllocations[pos].stream;
eraseAllocation(pos);
- CUDA_REPORT_IF_ERROR(cudaFreeAsync(p, stream));
+ cudaFreeAsync(p, stream);
} else {
- CUDA_REPORT_IF_ERROR(cudaFree(p));
+ cudaFree(p);
}
}
void *CUFAllocManaged(
std::size_t sizeInBytes, [[maybe_unused]] std::int64_t *asyncObject) {
void *p;
- CUDA_REPORT_IF_ERROR(
- cudaMallocManaged((void **)&p, sizeInBytes, cudaMemAttachGlobal));
+ cudaMallocManaged((void **)&p, sizeInBytes, cudaMemAttachGlobal);
return reinterpret_cast<void *>(p);
}
-void CUFFreeManaged(void *p) { CUDA_REPORT_IF_ERROR(cudaFree(p)); }
+void CUFFreeManaged(void *p) { cudaFree(p); }
void *CUFAllocUnified(
std::size_t sizeInBytes, [[maybe_unused]] std::int64_t *asyncObject) {