There was an error fetching the commit references. Please try again later.
feat: implement `__cuda_array_interface__`
https://numba.readthedocs.io/en/stable/cuda/cuda_array_interface.html 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__`
Showing
- pystencils/config.py 3 additions, 3 deletionspystencils/config.py
- pystencils/gpucuda/cudajit.py 43 additions, 14 deletionspystencils/gpucuda/cudajit.py
- pystencils_tests/test_cuda_array_interface.py 61 additions, 0 deletionspystencils_tests/test_cuda_array_interface.py
- pystencils_tests/test_cudagpu.py 36 additions, 5 deletionspystencils_tests/test_cudagpu.py
Please register or sign in to comment