Просмотр исходного кода

Only accept cudaErrorNotReady errors from cudaEventQuery

Samuel Thibault лет назад: 14
Родитель
Сommit
0bf975225f
1 измененных файлов с 4 добавлено и 1 удалено
  1. 4 1
      src/datawizard/copy_driver.c

+ 4 - 1
src/datawizard/copy_driver.c

@@ -340,10 +340,13 @@ unsigned _starpu_driver_test_request_completion(struct starpu_async_channel *asy
 #ifdef STARPU_USE_CUDA
 		case STARPU_CUDA_RAM:
 			event = (*async_channel).event.cuda_event;
+			CUresult cures = cudaEventQuery(event);
 
-			success = (cudaEventQuery(event) == cudaSuccess);
+			success = (cures == cudaSuccess);
 			if (success)
 				cudaEventDestroy(event);
+			else if (cures != cudaErrorNotReady)
+				STARPU_CUDA_REPORT_ERROR(cures);
 
 			break;
 #endif