spdl.io.CUDAArrayInterface¶
- class CUDAArrayInterface[source]¶
Protocol for objects that implement the CUDA Array Interface Protocol.
Objects implementing this protocol expose their CUDA data through the
__cuda_array_interface__
attribute, allowing zero-copy conversion to CUDA-aware libraries like PyTorch and Numba.See: https://numba.pydata.org/numba-doc/latest/cuda/cuda_array_interface.html