test/linalg.h

   1 #define MAIN(x) linalg##x
   2 #include "test.h"
   3 #include "a/linalg.h"
   4
   5 static void test_mul(void)
   6 {
   7     a_uint i;
   8     a_float z[3 * 3];
   9     a_float x[3 * 3] = {1, 2, 3, 4, 5, 6, 7, 8, 9};
  10     a_float y[3 * 3] = {9, 8, 7, 6, 5, 4, 3, 2, 1};
  11
  12     debug("X\t={");
  13     for (i = 0; i < 9; ++i)
  14     {
  15         debug("%" A_FLOAT_PRI "g%c", x[i], i < 8 ? ',' : '}');
  16     }
  17     debug("\n");
  18
  19     debug("Y\t={");
  20     for (i = 0; i < 9; ++i)
  21     {
  22         debug("%" A_FLOAT_PRI "g%c", y[i], i < 8 ? ',' : '}');
  23     }
  24     debug("\n");
  25
  26     a_linalg_mulmm(z, x, y, 3, 3, 3);
  27
  28     debug("XY\t={");
  29     for (i = 0; i < 9; ++i)
  30     {
  31         debug("%" A_FLOAT_PRI "g%c", z[i], i < 8 ? ',' : '}');
  32     }
  33     debug("\n");
  34
  35     a_linalg_mulTm(z, x, y, 3, 3, 3);
  36
  37     debug("X'Y\t={");
  38     for (i = 0; i < 9; ++i)
  39     {
  40         debug("%" A_FLOAT_PRI "g%c", z[i], i < 8 ? ',' : '}');
  41     }
  42     debug("\n");
  43
  44     a_linalg_mulmT(z, x, y, 3, 3, 3);
  45
  46     debug("XY'\t={");
  47     for (i = 0; i < 9; ++i)
  48     {
  49         debug("%" A_FLOAT_PRI "g%c", z[i], i < 8 ? ',' : '}');
  50     }
  51     debug("\n");
  52
  53     a_linalg_mulTT(z, x, y, 3, 3, 3);
  54
  55     debug("X'Y'\t={");
  56     for (i = 0; i < 9; ++i)
  57     {
  58         debug("%" A_FLOAT_PRI "g%c", z[i], i < 8 ? ',' : '}');
  59     }
  60     debug("\n");
  61
  62     debug("X.Y\t=%" A_FLOAT_PRI "g\n", a_linalg_dot(x, y, 9));
  63 }
  64
  65 int main(int argc, char *argv[]) /* NOLINT(misc-definitions-in-headers) */
  66 {
  67     main_init(argc, argv, 1);
  68     test_mul();
  69     return 0;
  70 }