fix(xnnpack): preserve PLE dequantization and sensitive projection precision
Browse files- README.md +11 -2
- e2b/xnnpack/gemma_4_e2b_xnnpack_8da4w.pte +2 -2
README.md
CHANGED
|
@@ -23,7 +23,7 @@ library as ExecuTorch `.pte` programs, ready to run on device.
|
|
| 23 |
| --- | --- | --- | --- |
|
| 24 |
| `e2b/mlx/gemma_4_e2b_mlx_int4.pte` | mlx | int4 | 4,096 |
|
| 25 |
| `e2b/vulkan/gemma_4_e2b_vulkan_8da4w.pte` | vulkan | 8da4w | 2,048 |
|
| 26 |
-
| `e2b/xnnpack/gemma_4_e2b_xnnpack_8da4w.pte` | xnnpack | 8da4w | 4,096 |
|
| 27 |
|
| 28 |
|
| 29 |
|
|
@@ -38,9 +38,18 @@ e2b/tokenizer_config.json 21.8 KiB
|
|
| 38 |
e2b/vulkan/config.json 1.3 KiB
|
| 39 |
e2b/vulkan/gemma_4_e2b_vulkan_8da4w.pte 2.8 GiB
|
| 40 |
e2b/xnnpack/config.json 1.3 KiB
|
| 41 |
-
e2b/xnnpack/gemma_4_e2b_xnnpack_8da4w.pte
|
| 42 |
```
|
| 43 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
## Compatibility
|
| 45 |
|
| 46 |
These files are published for the **ExecuTorch v1.4.1** runtime. ExecuTorch
|
|
|
|
| 23 |
| --- | --- | --- | --- |
|
| 24 |
| `e2b/mlx/gemma_4_e2b_mlx_int4.pte` | mlx | int4 | 4,096 |
|
| 25 |
| `e2b/vulkan/gemma_4_e2b_vulkan_8da4w.pte` | vulkan | 8da4w | 2,048 |
|
| 26 |
+
| `e2b/xnnpack/gemma_4_e2b_xnnpack_8da4w.pte` | xnnpack | 8da4w + fp16 | 4,096 |
|
| 27 |
|
| 28 |
|
| 29 |
|
|
|
|
| 38 |
e2b/vulkan/config.json 1.3 KiB
|
| 39 |
e2b/vulkan/gemma_4_e2b_vulkan_8da4w.pte 2.8 GiB
|
| 40 |
e2b/xnnpack/config.json 1.3 KiB
|
| 41 |
+
e2b/xnnpack/gemma_4_e2b_xnnpack_8da4w.pte 3.0 GiB
|
| 42 |
```
|
| 43 |
|
| 44 |
+
## XNNPACK precision
|
| 45 |
+
|
| 46 |
+
The XNNPACK export uses mixed precision: MLP and language-model head linears
|
| 47 |
+
use 8da4w, while attention projections and the PLE residual output projections
|
| 48 |
+
remain fp16. Token embeddings remain int4. The small PLE context projection
|
| 49 |
+
weights are materialized from their quantized values before lowering to
|
| 50 |
+
preserve dequantization semantics. The model filename and runtime API remain
|
| 51 |
+
unchanged; maximum prefill is 512 tokens and context length is 4,096 tokens.
|
| 52 |
+
|
| 53 |
## Compatibility
|
| 54 |
|
| 55 |
These files are published for the **ExecuTorch v1.4.1** runtime. ExecuTorch
|
e2b/xnnpack/gemma_4_e2b_xnnpack_8da4w.pte
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d11e8587cd20015e01aed66c0880e3a2cefd464073a2f80b74a666246ab71cc7
|
| 3 |
+
size 3215277312
|