diff --git a/runtime/starpu/codelets/codelet_zgemm.c b/runtime/starpu/codelets/codelet_zgemm.c index 8e52cb94b6268db109386e6957c39a5ac70709a5..842724a6550a70ce96106b61b319c001c04d6bc4 100644 --- a/runtime/starpu/codelets/codelet_zgemm.c +++ b/runtime/starpu/codelets/codelet_zgemm.c @@ -200,13 +200,14 @@ static void cl_zgemm_cuda_func(void *descr[], void *cl_arg) cuDoubleComplex beta; cuDoubleComplex *C; int ldc; + CUstream stream; A = (cuDoubleComplex *)STARPU_MATRIX_GET_PTR(descr[0]); B = (cuDoubleComplex *)STARPU_MATRIX_GET_PTR(descr[1]); C = (cuDoubleComplex *)STARPU_MATRIX_GET_PTR(descr[2]); starpu_codelet_unpack_args(cl_arg, &transA, &transB, &m, &n, &k, &alpha, &lda, &ldb, &beta, &ldc); - CUstream stream = starpu_cuda_get_local_stream(); + stream = starpu_cuda_get_local_stream(); cublasSetKernelStream( stream ); cublasZgemm(