gpu.js/test/features/add-custom-native-function.js
Robert Plummer b2df0ab604 fix #316
Change `HTMLImageArray` render strategy to `gl.NEAREST`, like all the others
Let `WebGL2FunctionNode` extend `WebGLFunctionNode` and lighten
Remove all the different html pages associated with tests, and just use one file to handle them and use qunit for filtering them
Bump version number
2018-06-12 14:07:41 -04:00

48 lines
1.1 KiB
JavaScript

function addCustomNativeFunctionDivide(mode) {
var gpu = new GPU({ mode: mode });
gpu.addNativeFunction('divide', `float divide(float a, float b) {
return a / b;
}`);
function divide(a,b) {
return a / b;
}
var f = gpu.createKernel(function(a, b) {
return divide(a[this.thread.x], b[this.thread.x]);
}, {
functions: [divide],
output : [6]
});
QUnit.assert.ok( f !== null, 'function generated test');
var a = [1, 2, 3, 5, 6, 7];
var b = [4, 5, 6, 1, 2, 3];
var res = f(a,b);
var exp = [0.25, 0.4, 0.5, 5, 3, 2.33];
for(var i = 0; i < exp.length; ++i) {
QUnit.assert.close(res[i], exp[i], 0.1, 'Result arr idx: '+i);
}
}
QUnit.test( 'addCustomNativeFunctionDivide (auto)', function() {
addCustomNativeFunctionDivide(null);
});
QUnit.test( 'addCustomNativeFunctionDivide (gpu)', function() {
addCustomNativeFunctionDivide('gpu');
});
QUnit.test( 'addCustomNativeFunctionDivide (webgl)', function() {
addCustomNativeFunctionDivide('webgl');
});
QUnit.test( 'addCustomNativeFunctionDivide (webgl2)', function() {
addCustomNativeFunctionDivide('webgl2');
});
//TODO: CPU?