Bläddra i källkod

Always make sure that kernels without the async flag are waiting for kernel termination

Samuel Thibault 10 år sedan
förälder
incheckning
2e3c01cb38
1 ändrade filer med 2 tillägg och 2 borttagningar
  1. 2 2
      src/drivers/cuda/driver_cuda.c

+ 2 - 2
src/drivers/cuda/driver_cuda.c

@@ -571,8 +571,8 @@ static void execute_job_on_cuda(struct starpu_task *task, struct _starpu_worker
 	else
 	/* Synchronous execution */
 	{
-#if defined(STARPU_DEBUG) && !defined(STARPU_SIMGRID)
-		STARPU_ASSERT_MSG(cudaStreamQuery(starpu_cuda_get_local_stream()) == cudaSuccess, "CUDA codelets have to wait for termination of their kernels on the starpu_cuda_get_local_stream() stream");
+#if !defined(STARPU_SIMGRID)
+		STARPU_ASSERT_MSG(cudaStreamQuery(starpu_cuda_get_local_stream()) == cudaSuccess, "Unless when using the STARPU_CUDA_ASYNC flag, CUDA codelets have to wait for termination of their kernels on the starpu_cuda_get_local_stream() stream");
 #endif
 		finish_job_on_cuda(j, worker);
 	}