cuda-python: add deprecation warning for raw stream pointers

flatironinstitute · Feb 5, 2024 · dfc434f · dfc434f
1 parent 410e9a6
commit dfc434f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/cufinufft/cufinufft/_compat.py b/python/cufinufft/cufinufft/_compat.py
@@ -1,7 +1,7 @@
 import inspect
 
 import numpy as np
-
+import warnings
 
 def get_array_ptr(data):
     try:
@@ -30,6 +30,10 @@ def get_array_module(obj):
 def get_stream_ptr(obj):
     framework = get_array_module(obj)
     if isinstance(obj, int):
+        warnings.warn("Raw stream pointers are deprecated and will be removed in version 2.4. "
+                      "Please use your python GPU framework's stream object type directly.",
+                      DeprecationWarning)
+
         return obj
     if framework == 'numba':
         return obj.handle.value