copy_wait

nvmath.device.copy_wait()

Creates synchronization point. It has to be called after nvmath.device.copy() to ensure that the copy operation has completed before any subsequent operations are executed.

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