Sfoglia il codice sorgente

max: The mux version can take local or CPU data at will

Samuel Thibault 4 anni fa
parent
commit
e5d472fe59
1 ha cambiato i file con 5 aggiunte e 4 eliminazioni
  1. 5 4
      tests/fpga/max_fpga_mux.c

+ 5 - 4
tests/fpga/max_fpga_mux.c

@@ -124,7 +124,7 @@ static struct starpu_codelet cl1 =
 	.nbuffers = 3,
 	.modes = {STARPU_R, STARPU_R, STARPU_W},
 	.specific_nodes = 1,
-	.nodes = {STARPU_SPECIFIC_NODE_CPU, STARPU_SPECIFIC_NODE_CPU, STARPU_SPECIFIC_NODE_LOCAL},
+	.nodes = {STARPU_SPECIFIC_NODE_LOCAL_OR_CPU, STARPU_SPECIFIC_NODE_LOCAL_OR_CPU, STARPU_SPECIFIC_NODE_LOCAL},
 };
 
 void fpga_impl2(void *buffers[], void *cl_arg)
@@ -172,8 +172,9 @@ static struct starpu_codelet cl2 =
 {
  	.fpga_funcs = {fpga_impl2},
 	.nbuffers = 3,
-	.modes = {STARPU_R, STARPU_R, STARPU_W}
-	/* local by default */
+	.modes = {STARPU_R, STARPU_R, STARPU_W},
+	.specific_nodes = 1,
+	.nodes = {STARPU_SPECIFIC_NODE_LOCAL_OR_CPU, STARPU_SPECIFIC_NODE_LOCAL_OR_CPU, STARPU_SPECIFIC_NODE_LOCAL},
 };
 
 void fpga_impl3(void *buffers[], void *cl_arg)
@@ -223,7 +224,7 @@ static struct starpu_codelet cl3 =
 	.nbuffers = 3,
 	.modes = {STARPU_R, STARPU_R, STARPU_W},
 	.specific_nodes = 1,
-	.nodes = {STARPU_SPECIFIC_NODE_LOCAL, STARPU_SPECIFIC_NODE_LOCAL, STARPU_SPECIFIC_NODE_CPU},
+	.nodes = {STARPU_SPECIFIC_NODE_LOCAL_OR_CPU, STARPU_SPECIFIC_NODE_LOCAL_OR_CPU, STARPU_SPECIFIC_NODE_CPU},
 };
 
 int main(int argc, char **argv)