copy#

nvmath.device.copy(
src: OpaqueTensor,
dst: OpaqueTensor,
alignment=None,
)[source]#

Copies data from the source tensor to the destination tensor.

Parameters:

  • src (OpaqueTensor) – The source tensor to copy from.

  • dst (OpaqueTensor) – The destination tensor to copy to.

  • alignment (int, optional) – The alignment (in bytes) for the copy operation.

Refer to the cuBLASDx documentation for more details on how to use this function: https://docs.nvidia.com/cuda/cublasdx/api/other_tensors.html#cooperative-global-shared-copying

On this page