gpu.js/test/benchmark.js

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 = 2048;


// SIMPLE

const myGPUFunc = gpuRunner
  .createKernel(function compute() {
    const i = this.thread.x;
    const j = 0.89;
    return i + j;
  })
  .setPipeline(true)
  .setOutput([size, size]);


const myCPUFunc = cpuRunner
  .createKernel(function compute() {
    const i = this.thread.x;
    const j = 0.89;
    return i + j;
  })
  .setOutput([size, size]);


// add tests
suite
  .add('gpu', () => {
    myGPUFunc();
  })
  .add('cpu', () => {
    myCPUFunc();
  })
  // add listeners
  .on('cycle', function(event) {
    console.log(String(event.target));
  })
  .on('complete', () => {
    console.log('Fastest is ' + this.filter('fastest').map('name'));
  })
  .run({ 'async': true });