Add benchmark test

arjunmehta · arjunmehta · commit 8fd007d37685 · 2018-12-01T18:11:13.000-08:00
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -19,6 +19,7 @@
   "devDependencies": {
     "babel-plugin-syntax-async-functions": "^6.5.0",
     "babel-preset-env": "^1.5.2",
+    "benchmark": "^2.1.4",
     "browser-sync": "^2.26.3",
     "browserify": "^16.2.3",
     "chai": "^4.2.0",
@@ -42,7 +43,7 @@
     "vinyl-source-stream": "^2.0.0"
   },
   "scripts": {
-    "test": "mocha ./test",
+    "test": "mocha ./test/index",
     "setup": "npm i -g gulp-cli",
     "build-docs": "./node_modules/.bin/jsdoc -c jsdoc.json src -r -d doc --debug && gulp injectCSS",
     "make": "gulp build && gulp beautify && gulp minify"
diff --git a/src/core/gpu.js b/src/core/gpu.js
@@ -31,7 +31,9 @@ class GPU extends GPUCore {
 		this._canvas = settings.canvas || null;
 		this._webGl = settings.webGl || null;
 
-		if (!mode || mode === 'webgl-validator') {
+		if (mode === 'cpu') {
+			detectedMode = 'cpu';
+		} else if (mode === 'webgl' || mode === 'webgl-validator') {
 			const context = createNodeContext(512, 512);
 			const canvas = createCanvas(512, 512);
 			detectedMode = mode || 'webgl';
diff --git a/test/benchmark.js b/test/benchmark.js
@@ -0,0 +1,47 @@
+const GPU = require('../src/index.js');
+const Benchmark = require('benchmark');
+
+const suite = new Benchmark.Suite;
+
+
+const gpuRunner = new GPU({ mode: 'webgl' });
+const cpuRunner = new GPU({ mode: 'cpu' });
+
+const size = 200;
+
+const myGPUFunc = gpuRunner
+  .createKernel(function compute() {
+    const i = this.thread.x;
+    const j = 0.89;
+    return i + j;
+  })
+  // .setOutputToTexture(true)
+  .setOutput([size, size, size]);
+
+const myCPUFunc = cpuRunner
+  .createKernel(function compute() {
+    const i = this.thread.x;
+    const j = 0.89;
+    return i + j;
+  })
+  .setOutput([size, size, size]);
+
+
+
+
+// add tests
+suite.add('cpu', function() {
+    myGPUFunc();
+  })
+  .add('gpu', function() {
+    myCPUFunc();
+  })
+  // add listeners
+  .on('cycle', function(event) {
+    console.log(String(event.target));
+  })
+  .on('complete', function() {
+    console.log('Fastest is ' + this.filter('fastest').map('name'));
+  })
+  // run async
+  .run();
diff --git a/test/index.js b/test/index.js
@@ -1,18 +1,17 @@
-const path = require('path');
 const { expect } = require('chai');
 
 const GPU = require('../src/index.js');
 
 
 describe('Test Node GPU', () => {
   it('should find gpu', () => {
-    const gpu = new GPU();
+    const gpu = new GPU({ mode: 'webgl' });
 
     const myFunc = gpu.createKernel(function compute() {
       const i = this.thread.x;
       const j = 0.89;
       return i + j;
-    }).setOutput([100]);
+    }).setOutput([1000, 1000]);
 
     console.log(myFunc());
 
