forked from gpujs/gpu.js
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathindex.js
More file actions
72 lines (54 loc) · 1.71 KB
/
index.js
File metadata and controls
72 lines (54 loc) · 1.71 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
const { expect } = require('chai');
const GPU = require('../src/index.js');
describe('Test Node GPU', () => {
describe('gpu mode', () => {
it('should find and use gpu runner', () => {
const gpu = new GPU({ mode: 'gpu' });
const kernel = gpu.createKernel(function() {
return 1;
}).setOutput([1]);
const result = kernel();
expect(gpu.runner.constructor).to.equal(GPU.HeadlessGLRunner);
expect(result[0]).to.equal(1);
});
it('supports 2x2 size', () => {
const gpu = new GPU({ mode: 'gpu' });
const kernel = gpu.createKernel(function() {
return this.thread.x * this.thread.y;
}).setOutput([2, 2]);
const result = kernel();
expect(gpu.runner.constructor).to.equal(GPU.HeadlessGLRunner);
expect(result).to.deep.equal(
[
Float32Array.from([0,0]),
Float32Array.from([0,1])
]
);
});
});
describe('cpu mode', () => {
it('should find and use gpu runner', () => {
const gpu = new GPU({ mode: 'cpu' });
const kernel = gpu.createKernel(function() {
return 1;
}).setOutput([1]);
const result = kernel();
expect(gpu.runner.constructor).to.equal(GPU.CPURunner);
expect(result[0]).to.equal(1);
});
it('supports 2x2 size', () => {
const gpu = new GPU({ mode: 'cpu' });
const kernel = gpu.createKernel(function() {
return this.thread.x * this.thread.y;
}).setOutput([2, 2]);
const result = kernel();
expect(gpu.runner.constructor).to.equal(GPU.CPURunner);
expect(result).to.deep.equal(
[
Float32Array.from([0,0]),
Float32Array.from([0,1])
]
);
});
});
});