saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-16 21:00:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[js/webgpu] fix maxpool / fp16 (#19981)

Browse source
This commit is contained in:
Guenther Schmuelling 2024-03-19 16:15:49 -07:00 committed by GitHub
parent 597e828aae
commit c45cff60cf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
js/web/lib/wasm/jsep/webgpu/ops/pool.ts
View file

@ -381,8 +381,9 @@ const createMaxPoolProgramInfo =
programUniforms
}),
getShaderSource: shaderHelper => generatePoolingCode(
shaderHelper, x, input.dims.length, outputShape.length, adjustedAttributes, op1, op2, -1e5, uniforms,
hasPads, pwStartEndNotZero, phStartEndNotZero),
shaderHelper, x, input.dims.length, outputShape.length, adjustedAttributes, op1, op2,
(input.dataType === DataType.float16) ? -65504 : -1e5, uniforms, hasPads, pwStartEndNotZero,
phStartEndNotZero),
};
};