|
@@ -65,7 +65,7 @@ static TYPE **(tmp_21_block[2]);
|
|
|
|
|
|
int get_block_rank(unsigned i, unsigned j);
|
|
|
|
|
|
-static void parse_args(int argc, char **argv, int rank)
|
|
|
+static void parse_args(int argc, char **argv)
|
|
|
{
|
|
|
int i;
|
|
|
for (i = 1; i < argc; i++) {
|
|
@@ -399,6 +399,7 @@ int main(int argc, char **argv)
|
|
|
int world_size;
|
|
|
int thread_support;
|
|
|
|
|
|
+#if 0
|
|
|
/*
|
|
|
* Initialization
|
|
|
*/
|
|
@@ -413,19 +414,21 @@ int main(int argc, char **argv)
|
|
|
|
|
|
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
|
|
|
MPI_Comm_size(MPI_COMM_WORLD, &world_size);
|
|
|
+#endif
|
|
|
|
|
|
starpu_srand48((long int)time(NULL));
|
|
|
|
|
|
- parse_args(argc, argv, rank);
|
|
|
-
|
|
|
- STARPU_ASSERT(p*q == world_size);
|
|
|
+ parse_args(argc, argv);
|
|
|
|
|
|
starpu_init(NULL);
|
|
|
|
|
|
/* We disable sequential consistency in this example */
|
|
|
starpu_data_set_default_sequential_consistency_flag(0);
|
|
|
|
|
|
- starpu_mpi_initialize();
|
|
|
+ starpu_mpi_initialize_extended(1, &rank, &world_size);
|
|
|
+
|
|
|
+ STARPU_ASSERT(p*q == world_size);
|
|
|
+
|
|
|
starpu_helper_cublas_init();
|
|
|
|
|
|
int barrier_ret = MPI_Barrier(MPI_COMM_WORLD);
|