init: HW1

part2/main.c

@@ -0,0 +1,91 @@
+#include <getopt.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include "fasttime.h"
+#include "test.h"
+
+void usage(const char *progname);
+void initValue(float *values1, float *values2, double *value3, float *output, unsigned int N);
+
+extern void test1(float *a, float *b, float *c, int N);
+extern void test2(float *__restrict a, float *__restrict b, float *__restrict c, int N);
+extern double test3(double *__restrict a, int N) ;
+
+int main(int argc, char **argv) {
+  int N = 1024;
+  int whichTestToRun = 1;
+
+  // parse commandline options
+  int opt;
+  static struct option long_options[] = {
+    {"size", 1, 0, 's'},
+    {"test", 1, 0, 't'},
+    {"help", 0, 0, '?'},
+    {0 ,0, 0, 0}
+  };
+
+  while ((opt = getopt_long(argc, argv, "st:?", long_options, NULL)) != EOF) {
+
+    switch (opt) {
+      case 's':
+        N = atoi(optarg);
+        if (N <= 0) {
+          printf("Error: Workload size is set to %d (<0).\n", N);
+          return -1;
+        }
+        break;
+      case 't':
+        whichTestToRun = atoi(optarg);
+        if (whichTestToRun <= 0 || whichTestToRun >= 4) {
+          printf("Error: test%d() is not available.\n", whichTestToRun);
+          return -1;
+        }
+        break;
+      case 'h':
+      default:
+        usage(argv[0]);
+        return 1;
+    }
+  }
+
+#define AVX_ALIGNMENT 256
+  float *values1 = (float *)__builtin_alloca_with_align(N * sizeof(float), AVX_ALIGNMENT);
+  float *values2 = (float *)__builtin_alloca_with_align(N * sizeof(float), AVX_ALIGNMENT);
+  double *values3 = (double *)__builtin_alloca_with_align(N * sizeof(double), AVX_ALIGNMENT);
+  float *output = (float *)__builtin_alloca_with_align(N * sizeof(float), AVX_ALIGNMENT);
+#undef AVX_ALIGNMENT
+  initValue(values1, values2, values3, output, N);
+
+  printf("Running test%d()...\n", whichTestToRun);
+  fasttime_t time1 = gettime();
+  switch (whichTestToRun) {
+    case 1: test1(values1, values2, output, N); break;
+    case 2: test2(values1, values2, output, N); break;
+    case 3: test3(values3, N); break;
+  }
+  fasttime_t time2 = gettime();
+
+  double elapsedf = tdiff(time1, time2);
+  printf("Elapsed execution time of the loop in test%d():\n", whichTestToRun);
+  printf("%lfsec (N: %d, I: %d)\n", elapsedf, N, I);
+  return 0;
+}
+
+void usage(const char *progname) {
+  printf("Usage: %s [options]\n", progname);
+  printf("Program Options:\n");
+  printf("  -s  --size <N>     Use workload size N (Default = 1024)\n");
+  printf("  -t  --test <N>     Just run the testN function (Default = 1)\n");
+  printf("  -h  --help         This message\n");
+}
+
+void initValue(float *values1, float *values2, double *values3, float *output, unsigned int N) {
+  for (unsigned int i=0; i<N; i++)
+  {
+    // random input values
+    values1[i] = -1.0f + 4.0f * rand() / (float)RAND_MAX;
+    values2[i] = -1.0f + 4.0f * rand() / (float)RAND_MAX;
+    values3[i] = -1.0 + 4.0 * rand() / (double)RAND_MAX;
+    output[i] = 0.0f;
+  }
+}