Skip to content

Commit 907c433

Browse files
authored
Merge pull request #2007 from Nafees01/cuda-graph-followup
cuda-ref: use cudaMemcpyAsync for QFunction context
2 parents a604ee3 + ec25369 commit 907c433

1 file changed

Lines changed: 1 addition & 9 deletions

File tree

backends/cuda-ref/ceed-cuda-ref-qfunctioncontext.c

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -37,15 +37,7 @@ static inline int CeedQFunctionContextSyncH2D_Cuda(const CeedQFunctionContext ct
3737
impl->d_data = impl->d_data_owned;
3838
}
3939

40-
// Use async memcpy during CUDA Graph capture for compatibility
41-
enum cudaStreamCaptureStatus capture_status;
42-
43-
cudaStreamIsCapturing(cudaStreamPerThread, &capture_status);
44-
if (capture_status != cudaStreamCaptureStatusNone) {
45-
CeedCallCuda(ceed, cudaMemcpyAsync(impl->d_data, impl->h_data, ctx_size, cudaMemcpyHostToDevice, cudaStreamPerThread));
46-
} else {
47-
CeedCallCuda(ceed, cudaMemcpy(impl->d_data, impl->h_data, ctx_size, cudaMemcpyHostToDevice));
48-
}
40+
CeedCallCuda(ceed, cudaMemcpyAsync(impl->d_data, impl->h_data, ctx_size, cudaMemcpyHostToDevice, cudaStreamPerThread));
4941

5042
CeedCallBackend(CeedDestroy(&ceed));
5143
return CEED_ERROR_SUCCESS;

0 commit comments

Comments
 (0)