Forráskód Böngészése

tests/datawizard/acquire_release.c: add synchronisation and check token value

Nathalie Furmento 14 éve
szülő
commit
2e0bac56e6
1 módosított fájl, 6 hozzáadás és 1 törlés
  1. 6 1
      tests/datawizard/acquire_release.c

+ 6 - 1
tests/datawizard/acquire_release.c

@@ -30,7 +30,7 @@ static starpu_codelet increment_cl = {
 	.nbuffers = 1
 	.nbuffers = 1
 };
 };
 
 
-unsigned token = 42;
+unsigned token = 0;
 starpu_data_handle token_handle;
 starpu_data_handle token_handle;
 
 
 void increment_token()
 void increment_token()
@@ -54,6 +54,8 @@ int main(int argc, char **argv)
         starpu_init(NULL);
         starpu_init(NULL);
 	starpu_variable_data_register(&token_handle, 0, (uintptr_t)&token, sizeof(unsigned));
 	starpu_variable_data_register(&token_handle, 0, (uintptr_t)&token, sizeof(unsigned));
 
 
+        fprintf(stderr, "Token: %d\n", token);
+
 	for(i=0; i<ntasks; i++)
 	for(i=0; i<ntasks; i++)
 	{
 	{
                 starpu_data_acquire(token_handle, STARPU_RW);
                 starpu_data_acquire(token_handle, STARPU_RW);
@@ -63,7 +65,10 @@ int main(int argc, char **argv)
                 starpu_data_acquire_cb(token_handle, STARPU_RW, callback, NULL);
                 starpu_data_acquire_cb(token_handle, STARPU_RW, callback, NULL);
 	}
 	}
 
 
+        starpu_task_wait_for_all();
 	starpu_shutdown();
 	starpu_shutdown();
+        fprintf(stderr, "Token: %d\n", token);
+        STARPU_ASSERT(token==ntasks);
 
 
 	return 0;
 	return 0;
 }
 }