numba - 有没有办法在 Numba 中固定阵列，以便快速向/从设备传输数据？

Question

在 Pytorch 中，有一个选项可以固定 CPU 阵列以快速传输到 GPU（但似乎不适用于 GPU -> CPU）。

我想知道是否有办法将 Numba 数组固定到内存，或者有任何替代技术可以从 CPU 快速传输到 GPU。我没有从文档中看到直接的选项，所以我的猜测是我们需要在数组创建期间指定分配。