Skip to content

Draft: feat: implement `__cuda_array_interface__`

Stephan Seitz requested to merge seitz/pystencils:cuda-array-interface into master

https://numba.readthedocs.io/en/stable/cuda/cuda_array_interface.html

This is supported by:

Also allow to execute with cupy (https://docs.cupy.dev/en/stable/index.html) instead of pycuda

TODO:

  • check that pointers in correct CUDA context and if not import into current
  • make execution with pycuda aware of __cuda_array_interface__
  • what/how to test
Edited by Stephan Seitz

Merge request reports