llama.cpp-openPangu-2.0-Flash / PROMPT_REPRODUCE.md
HelloSun
docs: add PROMPT_REPRODUCE.md (step-by-step task brief to reproduce this port + experiment project); fix diff stat in file list; link prompt in README
7263337
|
Raw History Blame Contribute Delete
9.16 kB

任務簡報(Prompt):重現「openPangu-2.0-Flash × llama.cpp v0.7.0 移植 + CPU 優化」專案

把這整份文件當作任務書交給一個具備 shell / git / 網路存取能力的 coding agent, 在一台 x86 CPU 機器上逐步完成。所有數字都是「本專案在我們的機器上」的參考值, 重現時應以你自己的基準量測為準(百分比方向應相近,絕對值依機器而異)。

一、任務目標

  1. 把 openPangu-2.0-Flash 的 llama.cpp 支援,從舊 fork 手工移植到最新的 llama.cpp v0.7.0(commit 620d2225cc436ab0b001e3200e590e3b286ef9ab, 2026-10-15)。
  2. 逐一實驗一批上游 CPU 加速 PR,用嚴謹的 A/B 協議量測,只有真實有幫助的才併入。
  3. 把全部結果(含失敗實驗)寫成 model card,打包成 diff + patch.sh, 上傳到 HuggingFace repo:**HelloSun/llama.cpp-openPangu-2.0-Flash**。

二、素材與來源

項目 位置
待移植的原始 fork https://github.com/mrexodia/llama.cpp-openPangu-2.0-Flash(HEAD 76c92350b,基於 llama.cpp 6f4f53f2b, 2026-06-29)
移植目標 base llama.cpp v0.7.0,commit 620d2225cc436ab0b001e3200e590e3b286ef9ab
模型 openPangu-2.0-Flash-base-Q3_K_M.gguf(92B MoE、約 6B active、512K context、40.7 GiB);另有 split MTP draft head openPangu-2.0-Flash-mtp-Q8_0.gguf(Q8_0)
模型下載 https://huggingface.co/mrexodia/openPangu-2.0-Flash-GGUF
最終上傳 repo https://huggingface.co/HelloSun/llama.cpp-openPangu-2.0-Flash(需 HF token)

三、環境與限制

  • CPU:Intel Xeon Platinum 8559C(Emerald Rapids,有 AMX)、2 NUMA nodes。
  • RAM 充足(TiB 級),但有記憶體上限規則:任何單一程式 RSS 不得超過 64 GiB、 總和 80 GiB,超過會被砍(exit 137)。→ 全程使用 Q3_K_M(40.7 GiB)量化; default + MTP draft 會超上限、跑不了(要記錄這件事)。
  • Host 是共享機器、噪音 ±2–4% → 所有效能比較必須同場 interleaved 交替量測。
  • 工具怪癖:bc、/usr/bin/time 可能缺失(用 $SECONDS 或 python 計時); llama-cli 極小 -c 512 的 decode 異常慢(用 -c 8192 做 E2E smoke); v0.7.0 的 llama-cli 沒有 -no-cnv 旗標(用 -p 單輪模式)。

四、工作流程

步驟 1:移植(port)

  • 在 v0.7.0 base 上逐步搬遷 fork 的 openpangu-v2 支援(逐檔手工適配, 別整包 cherry-pick,v0.7.0 的 KV-cache / batch / graph 都已重構)。
  • 已知坑(必須修):原始 fork 在 AMX 機器以 default 設定啟動會 GGML_ASSERT(*cur_backend_id != -1)(sched_reserve 階段)——根因是 AMX weight repacking 對 batched/3D 權重的打包 bug。移植時修掉它,讓 default 可跑。
  • 正確性錨點:llama-perplexity -m <Q3_K_M> -f <小型文字語料> -c 2048 -t 16 → 記下基準 PPL(本專案為 1.0361;絕對值取決於語料,重現時以「候選與基準 必須完全相同」為準)。每次併入優化後都要重跑確認相同。

步驟 2:建立效能基準

  • 基準指令:llama-bench -m <Q3_K_M> -t 16 -r 3 -p 512 -n 128 [--repack 0]
  • 原始移植的參考值(本機):default pp512 ≈ 57 / tg128 ≈ 10.9, -nr(--repack 0)pp512 ≈ 62–64 / tg128 ≈ 10.5–11。

步驟 3:A/B 協議(每個候選 PR 都照做)

  1. 從上一個已併入狀態的源碼分別建兩個 binary: bin_base(baseline)與 bin_cand(baseline + 該 PR)。
  2. 同一場 session 內交替執行 base/cand(各 3 輪以上),取中位數與每輪勝負 (0.5% 內算平手)。
  3. 增量 = vs 前一個已併入狀態(不是 vs 原始移植!),逐 PR 疊加才準。
  4. 正確性:PPL 與基準完全相同;受影響 kernel 跑 test-backend-ops -b CPU -o MUL_MAT -p 'type=<qX_K>' 子集。
  5. 只併入真實 positive;negative 只記錄數據,不併入。
  6. 重要陷阱:**llama-bench 不吃 LLAMA_ARG_REPACK=0 環境變數** (只認 --repack 0 CLI flag;env 變數對 llama-cli/perplexity 有效)。

