Bläddra i källkod

tools/starpu_machine_display.c: also display SCC and MPI master slaves nodes

Nathalie Furmento 8 år sedan
förälder
incheckning
e2a6f6ca5a
1 ändrade filer med 18 tillägg och 0 borttagningar
  1. 18 0
      tools/starpu_machine_display.c

+ 18 - 0
tools/starpu_machine_display.c

@@ -166,6 +166,14 @@ int main(int argc, char **argv)
 	unsigned nmiccores = starpu_mic_worker_get_count();
 #endif
 
+#ifdef STARPU_USE_SCC
+	unsigned scc = starpu_scc_worker_get_count();
+#endif
+
+#ifdef STARPU_USE_MPI_MASTER_SLAVE
+	unsigned mpi_ms = starpu_mpi_ms_worker_get_count();
+#endif
+
 	fprintf(stdout, "StarPU has found :\n");
 
 	fprintf(stdout, "\t%u CPU threads\n", ncpu);
@@ -182,6 +190,16 @@ int main(int argc, char **argv)
 	display_worker_names(STARPU_MIC_WORKER);
 #endif
 
+#ifdef STARPU_USE_SCC
+	fprintf(stdout, "\t%u SCC cores\n", scc);
+	display_worker_names(STARPU_SCC_WORKER);
+#endif
+
+#ifdef STARPU_USE_MPI_MASTER_SLAVE
+	fprintf(stdout, "\t%u MPI Master workers\n", mpi_ms);
+	display_worker_names(STARPU_MPI_MS_WORKER);
+#endif
+
 	display_all_combined_workers();
 
 	if (ret != -ENODEV)