Stan Seibert
Stan Seibert
Yes, all of these methods should accept Numba device allocations. What error did you see?
I assume DeviceNDArrays can only be C ordered at the moment? Otherwise, option 2 seems best.
However, option 1 might be a good interim solution, since Numba's release cycle might be too long to wait to fix this.
Note that we're encouraging people to migrate to [CuPy](https://cupy.chainer.org/), which provides wrappers around the same CUDA libraries that pyculib provided, but with a nicer NumPy interface.
I'm not sure what vqgan is, but also the pyculib project isn't getting updates anymore. (We strongly encourage folks to check out CuPy.)