瀏覽代碼

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

Nathalie Furmento 14 年之前
父節點
當前提交
2e0bac56e6
共有 1 個文件被更改,包括 6 次插入1 次删除
  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
 };
 
-unsigned token = 42;
+unsigned token = 0;
 starpu_data_handle token_handle;
 
 void increment_token()
@@ -54,6 +54,8 @@ int main(int argc, char **argv)
         starpu_init(NULL);
 	starpu_variable_data_register(&token_handle, 0, (uintptr_t)&token, sizeof(unsigned));
 
+        fprintf(stderr, "Token: %d\n", token);
+
 	for(i=0; i<ntasks; i++)
 	{
                 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_task_wait_for_all();
 	starpu_shutdown();
+        fprintf(stderr, "Token: %d\n", token);
+        STARPU_ASSERT(token==ntasks);
 
 	return 0;
 }