forked from gpujs/gpu.js
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathternary.js
More file actions
78 lines (61 loc) · 1.78 KB
/
ternary.js
File metadata and controls
78 lines (61 loc) · 1.78 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
const { assert, skip, test, module: describe, only } = require('qunit');
const { GPU } = require('../../src');
describe('feature: Ternary');
function ternaryTest(mode) {
const gpu = new GPU({ mode });
function ternaryFunction(value) {
return (value > 1 ? 1 : 0);
}
const kernel = gpu.createKernel(ternaryFunction, { output: [1] });
const truthyResult = kernel(100);
const falseyResult = kernel(-100);
assert.equal(truthyResult[0], 1);
assert.equal(falseyResult[0], 0);
gpu.destroy();
}
test('auto', () => {
ternaryTest();
});
test('gpu', () => {
ternaryTest('gpu');
});
(GPU.isWebGLSupported ? test : skip)('webgl', () => {
ternaryTest('webgl');
});
(GPU.isWebGL2Supported ? test : skip)('webgl2', () => {
ternaryTest('webgl2');
});
(GPU.isHeadlessGLSupported ? test : skip)('headlessgl', () => {
ternaryTest('headlessgl');
});
test('cpu', () => {
ternaryTest('cpu');
});
function ternaryWithVariableUsage(mode) {
const gpu = new GPU({ mode });
const kernel = gpu.createKernel(function(value1) {
const value2 = value1 + 1;
return value2 > 10 ? 1 : 0;
}, { output: [1] });
assert.equal(kernel(9)[0], 0);
assert.equal(kernel(10)[0], 1);
gpu.destroy();
}
test('with variable usage auto', () => {
ternaryWithVariableUsage();
});
test('with variable usage gpu', () => {
ternaryWithVariableUsage('gpu');
});
(GPU.isWebGLSupported ? test : skip)('with variable usage webgl', () => {
ternaryWithVariableUsage('webgl');
});
(GPU.isWebGL2Supported ? test : skip)('with variable usage webgl2', () => {
ternaryWithVariableUsage('webgl2');
});
(GPU.isHeadlessGLSupported ? test : skip)('with variable usage headlessgl', () => {
ternaryWithVariableUsage('headlessgl');
});
test('with variable usage cpu', () => {
ternaryWithVariableUsage('cpu');
});