sanmonga22 commited on
Commit
7d7504e
·
verified ·
1 Parent(s): a437e4e

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +62 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags: [qhexrt, hexagon, npu, qnn, llm, gated-delta-net]
4
+ base_model: Qwen/Qwen3.5-4B
5
+ pipeline_tag: text-generation
6
+ ---
7
+
8
+ # Qwen3.5-4B — Hexagon NPU (QHexRT) — v81
9
+
10
+ A prebuilt **QHexRT** bundle of **Qwen/Qwen3.5-4B** (the **Qwen3-Next "GatedDeltaNet" hybrid** lineage)
11
+ running on the **Qualcomm Hexagon v81 NPU** (SM8850). text→text.
12
+
13
+ The text decoder is a **hybrid**: 3 of every 4 layers are a **recurrent gated delta-rule linear attention**
14
+ (`Qwen3_5GatedDeltaNet`: a short causal-conv FIFO + an SSM state + gated RMSNorm), and every 4th layer is
15
+ **gated softmax attention** (q-proj output gate, per-head q/k RMSNorm, **partial-rotary** M-RoPE θ=1e7). The
16
+ runtime drives it with the `qwen3_5_generate` host-op (decode-over-prompt; conv + SSM + KV state carried
17
+ host-side), the lm-head as a separate graph.
18
+
19
+ **Multi-context decode.** The 32-layer W8 decode (3.6 GB) exceeds the v81 cDSP **per-context** memory ceiling
20
+ (`contextCreateFromBinary → 0x3ea MEM_ALLOC`). So the decode is split into **2 context shards** by layer range
21
+ (0–15 / 16–31, ~1.81 GB each); the host-op runs them in sequence, threading the residual stream shard→shard
22
+ (only the last shard applies the final norm). Each shard updates only its own layers' conv/SSM/KV state. This
23
+ is numerically identical to a single decode graph — parity stays greedy-exact.
24
+
25
+ ## What's here (`v81/` — the flat artifacts root)
26
+ | file | role | ~size |
27
+ |---|---|---|
28
+ | `qwen3.5-4b-1024.json` | QHexRT manifest (the declarative run plan) | 2 KB |
29
+ | `qwen354b_decode_s0_w8.bin` | decode shard 0 (layers 0–15, W8) | 1.81 GB |
30
+ | `qwen354b_decode_s1_w8.bin` | decode shard 1 (layers 16–31, W8) | 1.81 GB |
31
+ | `qwen354b_lmhead_f16.bin` | lm-head context binary (tied embed, fp16) | 1.27 GB |
32
+ | `qwen354b_embed_f16.bin` | embedding table (host lookup, fp16) | 1.27 GB |
33
+ | `tokenizer.json` | Qwen2-style BPE tokenizer | 20 MB |
34
+
35
+ Arch-pinned: a v81 binary will not load on another Hexagon arch. The QNN runtime libs come from the QAIRT SDK,
36
+ not this repo.
37
+
38
+ ## Run
39
+ ```bash
40
+ hf download runanywhere/qwen3_5_4b_HNPU --local-dir q35_4b
41
+ adb push q35_4b/v81 /data/local/tmp/wq/qwen35-4b
42
+ # also stage the QAIRT runtime libs + a qhx_generate built with the qwen3_5_generate host-op (see caveats)
43
+ adb shell "cd /data/local/tmp/wq && LD_LIBRARY_PATH=. \
44
+ ./qhx_generate qwen35-4b/qwen3.5-4b-1024.json libQnnHtp.so libQnnSystem.so qwen35-4b 64 'The capital of France is'"
45
+ ```
46
+
47
+ ## Measured (device: SM8850 / v81, QAIRT 2.47)
48
+ - **Decode: ~5.5 tok/s** (W8 weight-only, 2 context shards).
49
+ - **Parity: greedy-EXACT** — on-device tokens match the HF `Qwen/Qwen3.5-4B` greedy continuation **64/64**
50
+ (prompt "The capital of France is").
51
+
52
+ ## Caveats
53
+ - **W8 weight-only** decode (the ~7 GB fp16 decode exceeds device RAM; W8 ~3.6 GB) + **fp16** lm-head. Greedy
54
+ parity is exact under W8 (64/64) for this model.
55
+ - **Multi-context**: the decode is 2 shards (`decode_s0` + `decode_s1`) plus the lm-head — 3 context binaries
56
+ loaded together. A single decode context this size fails on v81 (per-context alloc ceiling, `0x3ea`).
57
+ - The W8 decode `.so` (per shard ~1.8 GB) is built with a **large-code-model** path (weight blob → `.lrodata`
58
+ + `-mcmodel=large`); the lm-head ONNX is built directly (MatMul + external fp16 weight) to dodge the 2 GB
59
+ protobuf limit. Both are baked into the published bins.
60
+ - **Text-LLM path only** — the vision tower + MTP head are not exported.
61
+ - Needs a `qhx_generate` that includes the **`qwen3_5_generate`** host-op (QHexRT branch `smonga/qwen_fam`).
62
+ - Built by the in-repo `forge` pipeline (oracle-gated export 10/10 → QAIRT-2.47 O3 compile → device greedy gate).