Browse Source

gcc: Add tests for `register' with non-vector matrices.

* gcc-plugin/tests/register.c (main): Add two tests.
Ludovic Courtès 13 years ago
parent
commit
60438a6f6d
1 changed files with 12 additions and 1 deletions
  1. 12 1
      gcc-plugin/tests/register.c

+ 12 - 1
gcc-plugin/tests/register.c

@@ -49,6 +49,7 @@ main (int argc, char *argv[])
   double *y;
   double *y;
   static char z[345];
   static char z[345];
   static float m[7][42];
   static float m[7][42];
+  static float m3d[14][11][80];
   short w[] = { 1, 2, 3 };
   short w[] = { 1, 2, 3 };
   size_t y_size = 234;
   size_t y_size = 234;
 
 
@@ -123,7 +124,17 @@ main (int argc, char *argv[])
   expected_register_arguments.element_size = sizeof m[0][0];
   expected_register_arguments.element_size = sizeof m[0][0];
 #pragma starpu register m[6]
 #pragma starpu register m[6]
 
 
-  assert (data_register_calls == 14);
+  expected_register_arguments.pointer = m;
+  expected_register_arguments.elements = 7;
+  expected_register_arguments.element_size = sizeof m[0];
+#pragma starpu register m
+
+  expected_register_arguments.pointer = m3d;
+  expected_register_arguments.elements = 14;
+  expected_register_arguments.element_size = sizeof m3d[0];
+#pragma starpu register m3d
+
+  assert (data_register_calls == 16);
 
 
   free (y);
   free (y);