有没有一种方法可以在Numba中固定阵列,以实现与设备之间的快速数据传输?

时间:2019-09-03 22:13:02

标签: numba numba-pro

在Pytorch中,有一个选项可以固定CPU阵列以快速传输到GPU(尽管似乎不适用于GPU-> CPU)。

我想知道是否有一种方法可以将Numba阵列固定到内存,或者有其他替代技术可以将CPU快速传输到GPU。我在文档中没有看到直接的选项,因此我想我们需要在数组创建期间指定分配。

0 个答案:

没有答案