Browse Source

Example/filters: use starpu_data_get_nb_children() to get the number of children instead of the value initially given to starpu_data_partition()

Nathalie Furmento 15 years ago
parent
commit
e69c1c8e6c
2 changed files with 3 additions and 2 deletions
  1. 1 1
      examples/filters/fmatrix.c
  2. 2 1
      examples/filters/fvector.c

+ 1 - 1
examples/filters/fmatrix.c

@@ -75,7 +75,7 @@ int main(int argc, char **argv)
 	starpu_data_partition(handle, &f);
 	starpu_data_partition(handle, &f);
 
 
         /* Submit a task on each sub-vector */
         /* Submit a task on each sub-vector */
-	for (i=0; i<PARTS; i++)
+	for (i=0; i<starpu_data_get_nb_children(handle); i++)
 	{
 	{
                 struct starpu_task *task = starpu_task_create();
                 struct starpu_task *task = starpu_task_create();
                 int factor = i;
                 int factor = i;

+ 2 - 1
examples/filters/fvector.c

@@ -67,7 +67,7 @@ int main(int argc, char **argv)
 	starpu_data_partition(handle, &f);
 	starpu_data_partition(handle, &f);
 
 
         /* Submit a task on each sub-vector */
         /* Submit a task on each sub-vector */
-	for (i=0; i<PARTS; i++)
+	for (i=0; i<starpu_data_get_nb_children(handle); i++)
 	{
 	{
                 starpu_data_handle sub_handle = starpu_data_get_sub_data(handle, 1, i);
                 starpu_data_handle sub_handle = starpu_data_get_sub_data(handle, 1, i);
                 struct starpu_task *task = starpu_task_create();
                 struct starpu_task *task = starpu_task_create();
@@ -83,6 +83,7 @@ int main(int argc, char **argv)
 		starpu_task_submit(task);
 		starpu_task_submit(task);
 	}
 	}
 
 
+	starpu_data_unpartition(handle, 0);
         starpu_data_unregister(handle);
         starpu_data_unregister(handle);
 	starpu_shutdown();
 	starpu_shutdown();