diff options
Diffstat (limited to 'flang/runtime/CUDA/descriptor.cpp')
| -rw-r--r-- | flang/runtime/CUDA/descriptor.cpp | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/flang/runtime/CUDA/descriptor.cpp b/flang/runtime/CUDA/descriptor.cpp new file mode 100644 index 000000000000..1031b1e601b6 --- /dev/null +++ b/flang/runtime/CUDA/descriptor.cpp @@ -0,0 +1,28 @@ +//===-- runtime/CUDA/descriptor.cpp ---------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#include "flang/Runtime/CUDA/descriptor.h" +#include "flang/Runtime/CUDA/allocator.h" + +namespace Fortran::runtime::cuda { +extern "C" { +RT_EXT_API_GROUP_BEGIN + +Descriptor *RTDEF(CUFAllocDesciptor)( + std::size_t sizeInBytes, const char *sourceFile, int sourceLine) { + return reinterpret_cast<Descriptor *>(CUFAllocManaged(sizeInBytes)); +} + +void RTDEF(CUFFreeDesciptor)( + Descriptor *desc, const char *sourceFile, int sourceLine) { + CUFFreeManaged(reinterpret_cast<void *>(desc)); +} + +RT_EXT_API_GROUP_END +} +} // namespace Fortran::runtime::cuda |
