cupyx.jit.cg.memcpy_async#

cupyx.jit.cg.memcpy_async(group, dst, dst_idx, src, src_idx, size, *, aligned_size=None) = <cupyx.jit function>#

调用 cg::memcpy_sync().

参数:

  • group – 一个有效的协作组

  • dst – 可以视为一维 C 连续数组的目标数组

  • dst_idx – 目标数组元素的起始索引

  • src – 可以视为一维 C 连续数组的源数组

  • src_idx – 源数组元素的起始索引

  • size (int) – 从 src[src_index] 复制到 dst[dst_idx] 的字节数

  • aligned_size (int) – 使用 cuda::aligned_size_t<N> 向编译器保证 src/dst 至少按 N 字节对齐。如果无法保证对齐，则行为未定义。

另请参阅

cg::memcpy_sync