Searched refs:PetscGetCurrentCUDAStream (Results 1 – 4 of 4) sorted by relevance
13 PetscCall(PetscGetCurrentCUDAStream(&stream)); in MatZeroRows_CUDA()
39 PetscErrorCode PetscGetCurrentCUDAStream(cudaStream_t *stream) in PetscGetCurrentCUDAStream() function
166 PETSC_EXTERN PetscErrorCode PetscGetCurrentCUDAStream(cudaStream_t *);
29 - Add utility functions `PetscGetCurrentCUDAStream()` and `PetscGetCurrentHIPStream()` to return th…