added driver

examples_ptx/nbody/nbody.cpp

@@ -0,0 +1,73 @@
+#include <cstdio>
+#include <cstdlib>
+#include <algorithm>
+#include <iostream>
+#include <cassert>
+#include <iomanip>
+#include "../timing.h"
+#include "../ispc_malloc.h"
+#include "nbody_ispc.h"
+#include "plummer.h"
+
+typedef double real;
+
+int main (int argc, char *argv[])
+{
+  int i, j, n = argc == 1 ? 1024*1024: atoi(argv[1]), m = n < 100 ? 1 : 50, l = n < 100 ? n : RAND_MAX;
+  double tISPC1 = 0.0, tISPC2 = 0.0, tSerial = 0.0;
+
+  Plummer plummer(n);
+
+  real *posx = new real[n];
+  real *posy = new real[n];
+  real *posz = new real[n];
+  real *velx = new real[n];
+  real *vely = new real[n];
+  real *velz = new real[n];
+  real *mass = new real[n];
+
+#pragma omp parallel for
+  for (int i = 0; i < n; i++)
+  {
+    posx[i] = plummer.pos[i].x;
+    posy[i] = plummer.pos[i].y;
+    posz[i] = plummer.pos[i].z;
+    velx[i] = plummer.vel[i].x;
+    vely[i] = plummer.vel[i].y;
+    velz[i] = plummer.vel[i].z;
+    mass[i] = plummer.mass[i];
+  }
+
+  ispcSetMallocHeapLimit(1024*1024*1024);
+  ispc::openNbody(n);
+
+  const int nSteps = 10;
+  const real dt = 0;
+  tISPC2 = 1e30;
+  for (i = 0; i < m; i ++)
+  {
+    reset_and_start_timer();
+    ispc::nbodyIntegrate(
+        nSteps, n, dt,
+        posx, posy, posz, mass,
+        velx, vely, velz,
+        NULL);
+    tISPC2 = get_elapsed_msec();
+    fprintf(stderr, " %d iterations took %g sec; perf= %g GFlops\n",
+        nSteps, tISPC2/1e3,
+        20*n*n/(tISPC2/1e3/1e9));
+  }
+
+  ispc::closeNbody();
+
+  delete posx;
+  delete posy;
+  delete posz;
+  delete velx;
+  delete vely;
+  delete velz;
+  delete mass;
+
+
+  return 0;
+}