torch::cuda Namespace Reference

Functions
void	synchronize (int64_t device_index)

Function Documentation

synchronize()

void torch::cuda::synchronize

(

int64_t

device_index

)

Referenced by NEML2ModelExecutor::deviceSynchronize().