diff options
| author | NAKAMURA Takumi <geek4civic@gmail.com> | 2025-01-09 17:50:40 +0900 |
|---|---|---|
| committer | NAKAMURA Takumi <geek4civic@gmail.com> | 2025-01-09 17:50:40 +0900 |
| commit | fea7da1b00cc97d742faede2df96c7d327950f49 (patch) | |
| tree | 4de1d6b4ddc69f4f32daabb11ad5c71ab0cf895e /flang/runtime/CUDA/descriptor.cpp | |
| parent | 9b99dde0d47102625d93c5d1cbbc04951025a6c9 (diff) | |
| parent | 0aa930a41f2d1ebf1fa90ec42da8f96d15a4dcbb (diff) | |
| download | llvm-users/chapuni/cov/single/nextcount.zip llvm-users/chapuni/cov/single/nextcount.tar.gz llvm-users/chapuni/cov/single/nextcount.tar.bz2 | |
Merge branch 'users/chapuni/cov/single/nextcount-base' into users/chapuni/cov/single/nextcountusers/chapuni/cov/single/nextcount
Diffstat (limited to 'flang/runtime/CUDA/descriptor.cpp')
| -rw-r--r-- | flang/runtime/CUDA/descriptor.cpp | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/flang/runtime/CUDA/descriptor.cpp b/flang/runtime/CUDA/descriptor.cpp index 816b145..947eeb6 100644 --- a/flang/runtime/CUDA/descriptor.cpp +++ b/flang/runtime/CUDA/descriptor.cpp @@ -20,8 +20,7 @@ RT_EXT_API_GROUP_BEGIN Descriptor *RTDEF(CUFAllocDescriptor)( std::size_t sizeInBytes, const char *sourceFile, int sourceLine) { - return reinterpret_cast<Descriptor *>( - CUFAllocManaged(sizeInBytes, kCudaNoStream)); + return reinterpret_cast<Descriptor *>(CUFAllocManaged(sizeInBytes)); } void RTDEF(CUFFreeDescriptor)( @@ -47,6 +46,13 @@ void RTDEF(CUFDescriptorSync)(Descriptor *dst, const Descriptor *src, (void *)dst, (const void *)src, count, cudaMemcpyHostToDevice)); } +void RTDEF(CUFSyncGlobalDescriptor)( + void *hostPtr, const char *sourceFile, int sourceLine) { + void *devAddr{RTNAME(CUFGetDeviceAddress)(hostPtr, sourceFile, sourceLine)}; + RTNAME(CUFDescriptorSync) + ((Descriptor *)devAddr, (Descriptor *)hostPtr, sourceFile, sourceLine); +} + RT_EXT_API_GROUP_END } } // namespace Fortran::runtime::cuda |