pycuda
pycuda copied to clipboard
Support tensor memory
Discussed in https://github.com/inducer/pycuda/discussions/443
https://docs.nvidia.com/cuda/cuda-driver-api/group__CUDA__TENSOR__MEMORY.html
cc @bastianhagedorn