步驟 4:候選清單與已知結論(重現時應得到相近方向)

PR 內容 本專案結論(增量 t/s)
#27024 AMX q8_K VNNI gemv ✅ 併入(pp +3.4% / tg +2.7%)
#22525 q4_K AVX-512 GEMM interleave/staging ✅ 併入(pp +4.0% / tg +4.9%)
#19706 Q6_K interleave gemm ❌ 不併(≈ 0,重測仍近似噪音)
#27590 Q5_K/Q6_K AVX-512/VNNI dot ✅ 併入(-nr 下 pp +0.7% / tg +1.4%,4/4 輪勝;上游已關閉未合併,本地驗證)
#18698 repack 平行化 + NUMA affinity ❌ 不併(pp −4.1% / tg −2.1% 回歸;但載入時間 24s→10s;手工移植版存檔備查)
#27032 draft:GET_ROWS+ADD 融合、get_rows fast path 等 ❌ 不併(只有 ggml kernel 子集可移植,≈ 中性;此模型圖中無相鄰 GET_ROWS→ADD、token_embd 是 Q3_K,機制無用武之地)
#27396 draft:speculative draft-model prefill(EAGLE3/DFlash/GLM4) ❌ 不適用(CPU 端不用投機解碼、MTP 在 CPU 是淨虧;7/17 檔衝突)
  • 融合/移植技巧:可只取 PR 中可獨立套用的子集(例如 git apply --include='ggml/*'),再單獨驗證。
  • 若某 PR 3-way apply 衝突太多、機制與本模型圖不符 → 判「不適用、未移植」, 寫進 model card 即可,不要硬改核心。

步驟 5:收尾實驗(重要發現,要寫進 card)

  • 最終組合 A/B(vs 原始移植,同一 session,3 輪):本專案得到 default **pp +1.2% / tg +3.4%**(每項 3/3 輪勝); -nr **pp +6.4% / tg +0.6%**。
  • -nr vs default:本 AMX 機器上 -nr 明顯更快, pp512 +16.6% / tg128 +6.1%(default 的 repack kernels 沒有 AMX 路徑)。
  • MTP draft head 在 CPU 上是淨虧(4.7 vs 11.5+ t/s),純記錄。

步驟 6:測試整合與驗收(shipped 樹必須全綠)

  1. 新 arch 加入後,完整 ctest 常見兩個既有缺口,需修:
    • test-generate-models(test-llama-archs):把 openpangu-v2 標為 MoE-mandatory、補齊其必需 hparams(MLA 幾何 key/value_length_mla、 hyper-connection trio、attention.sink_count / conv_kernel_size、 per-layer attention.layer_swa_window,fixture 層數 ≥ 4, 讓損壞態還原測試(rf)在寫入前被 metadata 檢查攔下)。
    • llama-model-saver:補寫 attention.sink_count / attention.conv_kernel_size / attention.layer_swa_window, 否則合成模型存檔後無法重載(test-llama-context 全紅)。
    • test-llama-context 的 replay case:K sink prefix 只支援 unified cache(多 stream 會 GGML_ABORT)→ 加 per-case skip (arch_requires_unified_kv,比照既有 arch_reserves_single_seq 寫法)。
  2. 驗收標準:ctest 61/61 全過;test-llama-context --models 對 132 個 合成模型 132/132 通過。
  3. 全新 E2E:乾淨目錄 → ./patch.sh(下載 v0.7.0 tarball、套 diff)→ 建置 → PPL 與基準相同 → bench spot-check → llama-cli -c 8192 -p ... 能生成。

五、交付物與上傳要求

最終 HF repo 應包含:

  • llama.cpp-openpangu-v0.7.0.diff — 完整 patch(本專案最終:48 檔、 +3340 / −368)
  • patch.sh — 下載 pinned commit、套 patch、印建置與執行說明
  • README.md — model card:
    • 移植說明、與原 fork 的相容性/效能比較(原 fork 在 AMX default 會 abort)
    • 逐 PR 表格,每個機制標「⭕ 有幫助 / ❌ 沒幫助」並列出前後 tok/s
    • 測試環境、測試結果、專有名詞(TW 用語)
  • LICENSE + LICENSE.header — GPL-3.0(patch 集本身), 內含使用來源的 MIT 參考清單:llama.cpp、mrexodia fork、ggml、 以及有被併入的上游 PR(#27024 @jinzihao、#22525 @HyeongiJeon、 #27590 @SongXiaoXi)——未被併入的 PR 不列入引用清單。
  • 上傳節奏:每測完一個機制就分階段上傳一次 HF(commit 訊息記清楚該階段 內容);最後一次為最終版(diff + README + patch.sh + LICENSE + 本 prompt)。

六、成功標準(Definition of Done)

  1. patch.sh 在乾淨 v0.7.0 上:套用成功 → 建置成功 → 載入模型正確生成 (PPL 與基準相同)。
  2. 完整 ctest 全綠(含 openpangu 相關 fixtures 與 context cases)。
  3. Model card 完整記錄全部實驗(成功與失敗都有數據、結論與原因)。
  4. HF repo 內含所有交付物且與本地檔案一致(雜湊一致)。