[flang][cuda][NFC] Fix naming of CUFGetAssociatedStream (llvm#169838)

Author: clementval

flang-rt/lib/cuda/allocator.cpp

@@ -132,7 +132,7 @@ void RTDEF(CUFRegisterAllocator)() {
       kUnifiedAllocatorPos, {&CUFAllocUnified, CUFFreeUnified});
 }
 
-cudaStream_t RTDECL(CUFAssociatedGetStream)(void *p) {
+cudaStream_t RTDECL(CUFGetAssociatedStream)(void *p) {
   int pos = findAllocation(p);
   if (pos >= 0) {
     cudaStream_t stream = deviceAllocations[pos].stream;

flang-rt/unittests/Runtime/CUDA/Allocatable.cpp

@@ -142,7 +142,7 @@ TEST(AllocatableAsyncTest, StreamDeviceAllocatable) {
   EXPECT_TRUE(a->IsAllocated());
   cudaDeviceSynchronize();
   EXPECT_EQ(cudaSuccess, cudaGetLastError());
-  cudaStream_t s = RTDECL(CUFAssociatedGetStream)(a->raw().base_addr);
+  cudaStream_t s = RTDECL(CUFGetAssociatedStream)(a->raw().base_addr);
   EXPECT_EQ(s, stream);
   RTNAME(AllocatableDeallocate)
   (*a, /*hasStat=*/false, /*errMsg=*/nullptr, __FILE__, __LINE__);
@@ -156,7 +156,7 @@ TEST(AllocatableAsyncTest, StreamDeviceAllocatable) {
   EXPECT_TRUE(a->IsAllocated());
   cudaDeviceSynchronize();
   EXPECT_EQ(cudaSuccess, cudaGetLastError());
-  cudaStream_t d = RTDECL(CUFAssociatedGetStream)(a->raw().base_addr);
+  cudaStream_t d = RTDECL(CUFGetAssociatedStream)(a->raw().base_addr);
   EXPECT_EQ(d, defaultStream);
   RTNAME(AllocatableDeallocate)
   (*a, /*hasStat=*/false, /*errMsg=*/nullptr, __FILE__, __LINE__);
@@ -169,6 +169,6 @@ TEST(AllocatableAsyncTest, StreamDeviceAllocatable) {
   EXPECT_TRUE(a->IsAllocated());
   cudaDeviceSynchronize();
   EXPECT_EQ(cudaSuccess, cudaGetLastError());
-  cudaStream_t empty = RTDECL(CUFAssociatedGetStream)(a->raw().base_addr);
+  cudaStream_t empty = RTDECL(CUFGetAssociatedStream)(a->raw().base_addr);
   EXPECT_EQ(empty, nullptr);
 }

flang/include/flang/Runtime/CUDA/allocator.h

@@ -20,7 +20,7 @@ namespace Fortran::runtime::cuda {
 extern "C" {
 
 void RTDECL(CUFRegisterAllocator)();
-cudaStream_t RTDECL(CUFAssociatedGetStream)(void *);
+cudaStream_t RTDECL(CUFGetAssociatedStream)(void *);
 }
 
 void *CUFAllocPinned(std::size_t, std::int64_t *);