Преглед на файлове

Make sure that we are not in exclusive thread mode when using memcpy peer

Samuel Thibault преди 13 години
родител
ревизия
942609f3bc
променени са 1 файла, в които са добавени 6 реда и са изтрити 0 реда
  1. 6 0
      src/drivers/cuda/driver_cuda.c

+ 6 - 0
src/drivers/cuda/driver_cuda.c

@@ -124,6 +124,12 @@ static void init_context(int devid)
 	cures = cudaGetDeviceProperties(&props[devid], devid);
 	if (STARPU_UNLIKELY(cures))
 		STARPU_CUDA_REPORT_ERROR(cures);
+#ifdef HAVE_CUDA_MEMCPY_PEER
+	if (props[devid].computeMode == cudaComputeModeExclusive) {
+		fprintf(stderr, "CUDA is in EXCLUSIVE-THREAD mode, but StarPU was built with multithread GPU control support, please reconfigure with --disable-cuda-memcpy-peer\n");
+		STARPU_ASSERT(0);
+	}
+#endif
 
 	limit_gpu_mem_if_needed(devid);