cuda_synchronize: Waits for all kernels in all streams on a CUDA device to...

View source: R/cuda.R

cuda_synchronizeR Documentation

Waits for all kernels in all streams on a CUDA device to complete.

Description

Waits for all kernels in all streams on a CUDA device to complete.

Usage

cuda_synchronize(device = NULL)

Arguments

device

device for which to synchronize. It uses the current device given by cuda_current_device() if no device is specified.


torch documentation built on May 29, 2024, 9:54 a.m.