commit 0c58388ced9d28704eb304ea94a5109a1d799bec
author Lei Zhang <antiagainst@google.com> Mon Jul 31 12:20:15 2023 -0400
committer GitHub <noreply@github.com> Mon Jul 31 09:20:15 2023 -0700
tree bd6f9c323c7640e7d694d880b23e47c3988ee332
parent b32e07574551dab08202922f05ef2c078d564da8

[cuda] Use proper symbols to fix compilation errors (#14523)

experimental/cuda2/cuda_allocator.c

diff --git a/experimental/cuda2/cuda_allocator.c b/experimental/cuda2/cuda_allocator.c
index d2752b5..e119529 100644
--- a/experimental/cuda2/cuda_allocator.c
+++ b/experimental/cuda2/cuda_allocator.c
@@ -615,7 +615,7 @@
     iree_hal_external_buffer_type_t requested_type,
     iree_hal_external_buffer_flags_t requested_flags,
     iree_hal_external_buffer_t* IREE_RESTRICT out_external_buffer) {
-  iree_hal_cuda_buffer_type_t buffer_type = iree_hal_cuda_buffer_type(buffer);
+  iree_hal_cuda2_buffer_type_t buffer_type = iree_hal_cuda2_buffer_type(buffer);
 
   switch (requested_type) {
     case IREE_HAL_EXTERNAL_BUFFER_TYPE_DEVICE_ALLOCATION:
@@ -624,7 +624,7 @@
           out_external_buffer->flags = requested_flags;
           out_external_buffer->type = requested_type;
           out_external_buffer->handle.device_allocation.ptr =
-              iree_hal_cuda_buffer_device_pointer(buffer);
+              iree_hal_cuda2_buffer_device_pointer(buffer);
           out_external_buffer->size = iree_hal_buffer_allocation_size(buffer);
           return iree_ok_status();