Explorar o código

Fix the case of GPU-GPU transfers

Cédric Augonnet %!s(int64=15) %!d(string=hai) anos
pai
achega
660fe66521
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      src/datawizard/coherency.c

+ 1 - 1
src/datawizard/coherency.c

@@ -119,7 +119,7 @@ static int worker_supports_direct_access(unsigned node, unsigned handling_node)
 #ifdef HAVE_CUDA_MEMCPY_PEER
 			/* GPUs not always allow direct remote access: if CUDA4
 			 * is enabled, we allow two CUDA devices to communicate. */
-			return (type_handling != STARPU_OPENCL_RAM);
+			return (_starpu_get_node_kind(handling_node) != STARPU_OPENCL_RAM);
 #else
 			/* Direct GPU-GPU transfers are not allowed in general */
 			return 0;