Преглед изворни кода

Only accept cudaErrorNotReady errors from cudaEventQuery

Samuel Thibault пре 14 година
родитељ
комит
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