We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d1b4705 commit 976ce48Copy full SHA for 976ce48
1 file changed
src/gpuarray_buffer_cuda.c
@@ -1388,12 +1388,16 @@ static int cuda_sync(gpudata *b) {
1388
1389
ASSERT_BUF(b);
1390
cuda_enter(ctx);
1391
- ctx->err = cuEventSynchronize(b->wev);
1392
- if (ctx->err != CUDA_SUCCESS)
1393
- err = GA_IMPL_ERROR;
1394
- ctx->err = cuEventSynchronize(b->rev);
1395
1396
+ if (ctx->flags & GA_CTX_SINGLE_STREAM) {
+ cuStreamSynchronize(ctx->s);
+ } else {
+ ctx->err = cuEventSynchronize(b->wev);
+ if (ctx->err != CUDA_SUCCESS)
+ err = GA_IMPL_ERROR;
1397
+ ctx->err = cuEventSynchronize(b->rev);
1398
1399
1400
+ }
1401
cuda_exit(ctx);
1402
return err;
1403
}
0 commit comments