Skip to content

Commit 976ce48

Browse files
committed
Fix sync() for single stream mode.
1 parent d1b4705 commit 976ce48

1 file changed

Lines changed: 10 additions & 6 deletions

File tree

src/gpuarray_buffer_cuda.c

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1388,12 +1388,16 @@ static int cuda_sync(gpudata *b) {
13881388

13891389
ASSERT_BUF(b);
13901390
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-
if (ctx->err != CUDA_SUCCESS)
1396-
err = GA_IMPL_ERROR;
1391+
if (ctx->flags & GA_CTX_SINGLE_STREAM) {
1392+
cuStreamSynchronize(ctx->s);
1393+
} else {
1394+
ctx->err = cuEventSynchronize(b->wev);
1395+
if (ctx->err != CUDA_SUCCESS)
1396+
err = GA_IMPL_ERROR;
1397+
ctx->err = cuEventSynchronize(b->rev);
1398+
if (ctx->err != CUDA_SUCCESS)
1399+
err = GA_IMPL_ERROR;
1400+
}
13971401
cuda_exit(ctx);
13981402
return err;
13991403
}

0 commit comments

Comments
 (0)