From 20334d126bf8fb880f146ea6e4c0079b6322bcb6 Mon Sep 17 00:00:00 2001 From: Andy Jost Date: Wed, 19 Nov 2025 12:25:37 -0800 Subject: [PATCH] Expand a 32-bit int to 64-bits for proper CUDA API call. --- .../core/experimental/_memory/_device_memory_resource.pyx | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cuda_core/cuda/core/experimental/_memory/_device_memory_resource.pyx b/cuda_core/cuda/core/experimental/_memory/_device_memory_resource.pyx index 47b6fd114e6..a2a8eccb68b 100644 --- a/cuda_core/cuda/core/experimental/_memory/_device_memory_resource.pyx +++ b/cuda_core/cuda/core/experimental/_memory/_device_memory_resource.pyx @@ -5,7 +5,7 @@ from __future__ import annotations from libc.limits cimport ULLONG_MAX -from libc.stdint cimport uintptr_t +from libc.stdint cimport uintptr_t, uint64_t from libc.string cimport memset from cuda.bindings cimport cydriver @@ -116,10 +116,10 @@ cdef DMRA_mempool_attribute(property_type: type): return decorator -cdef int DMRA_getattribute( +cdef uint64_t DMRA_getattribute( cydriver.CUmemoryPool pool_handle, cydriver.CUmemPool_attribute attr_enum ): - cdef int value + cdef uint64_t value = 0 with nogil: HANDLE_RETURN(cydriver.cuMemPoolGetAttribute(pool_handle, attr_enum, &value)) return value