Update wrappers for CUDA 12.2.

Author: maleadt

lib/cublas/libcublas.jl

@@ -13,7 +13,8 @@ mutable struct CuMemoryPool
 
     function CuMemoryPool(dev::CuDevice;
                           alloc_type::CUmemAllocationType=ALLOCATION_TYPE_PINNED,
-                          handle_type::CUmemAllocationHandleType=HANDLE_TYPE_NONE)
+                          handle_type::CUmemAllocationHandleType=HANDLE_TYPE_NONE,
+                          maxSize::Integer=0)
         props = Ref(CUmemPoolProps(
             alloc_type,
             handle_type,
@@ -22,7 +23,8 @@ mutable struct CuMemoryPool
                 deviceid(dev)
             ),
             C_NULL,
-            ntuple(i->Cuchar(0), 64)
+            maxSize,
+            ntuple(i->Cuchar(0), 56)
         ))
         handle_ref = Ref{CUmemoryPool}()
         cuMemPoolCreate(handle_ref, props)

lib/cudadrv/libcuda.jl

@@ -14,7 +14,7 @@ end
 
 function check(f, errs...)
     res = retry_reclaim(in((CUDNN_STATUS_ALLOC_FAILED, errs...))) do
-        f()
+        return f()
     end
 
     if res != CUDNN_STATUS_SUCCESS

lib/cudadrv/pool.jl

@@ -14,7 +14,7 @@ end
 
 function check(f, errs...)
     res = retry_reclaim(in((CUFFT_ALLOC_FAILED, errs...))) do
-        f()
+        return f()
     end
 
     if res != CUFFT_SUCCESS

lib/cudnn/src/libcudnn.jl

@@ -6,8 +6,9 @@ using ..CUDA_Runtime
 
 using ..CUDA
 using ..CUDA: retry_reclaim, initialize_context
-using ..CUDA: CUuuid, CUcontext, CUstream, CUdevice, CUdevice_attribute,
-              CUgraph, CUgraphNode, CUgraphNodeType, CUgraphExec, CUaccessPolicyWindow
+using ..CUDA: CUuuid, CUcontext, CUstream, CUdevice, CUdevice_attribute, CUstreamAttrID,
+              CUgraph, CUgraphNode, CUgraphNodeType, CUgraphExec, CUaccessPolicyWindow,
+              CUstreamAttrValue
 
 using CEnum: @cenum