Skip to content

[ExecuTorch][WebGPU] Add q8ta_add op (int8 elementwise add)#21191

Merged
meta-codesync[bot] merged 2 commits into
gh/JCNTH/151/basefrom
gh/JCNTH/151/head
Jul 24, 2026
Merged

[ExecuTorch][WebGPU] Add q8ta_add op (int8 elementwise add)#21191
meta-codesync[bot] merged 2 commits into
gh/JCNTH/151/basefrom
gh/JCNTH/151/head

Conversation

@JCNTH

@JCNTH JCNTH commented Jul 22, 2026

Copy link
Copy Markdown
Contributor

Stack from ghstack (oldest at bottom):

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port et_vk.q8ta_add (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute a + alpha*b, requantize. Reuses the C0 int8 buffer path (array<u32> pack/unpack, is_int8 guard, multiply-by-inv_scale).

Implementation: Q8taAdd.cpp registers et_vk.q8ta_add.default (out = args.back()), guards all three tensors int8 + equal numel + numel % 4 == 0, fail-loud; q8ta_add.wgsl unpacks 4 int8/word, dequantizes scale*(q - zero_point), forms a + alpha*b, and requantizes clamp(round(v * inv_output_scale) + zero_point, -128, 127). Mirrors Vulkan q8ta_binary.glsl + common.glslh, with one deliberate divergence: the Vulkan glsl buffer kernel drops alpha (computes a + b), so this port implements the full a + alpha*b to match the CPU reference for any alpha (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-numel guard rejects it, matching the add op baseline); supports_resize=False (no resize hook), mirroring the partitioner.
@exported-using-ghexport

Differential Revision: D112257597

Differential Revision: D112257597

[ghstack-poisoned]
@pytorch-bot

pytorch-bot Bot commented Jul 22, 2026

Copy link
Copy Markdown

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/21191

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit ff9b3d7 with merge base 4a26c64 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@github-actions

Copy link
Copy Markdown

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

This was referenced Jul 22, 2026
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
JCNTH added a commit that referenced this pull request Jul 24, 2026
Pull Request resolved: #21191

Problem: The WebGPU delegate has no quantized elementwise add — the first C1 q8ta op, on the int8 activation path C0 established.

Solution: Port `et_vk.q8ta_add` (two int8 tensors + per-tensor qparams -> int8): dequantize both, compute `a + alpha*b`, requantize. Reuses the C0 int8 buffer path (`array<u32>` pack/unpack, `is_int8` guard, multiply-by-`inv_scale`).

Implementation: `Q8taAdd.cpp` registers `et_vk.q8ta_add.default` (`out = args.back()`), guards all three tensors int8 + equal `numel` + `numel % 4 == 0`, fail-loud; `q8ta_add.wgsl` unpacks 4 int8/word, dequantizes `scale*(q - zero_point)`, forms `a + alpha*b`, and requantizes `clamp(round(v * inv_output_scale) + zero_point, -128, 127)`. Mirrors Vulkan `q8ta_binary.glsl` + `common.glslh`, with one deliberate divergence: the Vulkan glsl buffer kernel drops `alpha` (computes `a + b`), so this port implements the full `a + alpha*b` to match the CPU reference for any `alpha` (surfaced in the handler comment).

Constraints: same-shape only (no broadcast — the equal-`numel` guard rejects it, matching the `add` op baseline); `supports_resize=False` (no resize hook), mirroring the partitioner.
ghstack-source-id: 406366820
@exported-using-ghexport

Differential Revision: [D112257597](https://our.internmc.facebook.com/intern/diff/D112257597/)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants