是否可以在PyCuda中固定阵列,以实现与设备之间的快速数据传输?

时间:2019-09-03 17:59:46

标签: pycuda

在Pytorch中,有一个选项可以固定CPU阵列 https://pytorch.org/docs/stable/tensors.html#torch.Tensor.pin_memory

我想知道PyCuda中是否有这样的技术。我在文档中没有看到直接的选项,因此我的猜测是我们需要在数组创建期间指定分配。

0 个答案:

没有答案