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 Jan. 24, 2023, 1:05 a.m.