From 485170d22dce7254ecc842570676056a0477011f Mon Sep 17 00:00:00 2001
From: Keith Kraus <keith.j.kraus@gmail.com>
Date: Fri, 7 Nov 2025 16:18:17 -0500
Subject: [PATCH] Add __cuda_stream__ protocol to driver.CUStream class

---
 cuda_bindings/cuda/bindings/driver.pyx.in | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/cuda_bindings/cuda/bindings/driver.pyx.in b/cuda_bindings/cuda/bindings/driver.pyx.in
index 2921fe181..33c1c770f 100644
--- a/cuda_bindings/cuda/bindings/driver.pyx.in
+++ b/cuda_bindings/cuda/bindings/driver.pyx.in
@@ -7101,8 +7101,11 @@ cdef class CUstream:
         return <void_ptr>self._pvt_ptr[0]
     def getPtr(self):
         return <void_ptr>self._pvt_ptr
+    def __cuda_stream__(self):
+        return (0, <uintptr_t><void_ptr>(self._pvt_ptr[0]))
 {{endif}}
 
+
 {{if 'CUgraphicsResource' in found_types}}
 
 cdef class CUgraphicsResource: