bhanc commited on
Commit
27ecbb4
·
verified ·
1 Parent(s): 4ce8047

fix(xnnpack): preserve PLE dequantization and sensitive projection precision

Browse files
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 2.6 GiB
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:9f52126bea0f7293a2e45d62ec746a780b9a7fd936f2836b2eee8a22336311b8
3
- size 2809192448
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d11e8587cd20015e01aed66c0880e3a2cefd464073a2f80b74a666246ab71cc7
3
+ size 3215277312