8000 webgpu: change matmul_webgpu tile size back to 16 (#1711) · tensorflow/tfjs-core@5ed6901 · GitHub
[go: up one dir, main page]

Skip to content
This repository was archived by the owner on Aug 15, 2019. It is now read-only.

Commit 5ed6901

Browse files
kainino0xannxingyuan
authored andcommitted
webgpu: change matmul_webgpu tile size back to 16 (#1711)
INTERNAL Tile size of 32 is causing failures and weird stuff on Mac. This was untested because WEBGPU_MATMUL_WORK_PER_THREAD defaults to 2 (and uses the packed version instead).
1 parent 68375f5 commit 5ed6901

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/backends/webgpu/src/kernels/matmul_webgpu.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ export class MatMulProgram implements WebGPUProgram {
2323
dispatch: [number, number, number];
2424
variableNames = ['A', 'B'];
2525
uniforms = 'uint dimAOuter, dimInner, dimBOuter, batch;';
26-
tileSize: [number, number] = [32, 32]; // Must be square.
26+
tileSize: [number, number] = [16, 16]; // Must be square.
2727

2828
constructor(outputShape: [number, number, number]) {
2929
this.outputShape = outputShape;

0 commit comments

Comments
 (0)
0