let as_cu supports CUModule

Author: lijinf2

cuda_core/cuda/core/_cpp/resource_handles.hpp

@@ -754,6 +754,10 @@ inline CUlibrary as_cu(const LibraryHandle& h) noexcept {
     return h ? *h : nullptr;
 }
 
+inline CUmodule as_cu(const CUmodule& h) noexcept {
+    return h;
+}
+
 inline CUkernel as_cu(const KernelHandle& h) noexcept {
     return h ? *h : nullptr;
 }
@@ -838,10 +842,8 @@ inline std::intptr_t as_intptr(const LibraryHandle& h) noexcept {
     return reinterpret_cast<std::intptr_t>(as_cu(h));
 }
 
-// Regular pointer (CUmodule), not a shared_ptr: lifetime is owned by the
-// CUlibrary it was retrieved from.
 inline std::intptr_t as_intptr(const CUmodule& h) noexcept {
-    return reinterpret_cast<std::intptr_t>(h);
+    return reinterpret_cast<std::intptr_t>(as_cu(h));
 }
 
 inline std::intptr_t as_intptr(const KernelHandle& h) noexcept {

cuda_core/cuda/core/_resource_handles.pxd

@@ -85,6 +85,7 @@ cdef extern from "_cpp/resource_handles.hpp" namespace "cuda_core":
     cydriver.CUmemoryPool as_cu(MemoryPoolHandle h) noexcept nogil
     cydriver.CUdeviceptr as_cu(DevicePtrHandle h) noexcept nogil
     cydriver.CUlibrary as_cu(LibraryHandle h) noexcept nogil
+    cydriver.CUmodule as_cu(cydriver.CUmodule h) noexcept nogil
     cydriver.CUkernel as_cu(KernelHandle h) noexcept nogil
     cydriver.CUgraph as_cu(GraphHandle h) noexcept nogil
     cydriver.CUgraphExec as_cu(GraphExecHandle h) noexcept nogil