HelloSun commited on
Commit
2b6ab7c
·
verified ·
1 Parent(s): afafadc

Add EPQ pager source and reproducible scripts

Browse files
sdq-ling/0001-sdq-ling.patch ADDED
@@ -0,0 +1,231 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ diff --git a/ggml/src/ggml-cpu/amx/amx.cpp b/ggml/src/ggml-cpu/amx/amx.cpp
2
+ index e90c279..49be9e1 100644
3
+ --- a/ggml/src/ggml-cpu/amx/amx.cpp
4
+ +++ b/ggml/src/ggml-cpu/amx/amx.cpp
5
+ @@ -152,6 +152,13 @@ class extra_buffer_type : ggml::cpu::extra_buffer_type {
6
+ if (!ggml_is_contiguous(src0) || !ggml_is_contiguous(src1)) {
7
+ return false;
8
+ }
9
+ + // SDQ: the AMX weight repack (ggml_backend_amx_convert_weight) only converts the
10
+ + // ne0 x ne1 slice of a tensor, so a 3D/4D weight would be left with unconverted
11
+ + // data in the remaining slices. MLA attention feeds a 3D wv_b to MUL_MAT, which
12
+ + // lands here and then reads garbage (NaN) - keep 3D weights in the CPU buffer.
13
+ + if (src0->ne[2] > 1 || src0->ne[3] > 1) {
14
+ + return false;
15
+ + }
16
+ if (!src0->buffer || src0->buffer->buft != ggml_backend_amx_buffer_type()) {
17
+ return false;
18
+ }
19
+ diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
20
+ index 97250c2..1fcfb6e 100644
21
+ --- a/src/CMakeLists.txt
22
+ +++ b/src/CMakeLists.txt
23
+ @@ -18,6 +18,9 @@ set(LLAMA_CORE_SOURCES
24
+ llama-cparams.cpp
25
+ llama-grammar.cpp
26
+ llama-graph.cpp
27
+ + sdq_pager.cpp
28
+ + sdq_buft.cpp
29
+ + sdq_moe.cpp
30
+ llama-hparams.cpp
31
+ llama-impl.cpp
32
+ llama-io.cpp
33
+ @@ -51,6 +54,11 @@ add_library(llama
34
+ ${LLAMA_MODELS_SOURCES}
35
+ )
36
+
37
+ +# SDQ: chat / benchmark front end (links only llama + ggml)
38
+ +add_executable(sdq-chat sdq_cli.cpp)
39
+ +target_link_libraries(sdq-chat PRIVATE llama)
40
+ +target_include_directories(sdq-chat PRIVATE .)
41
+ +
42
+ set_target_properties(llama PROPERTIES
43
+ VERSION ${LLAMA_VERSION_BASE}
44
+ SOVERSION ${LLAMA_VERSION_MAJOR}
45
+ @@ -61,11 +69,14 @@ set_target_properties(llama PROPERTIES
46
+
47
+ # exclude non-model sources from unity build
48
+ set_source_files_properties(${LLAMA_CORE_SOURCES} ../include/llama.h unicode.h
49
+ + sdq_pager.cpp sdq_buft.cpp sdq_moe.cpp
50
+ PROPERTIES SKIP_UNITY_BUILD_INCLUSION ON)
51
+
52
+ configure_file(llama-version.h.in ${CMAKE_CURRENT_BINARY_DIR}/llama-version.h @ONLY)
53
+
54
+ target_include_directories(llama PRIVATE . ${CMAKE_CURRENT_BINARY_DIR})
55
+ +# SDQ needs ggml internal headers (custom buffer type + quantized dot kernels)
56
+ +target_include_directories(llama PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/../ggml/src)
57
+ target_include_directories(llama PUBLIC ../include)
58
+ target_compile_features (llama PRIVATE cxx_std_17) # don't bump
59
+
60
+ diff --git a/src/llama-context.cpp b/src/llama-context.cpp
61
+ index 9dcaf05..4fd893d 100644
62
+ --- a/src/llama-context.cpp
63
+ +++ b/src/llama-context.cpp
64
+ @@ -4,6 +4,9 @@
65
+ #include "llama-arch.h"
66
+ #include "llama-graph.h"
67
+ #include "llama-impl.h"
68
+ +
69
+ +// SDQ: expert pager
70
+ +#include "sdq_pager.h"
71
+ #include "llama-batch.h"
72
+ #include "llama-io.h"
73
+ #include "llama-memory.h"
74
+ @@ -2122,6 +2125,12 @@ int llama_context::decode(const llama_batch_ext & batch_inp) {
75
+ // wait for the computation to finish (automatically done when obtaining the model output)
76
+ //synchronize();
77
+
78
+ + // ---- SDQ: 收掉這一步 decode 的 accounting。
79
+ + // 這是所有前端都會經過的單一出口;sdq-chat 自己另外驅動,所以不會重複計數。
80
+ + if (sdq::sdq_enabled()) {
81
+ + sdq::sdq_on_decode((int) n_tokens_all);
82
+ + }
83
+ +
84
+ return 0;
85
+ }
86
+
87
+ diff --git a/src/llama-graph.cpp b/src/llama-graph.cpp
88
+ index 23e4c90..4769929 100644
89
+ --- a/src/llama-graph.cpp
90
+ +++ b/src/llama-graph.cpp
91
+ @@ -1,5 +1,7 @@
92
+ #include "llama-graph.h"
93
+
94
+ +#include "sdq_pager.h"
95
+ +
96
+ #include "llama-impl.h"
97
+ #include "llama-model.h"
98
+ #include "llama-moe-cache.h"
99
+ @@ -2211,6 +2213,40 @@ ggml_tensor * llm_graph_context::build_moe_ffn(
100
+ //call early so that topk-moe can be used
101
+ ggml_build_forward_expand(gf, weights);
102
+
103
+ + // SDQ expert pager: route the selected experts through the SSD->RAM pager
104
+ + // instead of ggml_mul_mat_id, so the expert weights never have to be resident.
105
+ + // The router (logits / gating / group top-k / weight norm) is still computed
106
+ + // by ggml above; only "evaluate the selected experts and accumulate" moves.
107
+ + if (const char * dp = getenv("SDQ_GRAPH_DUMP")) {
108
+ + const int dl = getenv("SDQ_GRAPH_DUMP_LAYER") ? atoi(getenv("SDQ_GRAPH_DUMP_LAYER")) : -1;
109
+ + if (dl < 0 || il == dl) {
110
+ + ggml_tensor * c3 = ggml_reshape_3d(ctx0, cur, n_embd, 1, n_tokens);
111
+ + ggml_build_forward_expand(gf, c3);
112
+ + ggml_build_forward_expand(gf, selected_experts);
113
+ + ggml_build_forward_expand(gf, weights);
114
+ + std::string path = std::string(dp) + "." + std::to_string(il);
115
+ + FILE * f = fopen(path.c_str(), "wb");
116
+ + if (f) {
117
+ + const int32_t meta[8] = { (int32_t) n_embd, (int32_t) n_tokens, (int32_t) n_expert_used,
118
+ + (int32_t) n_expert, (int32_t) (sdq::sdq_pager_active_for_layer(il) ? 1 : 0), 0, 0, 0 };
119
+ + fwrite(meta, sizeof(int32_t), 8, f);
120
+ + fwrite(c3->data, sizeof(float), (size_t) (n_embd * n_tokens), f);
121
+ + fwrite(selected_experts->data, sizeof(int32_t), (size_t) (n_expert_used * n_tokens), f);
122
+ + fwrite(weights->data, sizeof(float), (size_t) (n_expert_used * n_tokens), f);
123
+ + fclose(f);
124
+ + }
125
+ + }
126
+ + }
127
+ +
128
+ + if (sdq::sdq_pager_active_for_layer(il) && cur->ne[0] == n_embd) {
129
+ + const float swiglu_limit = il >= 0 ? hparams.swiglu_clamp_exp[il] : 0.0f;
130
+ + ggml_tensor * moe_out = sdq::sdq_build_moe(gf, ctx0, cur, selected_experts, weights, il,
131
+ + n_expert, n_expert_used, swiglu_limit);
132
+ + cb(moe_out, "ffn_moe_out", il);
133
+ + moe_out = sdq::sdq_maybe_dump_tensor(ctx0, moe_out, il);
134
+ + return moe_out;
135
+ + }
136
+ +
137
+ // the experts of host-resident layers may be read from the MoE cache
138
+ ggml_tensor * slots = build_moe_cache_slots(selected_experts, up_exps, gate_exps, down_exps, gate_up_exps, il);
139
+
140
+ @@ -2415,6 +2451,9 @@ ggml_tensor * llm_graph_context::build_moe_ffn(
141
+
142
+ cb(moe_out, "ffn_moe_out", il);
143
+
144
+ + // dump the native path too, so it can be diffed against the paged path
145
+ + moe_out = sdq::sdq_maybe_dump_tensor(ctx0, moe_out, il);
146
+ +
147
+ return moe_out;
148
+ }
149
+
150
+ @@ -2798,7 +2837,16 @@ ggml_tensor * llm_graph_context::build_attn_mha(
151
+ #else
152
+ // It's preferable to do the calculation as a matrix-matrix multiplication with n_tokens in dimension 1.
153
+ // The permutations are noops and only change how the tensor data is interpreted.
154
+ + // FLASH_ATTN_EXT produces F16, but the CPU backend cannot run MUL_MAT with a
155
+ + // quantized v_mla and F16 activations (widening requires vec_dot_type == F32),
156
+ + // and a quantized weight in the AMX buffer requires contiguous activations.
157
+ + if (cur->type != GGML_TYPE_F32 && v_mla->type != GGML_TYPE_F16 && v_mla->type != GGML_TYPE_F32) {
158
+ + cur = ggml_cast(ctx0, cur, GGML_TYPE_F32);
159
+ + }
160
+ cur = ggml_permute(ctx0, cur, 0, 2, 1, 3);
161
+ + if (!ggml_is_contiguous(cur)) {
162
+ + cur = ggml_cont(ctx0, cur);
163
+ + }
164
+ cur = ggml_mul_mat(ctx0, v_mla, cur);
165
+ cb(cur, "fattn_mla", il);
166
+ cur = ggml_permute(ctx0, cur, 0, 2, 1, 3);
167
+ diff --git a/src/llama-model.cpp b/src/llama-model.cpp
168
+ index 2e86914..fbf06cb 100644
169
+ --- a/src/llama-model.cpp
170
+ +++ b/src/llama-model.cpp
171
+ @@ -1,5 +1,7 @@
172
+ #include "llama-model.h"
173
+
174
+ +#include "sdq_pager.h"
175
+ +
176
+ #include "llama-arch.h"
177
+ #include "llama-ext.h"
178
+ #include "llama-hparams.h"
179
+ @@ -1834,7 +1836,9 @@ bool llama_model_base::load_tensors(llama_model_loader & ml) {
180
+ // per-tensor activation precision policy
181
+ prec_policy.load(ml, *this);
182
+
183
+ - ml.init_mappings(true, use_mlock ? &pimpl->mlock_mmaps : nullptr);
184
+ + // SDQ: the expert weights are read by the pager with its own pread calls, so the
185
+ + // loader must not MAP_POPULATE / WILLNEED the whole model file into page cache.
186
+ + ml.init_mappings(!sdq::sdq_enabled(), use_mlock ? &pimpl->mlock_mmaps : nullptr);
187
+ pimpl->mappings.reserve(ml.mappings.size());
188
+
189
+ // create the backend buffers
190
+ diff --git a/src/llama.cpp b/src/llama.cpp
191
+ index 9beb148..2633d96 100644
192
+ --- a/src/llama.cpp
193
+ +++ b/src/llama.cpp
194
+ @@ -11,6 +11,9 @@
195
+ #include "llama-model-saver.h"
196
+ #include "llama-model.h"
197
+
198
+ +// SDQ: expert pager(SSD -> RAM 分頁),見 sdq-ling/0001-sdq-ling.patch
199
+ +#include "sdq_pager.h"
200
+ +
201
+ #include "ggml.h"
202
+ #include "ggml-cpp.h"
203
+ #include "ggml-backend.h"
204
+ @@ -404,6 +407,27 @@ static struct llama_model * llama_model_load_from_file_impl(
205
+ // TODO: remove
206
+ ggml_time_init();
207
+
208
+ + // ---- SDQ: 把 MoE expert 權重交給 pager(SSD -> RAM arena 分頁)----
209
+ + // 放在模型載入的唯一入口,這樣 sdq-chat / llama-server / llama-cli 行為一致。
210
+ + // SDQ_PAGER=0 關閉,回到上游行為(A/B 對照用)
211
+ + // SDQ_RAM_BUDGET_MB RAM 總預算,arena 會配合縮減(沒有給就用實體 RAM 的一部分)
212
+ + // SDQ_ARENA_MB 直接指定 arena 大小
213
+ + // SDQ_RESERVE_MB KV/SSM state + compute buffer 的預留
214
+ + // SDQ_IO_THREADS I/O 執行緒數、SDQ_PREFETCH=0 關背景預取、SDQ_STATS_FILE 指標輸出
215
+ + {
216
+ + const char * sdq_off = getenv("SDQ_PAGER");
217
+ + if ((!sdq_off || strcmp(sdq_off, "0") != 0) && !params.vocab_only && !path_model.empty()) {
218
+ + if (sdq::sdq_init(path_model)) {
219
+ + // 沒有這一行,load_tensors() 會把整個模型檔要求預取進 page cache,
220
+ + // 正是分頁要避免的事(I/O 稽核見 results/io_audit.md)。
221
+ + sdq::sdq_set_enabled(true);
222
+ + sdq::sdq_install_buft_overrides(&params);
223
+ + // 權重是 pread 進 arena 的,不該再讓後端自己 offload 一次。
224
+ + params.n_gpu_layers = 0;
225
+ + }
226
+ + }
227
+ + }
228
+ +
229
+ if (!params.vocab_only && ggml_backend_reg_count() == 0) {
230
+ LLAMA_LOG_ERROR("%s: no backends are loaded. hint: use ggml_backend_load() or ggml_backend_load_all() to load a backend before calling this function\n", __func__);
231
+ return nullptr;
sdq-ling/__pycache__/verify_ling.cpython-312.pyc ADDED
Binary file (15.5 kB). View file
 
sdq-ling/audit_io.sh ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ # I/O honesty audit: prove the expert weights are read from the block device and
3
+ # are NOT served from (or hidden in) the page cache.
4
+ #
5
+ # ./audit_io.sh <model-first-shard.gguf> [sdq-chat] [ram-budget-mb] [n-predict]
6
+ #
7
+ # Two runs with identical settings:
8
+ # A) default pager uses O_DIRECT (bypasses the cache)
9
+ # B) SDQ_NO_DIRECT_IO=1 + --no-fadvise control: buffered reads kept in cache
10
+ #
11
+ # If this measurement were blind, A and B would look the same. They do not:
12
+ # B leaves gigabytes of the GGUF resident in the page cache, A does not.
13
+ set -u
14
+ MODEL="$(readlink -f "$1")"
15
+ BIN="$(readlink -f "${2:-./llama.cpp/build/bin/sdq-chat}")"
16
+ BUDGET="${3:-8192}"
17
+ NPRED="${4:-32}"
18
+ PROMPT="Explain in detail how paging mixture-of-experts weights from SSD works, \
19
+ and why the working set is much smaller than the weights."
20
+
21
+ cd "$(dirname "$0")/.."
22
+ SECOND="$(echo "$MODEL" | sed -E 's/-00001-of-/-00002-of-/')"
23
+ IFS=',' read -r -a SHARDS <<< "$MODEL,$SECOND"
24
+
25
+ if [ ! -x /tmp/pagecache_resident ]; then
26
+ echo "building the mincore helper ..."
27
+ g++ -O2 -o /tmp/pagecache_resident sdq-ling/pagecache_resident.cpp
28
+ fi
29
+
30
+ run() {
31
+ local label="$1"; shift
32
+ echo
33
+ echo "================================================================"
34
+ echo "[audit] $label"
35
+ echo "================================================================"
36
+ WATCH_INTERVAL=0.5 "$@" python3 sdq-ling/watch_io.py "$MODEL,$SECOND" \
37
+ "$BIN" -m "$MODEL" -p "$PROMPT" -n "$NPRED" -c 2048 -t 16 -b 128 \
38
+ --temp 0 --no-chat --no-interactive --ram-budget-mb "$BUDGET" \
39
+ 2>/dev/null | grep -E "^\[watch\]"
40
+ grep -E "全程 SSD 實際讀取|O_DIRECT 讀取模型檔|O_DIRECT 不可用" /tmp/watch_stderr.log | tail -3
41
+ grep -E "^assistant:" /tmp/watch_stdout.log | head -1
42
+ }
43
+
44
+ echo "[audit] model shards:"
45
+ /tmp/pagecache_resident "${SHARDS[@]}"
46
+
47
+ run "A) default (O_DIRECT)" env SDQ_RAM_BUDGET_MB="$BUDGET"
48
+ echo
49
+ echo "================================================================"
50
+ echo "[audit] B) control: buffered reads + page cache kept"
51
+ echo "================================================================"
52
+ WATCH_INTERVAL=0.5 SDQ_NO_DIRECT_IO=1 python3 sdq-ling/watch_io.py "$MODEL,$SECOND" \
53
+ "$BIN" -m "$MODEL" -p "$PROMPT" -n "$NPRED" -c 2048 -t 16 -b 128 \
54
+ --temp 0 --no-chat --no-interactive --no-fadvise --ram-budget-mb "$BUDGET" \
55
+ 2>/dev/null | grep -E "^\[watch\]"
56
+ grep -E "全程 SSD 實際讀取|O_DIRECT 不可用" /tmp/watch_stderr.log | tail -2
sdq-ling/ggml_deq.cpp ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ // Dump one ggml-dequantized row so the NumPy reference can be checked against
2
+ // ggml's own dequantizers. usage: ggml_deq <type> <file> <offset> <n>
3
+ #include "ggml.h"
4
+ #include "ggml-quants.h"
5
+
6
+ #include <cstdio>
7
+ #include <cstdlib>
8
+
9
+ int main(int argc, char ** argv) {
10
+ if (argc < 5) {
11
+ fprintf(stderr, "usage: %s <type> <file> <offset> <n>\n", argv[0]);
12
+ return 1;
13
+ }
14
+ const int tt = atoi(argv[1]);
15
+ const char * path = argv[2];
16
+ const long long off = atoll(argv[3]);
17
+ const int n = atoi(argv[4]);
18
+
19
+ FILE * f = fopen(path, "rb");
20
+ if (!f) { perror("open"); return 1; }
21
+ const size_t ts = ggml_row_size((ggml_type) tt, n);
22
+ void * buf = malloc(ts);
23
+ if (fseek(f, (long) off, SEEK_SET) != 0) { fprintf(stderr, "seek\n"); return 1; }
24
+ if (fread(buf, 1, ts, f) != ts) { fprintf(stderr, "short read\n"); return 1; }
25
+ fclose(f);
26
+
27
+ float * y = (float *) malloc(sizeof(float) * n);
28
+ switch (tt) {
29
+ case 12: dequantize_row_q4_K((const block_q4_K *) buf, y, n); break;
30
+ case 13: dequantize_row_q5_K((const block_q5_K *) buf, y, n); break;
31
+ case 14: dequantize_row_q6_K((const block_q6_K *) buf, y, n); break;
32
+ case 7: dequantize_row_q5_1((const block_q5_1 *) buf, y, n); break;
33
+ case 8: dequantize_row_q8_0((const block_q8_0 *) buf, y, n); break;
34
+ default: fprintf(stderr, "type %d not handled\n", tt); return 1;
35
+ }
36
+ fwrite(y, sizeof(float), n, stdout);
37
+ free(y);
38
+ free(buf);
39
+ return 0;
40
+ }
sdq-ling/make_io_audit.py ADDED
@@ -0,0 +1,148 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """Turn results/io_audit_raw.log (from sdq-ling/audit_io.sh) into io_audit.{json,md}."""
3
+ import json
4
+ import re
5
+ from pathlib import Path
6
+
7
+ R = Path('/root/app/results')
8
+ raw = (R / 'io_audit_raw.log').read_text()
9
+
10
+
11
+ def seg(marker):
12
+ i = raw.index(marker)
13
+ j = raw.find('[audit]', i + 10)
14
+ return raw[i:j if j > 0 else len(raw)]
15
+
16
+
17
+ def one(s, pat, cast=float):
18
+ m = re.search(pat, s)
19
+ return cast(m.group(1)) if m else None
20
+
21
+
22
+ def allv(s, pat):
23
+ return [float(m.group(1)) for m in re.finditer(pat, s)]
24
+
25
+
26
+ def parse(marker, mode, note):
27
+ s = seg(marker)
28
+ rb = allv(s, r'read_bytes=\s+([\d.]+) GiB')
29
+ res = allv(s, r'model_file_resident=\s*([\d.]+) GiB')
30
+ return {
31
+ 'mode': mode,
32
+ 'device_read_gib_max': max(rb) if rb else None,
33
+ 'device_read_gib_samples': len(rb),
34
+ 'model_file_resident_gib_peak': one(s, r'peak model_file_resident in page cache :\s+([\d.]+)'),
35
+ 'model_file_resident_gib_final': one(s, r'final model_file_resident\s+:\s+([\d.]+)'),
36
+ 'rss_anon_gib_peak': one(s, r'peak process RssAnon\s+:\s+([\d.]+)'),
37
+ 'rss_file_gib_peak': one(s, r'peak process RssFile\s+:\s+([\d.]+)'),
38
+ 'system_cached_delta_gib_peak': one(s, r'system Cached delta \(peak\)\s+:\s+([+\-][\d.]+)'),
39
+ 'sdq_chat_reported_ssd_gb': one(s, r'全程 SSD 實際讀取: ([\d.]+) GB'),
40
+ 'samples_tail': [f'read_bytes={b:.3f} GiB model_file_resident={r:.3f} GiB'
41
+ for b, r in list(zip(rb, res))[-6:]],
42
+ 'note': note,
43
+ }
44
+
45
+
46
+ doc = {
47
+ 'what': 'I/O honesty audit: are the expert weights really coming from the block device, '
48
+ 'or is the page cache quietly serving them?',
49
+ 'method': 'per-file page-cache eviction (posix_fadvise(DONTNEED) on both GGUF shards), '
50
+ 'then mincore(2) over the whole 71.72 GiB of shard data every 0.5 s while '
51
+ '/proc/<pid>/io read_bytes and /proc/<pid>/status RssAnon/RssFile are sampled.',
52
+ 'tools': ['sdq-ling/pagecache_resident.cpp', 'sdq-ling/watch_io.py', 'sdq-ling/audit_io.sh'],
53
+ 'settings': {'ram_budget_mb': 8192, 'ctx': 2048, 'ubatch': 128, 'threads': 16,
54
+ 'n_predict': 32, 'temp': 0},
55
+ 'results': {
56
+ 'A_default_o_direct': parse('A) default (O_DIRECT)', 'O_DIRECT (pager default)',
57
+ 'the model file stays at ~2.09 GiB resident, which is exactly the non-expert '
58
+ 'weights; the tens of GiB of expert reads leave no page-cache footprint'),
59
+ 'B_control_buffered_cache_kept': parse(
60
+ 'B) control: buffered reads + page cache kept',
61
+ 'SDQ_NO_DIRECT_IO=1 + --no-fadvise',
62
+ 'same reads, but the page cache retains them; residency tracks the read '
63
+ 'volume ~1:1, which is what a buffered implementation would cost'),
64
+ },
65
+ 'conclusion': [
66
+ 'The pager reads expert weights with O_DIRECT: read_bytes (real device I/O) grows '
67
+ 'while mincore residency of the GGUF stays flat.',
68
+ 'The only resident part of the model file is the 2.26 GiB of non-expert weights '
69
+ 'that llama.cpp keeps in RAM (1.54 GiB of it file-backed via mmap). Not one byte '
70
+ 'of the 69.46 GiB of expert weights reaches the page cache.',
71
+ 'The control run proves the measurement is sensitive: with buffered reads the same '
72
+ 'file accumulates tens of GiB of page cache.',
73
+ 'Memory numbers: an 8 GiB pager budget yields 5.83 GiB anonymous RSS '
74
+ '(3.54 GiB arena + ~2.3 GiB non-expert weights) + 1.73 GiB file-backed RSS '
75
+ '= 7.0 GiB total RSS.',
76
+ 'sdq-chat\'s "SSD read" line starts its counter after model load, so it excludes '
77
+ 'the ~3.9 GiB the loader itself reads; the watcher numbers above cover the whole '
78
+ 'process lifetime.',
79
+ ],
80
+ }
81
+ (R / 'io_audit.json').write_text(json.dumps(doc, ensure_ascii=False, indent=2) + '\n')
82
+
83
+ A = doc['results']['A_default_o_direct']
84
+ B = doc['results']['B_control_buffered_cache_kept']
85
+ md = f"""# I/O 稽核:expert 權重真的有從 block device 讀嗎?
86
+
87
+ 問題:如果 pager 其實是被 page cache 服務的,那「69 GiB 的權重留在 SSD、每次按需讀取」這句話就是假的。
88
+
89
+ ## 方法
90
+
91
+ 1. 用 `posix_fadvise(POSIX_FADV_DONTNEED)` 把兩個 GGUF shard(71.72 GiB)**逐檔案**從 page cache 驅逐
92
+ (不依賴 `/proc/sys/vm/drop_caches`,容器裡也做得到),並用 `mincore` 確認回到 0.000 GiB resident。
93
+ 2. 開跑 `sdq-chat`,每 0.5 s 抽樣:
94
+ * `/proc/<pid>/io` 的 `read_bytes` = 真正從 block device 讀了多少
95
+ * `mincore(2)` 掃過整個模型檔 = 有多少頁留在 page cache
96
+ * `/proc/<pid>/status` 的 `RssAnon` / `RssFile`
97
+ 3. 跑兩次,其他設定完全相同:
98
+ * **A(預設)**:pager 用 `O_DIRECT` 讀
99
+ * **B(對照組)**:`SDQ_NO_DIRECT_IO=1` + `--no-fadvise` = buffered 讀且**保留** page cache
100
+
101
+ B 的存在是為了證明這個量測**有鑑別力**:如果看不出差別,A 的數字也沒有意義。
102
+
103
+ ## 結果
104
+
105
+ | | A:預設(O_DIRECT) | B:對照組(buffered + 保留 cache) |
106
+ |---|---:|---:|
107
+ | 真正讀進來的量(read_bytes 峰值) | {A['device_read_gib_max']:.2f} GiB | {B['device_read_gib_max']:.2f} GiB |
108
+ | **模型檔留在 page cache 的峰值** | **{A['model_file_resident_gib_peak']:.3f} GiB** | **{B['model_file_resident_gib_peak']:.2f} GiB** |
109
+ | 程序 RssAnon 峰值 | {A['rss_anon_gib_peak']:.2f} GiB | {B['rss_anon_gib_peak']:.2f} GiB |
110
+ | 程序 RssFile 峰值 | {A['rss_file_gib_peak']:.2f} GiB | {B['rss_file_gib_peak']:.2f} GiB |
111
+ | 系統 Cached 增量(峰值) | {A['system_cached_delta_gib_peak']:+.2f} GiB | {B['system_cached_delta_gib_peak']:+.2f} GiB |
112
+ | `sdq-chat` 自報「SSD 實際讀取」 | {A['sdq_chat_reported_ssd_gb']:.2f} GB | {B['sdq_chat_reported_ssd_gb']:.2f} GB |
113
+
114
+ A 的時間序列(最後 6 個抽樣)——`read_bytes` 一路爬到
115
+ {A['device_read_gib_max']:.1f} GiB,而模型檔 resident 從 2.09 GiB 之後**完全不動**:
116
+
117
+ ```
118
+ {chr(10).join(A['samples_tail'])}
119
+ ```
120
+
121
+ ## 結論
122
+
123
+ 1. **pager 讀 expert 權重走 `O_DIRECT`**:`read_bytes`(真實裝置 I/O)上升時,
124
+ 模型檔的 page cache 用量沒有增加。
125
+ 2. 模型檔唯一留在 page cache 的部分是 **{A['model_file_resident_gib_peak']:.2f} GiB 的非 expert 權重**
126
+ (shard1 1.54 + shard2 0.55 GiB;非 expert 權重總共 2.26 GiB),
127
+ 那是 llama.cpp 本來就該留在 RAM 的東西,其中約 1.54 GiB 以 mmap file-backed 方式存在。
128
+ **69.46 GiB 的 expert 權重一個位元組都沒有進 page cache。**
129
+ 3. **對照組證明量測有鑑別力**:同樣的讀取量改成 buffered,模型檔會在 page cache 堆到
130
+ {B['model_file_resident_gib_peak']:.1f} GiB。如果量測是瞎的,A 和 B 會一樣。
131
+ 4. 對記憶體數字的影響:8 GiB 的 pager 預算實際是
132
+ **{A['rss_anon_gib_peak']:.2f} GiB 匿名 RSS(其中 arena 3.54 GiB、非 expert 權重 ~2.3 GiB)
133
+ + {A['rss_file_gib_peak']:.2f} GiB 檔案對映 RSS = 7.0 GiB 總 RSS**。
134
+ 5. 兩個數字的差異是可解釋的:`sdq-chat` 的「SSD 實際讀取」在**載入模型之後**才開始計,
135
+ 所以不含 loader 自己讀的 ~3.9 GiB;watcher 的數字涵蓋整個行程生命週期。
136
+
137
+ > 順帶修掉一個原本會誤導的報表 bug:`sdq-chat` 的記憶體摘要讀 `/proc/self/smaps_rollup` 的
138
+ > `RssFile:`,但本機核心的 `smaps_rollup` **沒有這個 key**(只有 `Anonymous:`),
139
+ > 於是長期都印「檔案對映 RSS 0 MB」。已改成讀 `/proc/self/status` 的
140
+ > `RssAnon:` / `RssFile:` / `VmRSS:`,並在記憶體實驗裡一併記錄三個數字。
141
+
142
+ 原始輸出:`results/io_audit_raw.log`;結構化資料:`results/io_audit.json`
143
+ 重跑:`./sdq-ling/audit_io.sh <model.gguf> <sdq-chat> 8192 32`
144
+ """
145
+ (R / 'io_audit.md').write_text(md)
146
+ print('wrote io_audit.json / io_audit.md')
147
+ print('A:', A['device_read_gib_max'], A['model_file_resident_gib_peak'])
148
+ print('B:', B['device_read_gib_max'], B['model_file_resident_gib_peak'])
sdq-ling/pagecache_resident.cpp ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ // Count how many pages of a file are resident in the page cache (mincore(2)).
2
+ // This is how we check "did the pager really bypass the page cache?": if the
3
+ // expert weights were served from cache, the model file would show up as resident.
4
+ //
5
+ // usage: pagecache_resident <file> [file...]
6
+ #include <fcntl.h>
7
+ #include <stdio.h>
8
+ #include <stdlib.h>
9
+ #include <string.h>
10
+ #include <sys/mman.h>
11
+ #include <fcntl.h>
12
+ #include <sys/stat.h>
13
+ #include <unistd.h>
14
+
15
+ int main(int argc, char ** argv) {
16
+ if (argc < 2) { fprintf(stderr, "usage: %s [--evict] <file> [file...]\n", argv[0]); return 1; }
17
+ int argi = 1;
18
+ int evict = 0;
19
+ if (strcmp(argv[1], "--evict") == 0) { evict = 1; argi = 2; }
20
+ const long page = sysconf(_SC_PAGESIZE);
21
+ unsigned long long tot_r = 0, tot_s = 0;
22
+
23
+ for (int i = argi; i < argc; i++) {
24
+ const char * path = argv[i];
25
+ int fd = open(path, O_RDONLY);
26
+ if (fd < 0) { perror(path); return 1; }
27
+ struct stat st;
28
+ fstat(fd, &st);
29
+ if (evict) {
30
+ // per-file eviction of clean page-cache pages (no root needed)
31
+ posix_fadvise(fd, 0, 0, POSIX_FADV_DONTNEED);
32
+ }
33
+ const size_t size = (size_t) st.st_size;
34
+ char * addr = (char *) mmap(NULL, size, PROT_READ, MAP_SHARED, fd, 0);
35
+ if (addr == MAP_FAILED) { perror("mmap"); return 1; }
36
+
37
+ const size_t npages = (size + page - 1) / page;
38
+ unsigned char * vec = (unsigned char *) malloc(npages);
39
+ if (mincore(addr, size, vec) != 0) { perror("mincore"); return 1; }
40
+ unsigned long long res = 0;
41
+ for (size_t p = 0; p < npages; p++) {
42
+ res += (vec[p] & 1);
43
+ }
44
+ free(vec);
45
+ munmap(addr, size);
46
+ close(fd);
47
+
48
+ const char * base = strrchr(path, '/');
49
+ base = base ? base + 1 : path;
50
+ printf("%-48s resident %8.3f GiB / %8.3f GiB (%5.1f %%)\n",
51
+ base, res * (double) page / 1073741824.0, size / 1073741824.0,
52
+ 100.0 * (double) res * (double) page / (double) size);
53
+ tot_r += res * (unsigned long long) page;
54
+ tot_s += size;
55
+ }
56
+ if (argc > argi + 1) {
57
+ printf("%-48s resident %8.3f GiB / %8.3f GiB\n", "TOTAL",
58
+ tot_r / 1073741824.0, tot_s / 1073741824.0);
59
+ }
60
+ return 0;
61
+ }
sdq-ling/sdq_buft.cpp ADDED
@@ -0,0 +1,240 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ // SDQ expert buffer type
2
+ //
3
+ // 這是一個「假」buffer type:它宣稱接受所有張量,但實際上
4
+ // * 不配置任何 RAM(alloc_buffer 回傳 0 大小的 buffer)
5
+ // * set_tensor 忽略傳入的資料(資料來自 llama.cpp 的 mmap,我們不碰)
6
+ // * 真正的資料由 sdq::pager 從檔案 pread 到自己的 arena
7
+ //
8
+ // 用途:讓 llama.cpp 把 18.2 GiB 的 MoE 專家權重「掛在模型上」卻不放進 RAM,
9
+ // 改由分頁器管理 SSD → RAM → CPU 的三層資料流。
10
+
11
+ #include "sdq_pager.h"
12
+
13
+ #include "ggml-backend-impl.h"
14
+ #include "llama-mmap.h"
15
+
16
+ #include <cstdio>
17
+ #include <string>
18
+ #include <cstdlib>
19
+ #include <cstring>
20
+ #include <algorithm>
21
+ #include <mutex>
22
+ #include <unordered_map>
23
+ #include <vector>
24
+
25
+ namespace sdq {
26
+
27
+ // 給 tensor->data 的假位址。每個 expert 權重張量分到一小塊(預設 4 KB),
28
+ // 指標必須有效(ggml 會做指標運算),但內容永遠不會被讀。
29
+ struct expert_buffer {
30
+ std::vector<uint8_t> scratch;
31
+ };
32
+
33
+ static bool g_enabled = false;
34
+
35
+ void sdq_set_enabled(bool on) { g_enabled = on; }
36
+ bool sdq_enabled() { return g_enabled; }
37
+
38
+ static config g_cfg;
39
+ void sdq_set_config(const config & c) { g_cfg = c; }
40
+ const config & sdq_get_config() { return g_cfg; }
41
+
42
+ // ---------------------------------------------------------------- buffer iface
43
+
44
+ // ggml_backend_buffer_i 的欄位依賴下面的實作,先給宣告
45
+ struct ggml_backend_buffer_i sdq_buffer_iface();
46
+
47
+ static const char * sdq_buft_get_name(ggml_backend_buffer_type_t buft) {
48
+ (void) buft;
49
+ return "SDQ_ExpertSSD";
50
+ }
51
+
52
+ static size_t sdq_buft_get_alignment(ggml_backend_buffer_type_t buft) {
53
+ (void) buft;
54
+ return 4096;
55
+ }
56
+
57
+ static size_t sdq_buft_get_max_size(ggml_backend_buffer_type_t buft) {
58
+ (void) buft;
59
+ return SIZE_MAX;
60
+ }
61
+
62
+ // 每個 expert 權重張量只回報 4 KB:18.2 GiB 的權重「不會」被載入 RAM 的關鍵。
63
+ static size_t sdq_buft_get_alloc_size(ggml_backend_buffer_type_t buft, const ggml_tensor * tensor) {
64
+ (void) buft;
65
+ (void) tensor;
66
+ return 4096;
67
+ }
68
+
69
+ static bool sdq_buft_is_host(ggml_backend_buffer_type_t buft) {
70
+ (void) buft;
71
+ return false;
72
+ }
73
+
74
+ static ggml_backend_buffer_t sdq_buft_alloc_buffer(ggml_backend_buffer_type_t buft, size_t size) {
75
+ // llama.cpp 會用 get_alloc_size() 計算總大小後呼叫本函式;
76
+ // 我們只保留真正需要的假位址空間(每個張量 4 KB)。
77
+ const size_t bytes = size < 4096 ? 4096 : size;
78
+ auto * buf = new expert_buffer();
79
+ buf->scratch.assign(bytes, 0);
80
+ auto * b = new ggml_backend_buffer{
81
+ /* .iface = */ sdq_buffer_iface(),
82
+ /* .buft = */ buft,
83
+ /* .context = */ buf,
84
+ /* .size = */ bytes,
85
+ /* .usage = */ GGML_BACKEND_BUFFER_USAGE_WEIGHTS,
86
+ };
87
+ return b;
88
+ }
89
+
90
+ // 一次配置整個 context 的張量:每個張量 4 KB 假位址,實際權重 0 RAM。
91
+ static ggml_backend_buffer_t sdq_buft_alloc_buffer_n(ggml_backend_buffer_type_t buft,
92
+ ggml_tensor ** tensors, int n_tensors) {
93
+ const size_t slot = 4096;
94
+ const size_t bytes = slot * (size_t) std::max(n_tensors, 1);
95
+ ggml_backend_buffer_t buffer = sdq_buft_alloc_buffer(buft, bytes);
96
+ if (!buffer) {
97
+ return nullptr;
98
+ }
99
+ uint8_t * base = (uint8_t *) buffer->iface.get_base(buffer);
100
+ size_t off = 0;
101
+ for (int i = 0; i < n_tensors; i++) {
102
+ ggml_tensor * t = tensors[i];
103
+ t->data = base + off;
104
+ t->buffer = buffer;
105
+ off += slot;
106
+ buffer->iface.init_tensor(buffer, t);
107
+ }
108
+ return buffer;
109
+ }
110
+
111
+ static void sdq_buft_free_buffer(ggml_backend_buffer_t buffer) {
112
+ // 只釋放自己的資料;ggml_backend_buffer_free() 會負責 delete buffer 本體,
113
+ // 這裡再 delete 一次會造成 double free。
114
+ delete (expert_buffer *) buffer->context;
115
+ buffer->context = nullptr;
116
+ }
117
+
118
+ static void * sdq_buft_get_base(ggml_backend_buffer_t buffer) {
119
+ return ((expert_buffer *) buffer->context)->scratch.data();
120
+ }
121
+
122
+ static ggml_status sdq_buft_init_tensor(ggml_backend_buffer_t buffer, ggml_tensor * tensor) {
123
+ (void) buffer;
124
+ (void) tensor;
125
+ return GGML_STATUS_SUCCESS;
126
+ }
127
+
128
+ static void sdq_buft_memset_tensor(ggml_backend_buffer_t buffer, ggml_tensor * tensor,
129
+ uint8_t value, size_t offset, size_t size) {
130
+ (void) buffer; (void) tensor; (void) value; (void) offset; (void) size;
131
+ }
132
+
133
+ static void sdq_buft_set_tensor(ggml_backend_buffer_t buffer, ggml_tensor * tensor,
134
+ const void * data, size_t offset, size_t size) {
135
+ // 關鍵:忽略。llama.cpp 會把 mmap 的資料交給我們,這裡一個位元組都不落地,
136
+ // 專家權利永遠不會佔用 RAM。
137
+ (void) buffer; (void) tensor; (void) data; (void) offset; (void) size;
138
+ }
139
+
140
+ static void sdq_buft_get_tensor(ggml_backend_buffer_t buffer, const ggml_tensor * tensor,
141
+ void * data, size_t offset, size_t size) {
142
+ (void) buffer; (void) tensor; (void) offset;
143
+ memset(data, 0, size);
144
+ }
145
+
146
+ static bool sdq_buft_cpy_tensor(ggml_backend_buffer_t buffer, const ggml_tensor * src, ggml_tensor * dst) {
147
+ (void) buffer; (void) src; (void) dst;
148
+ return true;
149
+ }
150
+
151
+ static void sdq_buft_clear(ggml_backend_buffer_t buffer, uint8_t value) {
152
+ (void) buffer; (void) value;
153
+ }
154
+
155
+ struct ggml_backend_buffer_i sdq_buffer_iface() {
156
+ static ggml_backend_buffer_i i = [] {
157
+ ggml_backend_buffer_i x {};
158
+ x.free_buffer = sdq_buft_free_buffer;
159
+ x.get_base = sdq_buft_get_base;
160
+ x.init_tensor = sdq_buft_init_tensor;
161
+ x.memset_tensor = sdq_buft_memset_tensor;
162
+ x.set_tensor = sdq_buft_set_tensor;
163
+ x.get_tensor = sdq_buft_get_tensor;
164
+ x.cpy_tensor = sdq_buft_cpy_tensor;
165
+ x.clear = sdq_buft_clear;
166
+ return x;
167
+ }();
168
+ return i;
169
+ }
170
+
171
+ ggml_backend_buffer_type_t sdq_expert_buft() {
172
+ static ggml_backend_buffer_type_t buft = [] {
173
+ return new ggml_backend_buffer_type{
174
+ /* .iface = */ {
175
+ /* .get_name = */ sdq_buft_get_name,
176
+ /* .alloc_buffer = */ sdq_buft_alloc_buffer,
177
+ /* .alloc_buffer_n = */ sdq_buft_alloc_buffer_n,
178
+ /* .get_alignment = */ sdq_buft_get_alignment,
179
+ /* .get_max_size = */ sdq_buft_get_max_size,
180
+ /* .get_alloc_size = */ sdq_buft_get_alloc_size,
181
+ /* .get_alloc_size_n = */ nullptr,
182
+ /* .is_host = */ sdq_buft_is_host,
183
+ },
184
+ /* .device = */ nullptr,
185
+ /* .context = */ nullptr,
186
+ };
187
+ }();
188
+ return buft;
189
+ }
190
+
191
+ // llama.cpp 端需要的介面宣告
192
+ void sdq_install_buft_overrides(llama_model_params * params);
193
+
194
+ // SDQ_BUFT_KEEP 保留開關:std::regex 的負向先行斷言在 llama.cpp 的比對方式下
195
+ // 不可靠(會讓整個 override 失效、權重被載進 RAM),所以先不開放這個 knob。
196
+ static std::string sdq_expert_pattern() {
197
+ return "blk\\.[0-9]+\\.ffn_(gate|up|down|gate_up)_exps\\.weight";
198
+ }
199
+
200
+ // 這兩個 static 只能用「初始化一次」的方式寫:libstdc++ 的 std::string move-assign
201
+ // 會把舊的 buffer 交給右邊的 temporary,temporary 析構時就 free 掉。若改成每次呼叫
202
+ // pattern = sdq_expert_pattern(),先前存進 override 陣列的 .pattern 會變成懸空指標;
203
+ // 讀到空字串時 std::regex("") 會匹配「任何」張量名,於是整個模型(含 token_embd)
204
+ // 都被塞進假 buffer,server 直接在 sched_reserve 裡 ggml_abort。
205
+ static const llama_model_tensor_buft_override * sdq_overrides() {
206
+ static const std::string pattern = sdq_expert_pattern();
207
+ static const llama_model_tensor_buft_override ov[2] = {
208
+ { pattern.c_str(), sdq_expert_buft() },
209
+ { nullptr, nullptr },
210
+ };
211
+ return ov;
212
+ }
213
+
214
+ void sdq_install_buft_overrides(llama_model_params * params) {
215
+ if (!params) {
216
+ return;
217
+ }
218
+ if (params->tensor_buft_overrides == nullptr) {
219
+ params->tensor_buft_overrides = sdq_overrides();
220
+ } else {
221
+ // 使用者已有 override:把手動指定的順序原樣保留,附加在尾端
222
+ size_t n = 0;
223
+ while (params->tensor_buft_overrides[n].pattern != nullptr) {
224
+ n++;
225
+ }
226
+ static std::vector<llama_model_tensor_buft_override> merged;
227
+ merged.clear();
228
+ for (size_t i = 0; i < n; i++) {
229
+ merged.push_back(params->tensor_buft_overrides[i]);
230
+ }
231
+ const llama_model_tensor_buft_override * sdq_ov = sdq_overrides();
232
+ for (size_t i = 0; sdq_ov[i].pattern != nullptr; i++) {
233
+ merged.push_back(sdq_ov[i]);
234
+ }
235
+ merged.push_back({ nullptr, nullptr });
236
+ params->tensor_buft_overrides = merged.data();
237
+ }
238
+ }
239
+
240
+ } // namespace sdq
sdq-ling/sdq_cli.cpp ADDED
@@ -0,0 +1,844 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ // SDQ 聊天 / benchmark 主程式
2
+ //
3
+ // sdq-chat -m <model.gguf> [-c ctx] [-t threads] [-n n_predict] [-p prompt] ...
4
+ //
5
+ // 這個工具存在的理由:官方 llama-cli 不會安裝我們的 tensor buffer override,
6
+ // 而且需要我們在載入模型前後插入 pager 的初始化/收尾。用最少的相依度
7
+ // (只鏈 llama + ggml)自己寫一支,行為完全可控。
8
+
9
+ #include "llama.h"
10
+ #include "sdq_pager.h"
11
+
12
+ #include <algorithm>
13
+ #include <chrono>
14
+ #include <cinttypes>
15
+ #include <cmath>
16
+ #include <cstdio>
17
+ #include <cstdlib>
18
+ #include <cstring>
19
+ #include <functional>
20
+ #include <thread>
21
+ #include <iostream>
22
+ #include <string>
23
+ #include <vector>
24
+
25
+ using namespace sdq;
26
+
27
+ namespace {
28
+
29
+ struct options {
30
+ std::string model;
31
+ std::string prompt;
32
+ std::string system = "You are a helpful assistant. Answer in the user's language.";
33
+ std::string stats_file;
34
+ int32_t n_ctx = 4096; // 驗證範圍:4K(理論支援 1M)
35
+ int32_t n_predict = 256;
36
+ int32_t n_threads = 0;
37
+ int32_t n_ubatch = 0;
38
+ int32_t n_keep = 0;
39
+ float temp = 0.7f;
40
+ float top_p = 0.95f;
41
+ int32_t top_k = 20;
42
+ bool chat = true;
43
+ bool interactive = true;
44
+ bool bench = false;
45
+ bool no_prefetch = false;
46
+ bool dump_logits = false;
47
+ std::string logits_out;
48
+ std::string selftest;
49
+ std::string span_test;
50
+ std::string dump_tensors; // ggml eval callback:把指定張量的資料存檔(A/B 比對用)
51
+ bool no_pager = false; // A/B 測試用:不啟用分頁,走上游 mul_mat_id(整個模型在 RAM)
52
+ int32_t arena_mb = 0;
53
+ int32_t ram_budget_mb = 0;
54
+ bool fadvise = true;
55
+ };
56
+
57
+ double now_s() {
58
+ static auto t0 = std::chrono::steady_clock::now();
59
+ return std::chrono::duration<double>(std::chrono::steady_clock::now() - t0).count();
60
+ }
61
+
62
+ size_t rss_mb() { return process_rss_bytes() / (1024 * 1024); }
63
+
64
+ // 從 /proc/self/status 讀 VmHWM(峰值 RSS)/ VmSwap(這個行程真的用了多少 swap)
65
+ size_t proc_status_kb(const char * key) {
66
+ FILE * f = fopen("/proc/self/status", "r");
67
+ if (!f) return 0;
68
+ char line[256];
69
+ const size_t klen = strlen(key);
70
+ size_t val = 0;
71
+ while (fgets(line, sizeof(line), f)) {
72
+ if (strncmp(line, key, klen) == 0) {
73
+ val = (size_t) strtoull(line + klen, nullptr, 10);
74
+ break;
75
+ }
76
+ }
77
+ fclose(f);
78
+ return val;
79
+ }
80
+
81
+ size_t peak_rss_mb() { return proc_status_kb("VmHWM:") / 1024; }
82
+ size_t swap_mb() { return proc_status_kb("VmSwap:") / 1024; }
83
+
84
+ // /proc/self/smaps_rollup:把 RSS 拆成「檔案對映」與「匿名」兩塊。
85
+ // 這很重要:模型檔是 mmap 進來的,核心預讀會讓「檔案對映頁」也算進 RSS,
86
+ // 報告時必須分開看,否則會誤以為 RAM 用量比實際少(或多)。
87
+ size_t smaps_kb(const char * key) {
88
+ FILE * f = fopen("/proc/self/smaps_rollup", "r");
89
+ if (!f) return 0;
90
+ char line[256];
91
+ const size_t klen = strlen(key);
92
+ size_t val = 0;
93
+ while (fgets(line, sizeof(line), f)) {
94
+ if (strncmp(line, key, klen) == 0) {
95
+ val = (size_t) strtoull(line + klen, nullptr, 10);
96
+ break;
97
+ }
98
+ }
99
+ fclose(f);
100
+ return val;
101
+ }
102
+
103
+ // smaps_rollup 並不是每個核心都有 "RssFile:"(本機的核心就沒有,只有 "Anonymous:"),
104
+ // 讀不到就會印出「檔案對映 RSS 0 MB」這種假的數字。
105
+ // RssAnon / RssFile / VmRSS 在 /proc/self/status 一定有,而且正是同一組語意。
106
+ size_t status_kb(const char * key) {
107
+ FILE * f = fopen("/proc/self/status", "r");
108
+ if (!f) return 0;
109
+ char line[256];
110
+ const size_t klen = strlen(key);
111
+ size_t val = 0;
112
+ while (fgets(line, sizeof(line), f)) {
113
+ if (strncmp(line, key, klen) == 0) {
114
+ val = (size_t) strtoull(line + klen, nullptr, 10);
115
+ break;
116
+ }
117
+ }
118
+ fclose(f);
119
+ return val;
120
+ }
121
+
122
+ void print_mem_report(const char * when) {
123
+ fprintf(stderr,
124
+ "[sdq] 記憶體(%s):匿名 RSS %zu MB / 檔案對映 RSS %zu MB / 總 RSS %zu MB / 峰值 %zu MB / swap %zu MB\n",
125
+ when, status_kb("RssAnon:") / 1024, status_kb("RssFile:") / 1024,
126
+ status_kb("VmRSS:") / 1024, peak_rss_mb(), swap_mb());
127
+ }
128
+
129
+ // 讀 /proc/self/io:從 block device 實際讀了多少(SSD 讀取)
130
+ uint64_t read_bytes_io() {
131
+ FILE * f = fopen("/proc/self/io", "r");
132
+ if (!f) return 0;
133
+ char line[256];
134
+ uint64_t v = 0;
135
+ while (fgets(line, sizeof(line), f)) {
136
+ if (strncmp(line, "read_bytes:", 11) == 0) {
137
+ v = strtoull(line + 11, nullptr, 10);
138
+ }
139
+ }
140
+ fclose(f);
141
+ return v;
142
+ }
143
+
144
+ struct chat_state {
145
+ std::vector<llama_chat_message> msgs;
146
+ };
147
+
148
+ std::string build_prompt(const options & opt, const chat_state & st) {
149
+ if (!opt.chat) {
150
+ return opt.prompt;
151
+ }
152
+ const char * tmpl = llama_model_chat_template(nullptr, nullptr);
153
+ // 沒有 model 指標時用內建模板;��面 main() 會用實際模型的模板重做
154
+ (void) tmpl;
155
+ size_t n_chars = 0;
156
+ for (const auto & m : st.msgs) {
157
+ n_chars += strlen(m.role) + strlen(m.content) + 32;
158
+ }
159
+ std::vector<char> buf(2 * n_chars + 1024);
160
+ const char * t = st.msgs.empty() ? nullptr : nullptr;
161
+ (void) t;
162
+ int32_t len = llama_chat_apply_template(tmpl, st.msgs.data(), st.msgs.size(), true,
163
+ buf.data(), (int32_t) buf.size());
164
+ if (len > (int32_t) buf.size()) {
165
+ buf.resize(len + 1);
166
+ llama_chat_apply_template(tmpl, st.msgs.data(), st.msgs.size(), true,
167
+ buf.data(), (int32_t) buf.size());
168
+ }
169
+ return std::string(buf.data());
170
+ }
171
+
172
+ } // namespace
173
+
174
+ #include <execinfo.h>
175
+ #include <signal.h>
176
+ #include <unistd.h>
177
+ static void sdq_segv(int sig) {
178
+ void * bt[64];
179
+ const int n = backtrace(bt, 64);
180
+ fprintf(stderr, "\n*** sdq: signal %d, backtrace:\n", sig);
181
+ backtrace_symbols_fd(bt, n, 2);
182
+ _exit(9);
183
+ }
184
+
185
+ // eval callback:把感興趣的張量(用 SDQ_DUMP_NAMES 指定)寫進檔案
186
+ struct dump_state {
187
+ FILE * fp = nullptr;
188
+ uint64_t step = 0;
189
+ std::vector<std::string> names;
190
+ };
191
+
192
+ static dump_state g_dump;
193
+
194
+ static bool sdq_eval_callback(ggml_tensor * t, bool ask, void * user_data) {
195
+ (void) user_data;
196
+ auto & ds = g_dump;
197
+ if (ask) {
198
+ // 除錯模式:每個節點都要資料 → 會強制逐節點同步(慢,但能逐節點比對)
199
+ return ds.fp != nullptr;
200
+ }
201
+ if (!ds.fp) {
202
+ return false;
203
+ }
204
+ const char * nm = ggml_get_name(t);
205
+ const size_t n_len = nm ? strlen(nm) : 0;
206
+ bool want = ds.names.empty();
207
+ for (const auto & n : ds.names) {
208
+ if (nm && strstr(nm, n.c_str())) {
209
+ want = true;
210
+ break;
211
+ }
212
+ }
213
+ const int32_t ne[4] = {(int32_t) t->ne[0], (int32_t) t->ne[1], (int32_t) t->ne[2], (int32_t) t->ne[3]};
214
+ const int32_t type = (int32_t) t->type;
215
+ fwrite(&n_len, sizeof(size_t), 1, ds.fp);
216
+ if (n_len) {
217
+ fwrite(nm, 1, n_len, ds.fp);
218
+ }
219
+ fwrite(&ds.step, sizeof(uint64_t), 1, ds.fp);
220
+ fwrite(ne, sizeof(int32_t), 4, ds.fp);
221
+ fwrite(&type, sizeof(int32_t), 1, ds.fp);
222
+
223
+ // 永遠寫「摘要」(前 8 個值 + 平方和 + 最大值),這樣即使沒有名字也能 A/B 找出第一個分歧點
224
+ float summary[10] = {0};
225
+ if (t->type == GGML_TYPE_F32 && t->data) {
226
+ const size_t n = (size_t) ne[0] * ne[1] * ne[2] * ne[3];
227
+ const float * p = (const float *) t->data;
228
+ double ss = 0;
229
+ float mx = 0;
230
+ for (size_t i = 0; i < n; i++) {
231
+ ss += (double) p[i] * p[i];
232
+ if (std::fabs(p[i]) > std::fabs(mx)) {
233
+ mx = p[i];
234
+ }
235
+ }
236
+ for (int i = 0; i < 8 && (size_t) i < n; i++) {
237
+ summary[i] = p[i];
238
+ }
239
+ summary[8] = (float) std::sqrt(ss);
240
+ summary[9] = mx;
241
+ fwrite(summary, sizeof(float), 10, ds.fp);
242
+ if (want) {
243
+ fwrite(p, sizeof(float), n, ds.fp);
244
+ }
245
+ } else {
246
+ fwrite(summary, sizeof(float), 10, ds.fp);
247
+ }
248
+ ds.step++;
249
+ return true; // 這個節點的資料我們已經讀過了,標記成已完成
250
+ }
251
+
252
+ int main(int argc, char ** argv) {
253
+ signal(SIGSEGV, sdq_segv);
254
+ signal(SIGBUS, sdq_segv);
255
+ signal(SIGABRT, sdq_segv);
256
+ options opt;
257
+ const char * env_model = getenv("SDQ_MODEL_PATH");
258
+ if (env_model) {
259
+ opt.model = env_model;
260
+ }
261
+
262
+ for (int i = 1; i < argc; i++) {
263
+ std::string a = argv[i];
264
+ auto next = [&](const char * what) -> std::string {
265
+ if (i + 1 >= argc) {
266
+ fprintf(stderr, "%s 需要參數\n", what);
267
+ exit(1);
268
+ }
269
+ return argv[++i];
270
+ };
271
+ if (a == "-m" || a == "--model") opt.model = next("-m");
272
+ else if (a == "-p" || a == "--prompt") opt.prompt = next("-p");
273
+ else if (a == "--sys") opt.system = next("--sys");
274
+ else if (a == "-c" || a == "--ctx") opt.n_ctx = atoi(next("-c").c_str());
275
+ else if (a == "-n") opt.n_predict = atoi(next("-n").c_str());
276
+ else if (a == "-t") opt.n_threads = atoi(next("-t").c_str());
277
+ else if (a == "-b" || a == "--ubatch") opt.n_ubatch = atoi(next("-b").c_str());
278
+ else if (a == "--temp") opt.temp = atof(next("--temp").c_str());
279
+ else if (a == "--top-p") opt.top_p = atof(next("--top-p").c_str());
280
+ else if (a == "--top-k") opt.top_k = atoi(next("--top-k").c_str());
281
+ else if (a == "--no-chat") opt.chat = false;
282
+ else if (a == "--no-interactive") opt.interactive = false;
283
+ else if (a == "--bench") opt.bench = true;
284
+ else if (a == "--no-prefetch") opt.no_prefetch = true;
285
+ else if (a == "--arena-mb") opt.arena_mb = atoi(next("--arena-mb").c_str());
286
+ else if (a == "--ram-budget-mb") opt.ram_budget_mb = atoi(next("--ram-budget-mb").c_str());
287
+ else if (a == "--no-fadvise") opt.fadvise = false;
288
+ else if (a == "--stats-file") opt.stats_file = next("--stats-file");
289
+ else if (a == "--dump-logits") opt.dump_logits = true;
290
+ else if (a == "--logits-out") opt.logits_out = next("--logits-out");
291
+ else if (a == "--selftest") opt.selftest = next("--selftest");
292
+ else if (a == "--dump-tensors") opt.dump_tensors = next("--dump-tensors");
293
+ else if (a == "--no-pager") opt.no_pager = true;
294
+ else if (a == "--span-test") opt.span_test = next("--span-test");
295
+ else if (a == "-h" || a == "--help") {
296
+ printf(
297
+ "用法:sdq-chat -m <model.gguf> [選項]\n"
298
+ " -c, --ctx N 上下文長度(預設 4096;模型原生 262144)\n"
299
+ " -n N 要生成的 token 數\n"
300
+ " -t N CPU 執行緒數\n"
301
+ " -b N ubatch 大小\n"
302
+ " -p, --prompt TEXT 提示詞\n"
303
+ " --sys TEXT system 訊息\n"
304
+ " --no-chat 不要套 chat template(純 completion)\n"
305
+ " --no-interactive 跑完提示詞就結束\n"
306
+ " --bench benchmark 模式(輸出 JSON 到 stats 檔)\n"
307
+ " --arena-mb N expert arena 大小(MB)\n"
308
+ " --ram-budget-mb N RAM 總預算(MB),arena 會配合縮減\n"
309
+ " --no-prefetch 關閉背景預取\n"
310
+ " --no-fadvise 讀完不丟 kernel page cache(會讓 RAM 帳目失真)\n"
311
+ " --stats-file PATH 指標 JSONL 輸出\n"
312
+ " --dump-logits 把每個 token 的 logits 存檔(驗證用)\n"
313
+ " --logits-out PATH logits 輸出路徑\n");
314
+ return 0;
315
+ } else {
316
+ fprintf(stderr, "未知參數:%s(用 --help 看用法)\n", a.c_str());
317
+ return 1;
318
+ }
319
+ }
320
+
321
+ if (opt.model.empty()) {
322
+ fprintf(stderr, "缺少模型檔(-m 或 SDQ_MODEL_PATH)\n");
323
+ return 1;
324
+ }
325
+
326
+ // ---- 環境變數(pager 讀)
327
+ setenv("SDQ_MODEL_PATH", opt.model.c_str(), 1);
328
+ // I/O 讀取執行緒數:預設 3 × 計算執行緒(上限 48)。
329
+ // 必須在這裡設定 —— pager 的 I/O 池是在「載入模型」時啟動的,那已經比
330
+ // context 建立早了;放到後面設定會讓池子只拿到下限 4 條執行緒。
331
+ // 量測(repeat=3,中位數):序列讀 3.79 → 12 條 5.84 → 24 條 6.68 tok/s。
332
+ if (!getenv("SDQ_IO_THREADS")) {
333
+ const int hw_io = (int) std::max(1u, std::thread::hardware_concurrency());
334
+ const int dflt_io = hw_io < 8 ? hw_io : 8;
335
+ const int n_io = opt.n_threads > 0 ? opt.n_threads : dflt_io;
336
+ setenv("SDQ_IO_THREADS", std::to_string(n_io * 3).c_str(), 1);
337
+ }
338
+ // 只有明確指定時才設定,否則保留呼叫者從環境變數傳入的值
339
+ // (舊版無條件 setenv 會讓 SDQ_PREFETCH=0 被悄悄蓋回 1,除錯時極易誤判)
340
+ if (opt.no_prefetch) {
341
+ setenv("SDQ_PREFETCH", "0", 1);
342
+ }
343
+ setenv("SDQ_FADV_DONTNEED", opt.fadvise ? "1" : "0", 1);
344
+ if (opt.arena_mb > 0) {
345
+ setenv("SDQ_ARENA_MB", std::to_string(opt.arena_mb).c_str(), 1);
346
+ }
347
+ if (opt.ram_budget_mb > 0) {
348
+ setenv("SDQ_RAM_BUDGET_MB", std::to_string(opt.ram_budget_mb).c_str(), 1);
349
+ }
350
+ if (!opt.stats_file.empty()) {
351
+ setenv("SDQ_STATS_FILE", opt.stats_file.c_str(), 1);
352
+ }
353
+
354
+ llama_backend_init();
355
+ llama_numa_init(GGML_NUMA_STRATEGY_DISABLED);
356
+
357
+ // ---- 模型載入:expert 權重交給 pager 的 buffer type(0 RAM)
358
+ llama_model_params mparams = llama_model_default_params();
359
+ mparams.n_gpu_layers = 0;
360
+ mparams.load_mode = LLAMA_LOAD_MODE_MMAP;
361
+ if (!opt.no_pager) {
362
+ sdq_set_enabled(true);
363
+ sdq_install_buft_overrides(&mparams);
364
+ } else {
365
+ printf("[sdq] --no-pager:對照模式,專家權重全部留在 RAM(需要 %s RAM)\n", "22GB");
366
+ }
367
+
368
+ printf("[sdq] 載入模型:%s\n", opt.model.c_str());
369
+ const double t_load0 = now_s();
370
+ llama_model * model = llama_model_load_from_file(opt.model.c_str(), mparams);
371
+ if (!model) {
372
+ fprintf(stderr, "[sdq] 模型載入失敗\n");
373
+ return 1;
374
+ }
375
+ const double t_load = now_s() - t_load0;
376
+
377
+ // ---- pager 初始化(讀 GGUF 檔頭、建頁面表、配置 arena)
378
+ auto & P = pager::instance();
379
+ if (!opt.no_pager && !sdq_init(opt.model)) {
380
+ fprintf(stderr, "[sdq] pager 初始化失敗\n");
381
+ return 1;
382
+ }
383
+
384
+ // ---- 張量 dump(A/B 比對用):在 op 真的算完的那一刻把資��寫出來
385
+ FILE * tensor_fp = nullptr;
386
+ dump_state dump_state_;
387
+ if (!opt.dump_tensors.empty()) {
388
+ tensor_fp = fopen(opt.dump_tensors.c_str(), "wb");
389
+ dump_state_.fp = tensor_fp;
390
+ if (const char * names = getenv("SDQ_DUMP_NAMES")) {
391
+ std::string s = names;
392
+ size_t pos = 0;
393
+ while (pos <= s.size()) {
394
+ size_t c = s.find(',', pos);
395
+ if (c == std::string::npos) c = s.size();
396
+ dump_state_.names.push_back(s.substr(pos, c - pos));
397
+ pos = c + 1;
398
+ }
399
+ }
400
+ g_dump = dump_state_;
401
+ }
402
+
403
+ // ---- context
404
+ llama_context_params cparams = llama_context_default_params();
405
+ cparams.n_ctx = (uint32_t) opt.n_ctx;
406
+ cparams.n_batch = (uint32_t) opt.n_ctx;
407
+ // ubatch 預設 128(不是 512)。理由是「8 GB 預算要真的裝得下」:
408
+ // llama.cpp 的 CPU compute buffer 是按 n_ubatch 線性成長的,實測
409
+ // ubatch=512 → 502 MiB、256 → 251 MiB、128 → 125.5 MiB
410
+ // 而 compute buffer 是在 pager 算完 arena 大小之後才配置的,所以它不算在
411
+ // arena 的扣減裡。實測在 RLIMIT_DATA 8 GB 下:
412
+ // ubatch=512 → 提示詞超過 ~256 token 直接 std::bad_alloc(崩潰)
413
+ // ubatch=256 → 同一個提示詞一樣崩潰
414
+ // ubatch=128 → 正常,681 token 的 prefill 跑到 28.8 tok/s
415
+ // 而且 ubatch=128 還更快(docs/04 實測 +7%),因為 prefill 分更多批、
416
+ // 每批的 expert 聯集比較小、命中率比較高。
417
+ cparams.n_ubatch = opt.n_ubatch > 0 ? (uint32_t) opt.n_ubatch : (uint32_t) std::min(opt.n_ctx, 128);
418
+ // 執行緒預設值:不要直接用 hardware_concurrency()。
419
+ // 實測(docs/04,repeat=3):mode C 下 t=8 最快,給更多執行緒反而變慢 ——
420
+ // t=8 4.02 / t=10 3.79 / t=12 3.55 / t=16 3.60 tok/s
421
+ // 原因很直接:每層每 token 只有 n_expert_used = 8 個 expert 可以平行,
422
+ // 執行緒超過 8 之後就只是多開執行緒搶同一批 CPU。
423
+ // 下限取實際核心數,避免在 4 核心的目標機器上開太多。
424
+ const int hw = (int) std::max(1u, std::thread::hardware_concurrency());
425
+ const int dflt_threads = hw < 8 ? hw : 8;
426
+ cparams.n_threads = opt.n_threads > 0 ? opt.n_threads : (int32_t) dflt_threads;
427
+ cparams.n_threads_batch = cparams.n_threads;
428
+ if (!opt.stats_file.empty()) {
429
+ cparams.offload_kqv = false;
430
+ }
431
+
432
+ if (tensor_fp) {
433
+ cparams.cb_eval = sdq_eval_callback;
434
+ cparams.cb_eval_user_data = &g_dump;
435
+ }
436
+
437
+ llama_context * ctx = llama_init_from_model(model, cparams);
438
+ if (!ctx) {
439
+ fprintf(stderr, "[sdq] context 建立失敗(ctx=%d 太大?)\n", opt.n_ctx);
440
+ return 1;
441
+ }
442
+
443
+ printf("[sdq] ctx=%d threads=%d ubatch=%u RAM(RSS)=%zu MB 模型載入 %.1fs\n",
444
+ opt.n_ctx, cparams.n_threads, cparams.n_ubatch, rss_mb(), t_load);
445
+ print_mem_report("載入 + pager 後");
446
+
447
+ // ---- sampler
448
+ auto sparams = llama_sampler_chain_default_params();
449
+ sparams.no_perf = true;
450
+ llama_sampler * smpl = llama_sampler_chain_init(sparams);
451
+ llama_sampler_chain_add(smpl, llama_sampler_init_top_k(opt.top_k));
452
+ llama_sampler_chain_add(smpl, llama_sampler_init_top_p(opt.top_p, 1));
453
+ llama_sampler_chain_add(smpl, llama_sampler_init_temp(opt.temp));
454
+ llama_sampler_chain_add(smpl, llama_sampler_init_dist(LLAMA_DEFAULT_SEED));
455
+
456
+ chat_state st;
457
+ if (!opt.system.empty()) {
458
+ st.msgs.push_back({"system", opt.system.c_str()});
459
+ }
460
+
461
+ std::vector<llama_token> decoded;
462
+
463
+ const llama_vocab * vocab = llama_model_get_vocab(model);
464
+
465
+ auto tokenize = [&](const std::string & text) {
466
+ std::vector<llama_token> toks(text.size() + 16);
467
+ int32_t n = llama_tokenize(vocab, text.c_str(), (int32_t) text.size(),
468
+ toks.data(), (int32_t) toks.size(), true, true);
469
+ if (n < 0) {
470
+ toks.resize((size_t) -n);
471
+ n = llama_tokenize(vocab, text.c_str(), (int32_t) text.size(),
472
+ toks.data(), (int32_t) toks.size(), true, true);
473
+ }
474
+ toks.resize(n);
475
+ return toks;
476
+ };
477
+
478
+ // ---- 送進去一個 ubatch,回傳生成的 token(-1 = 需要繼續生成)
479
+ std::function<int(llama_token, bool)> submit;
480
+ uint64_t n_prefill_tok = 0;
481
+ double t_prefill = 0;
482
+ uint64_t n_decode_tok = 0;
483
+ double t_decode = 0;
484
+ uint64_t io_read0 = read_bytes_io();
485
+
486
+ std::vector<float> logits_store;
487
+ FILE * logits_fp = nullptr;
488
+ if (opt.dump_logits) {
489
+ logits_fp = fopen(opt.logits_out.empty() ? "sdq_logits.bin" : opt.logits_out.c_str(), "wb");
490
+ }
491
+
492
+ submit = [&](llama_token tok, bool last) -> int {
493
+ const uint32_t ctx_size = llama_n_ctx(ctx);
494
+ llama_memory_t mem = llama_get_memory(ctx);
495
+ if (tok == 1) { // BOS:重開記憶體
496
+ llama_memory_clear(mem, true);
497
+ decoded.clear();
498
+ }
499
+ if (decoded.size() >= opt.n_ctx - 4) {
500
+ // 環形丟掉最舊的 token(保留 opt.n_keep)
501
+ const size_t drop = decoded.size() - (size_t) opt.n_keep;
502
+ llama_memory_seq_rm(mem, 0, 0, (int32_t) drop);
503
+ decoded.erase(decoded.begin(), decoded.begin() + (long) drop);
504
+ }
505
+
506
+ llama_batch b = llama_batch_get_one((llama_token *) &tok, 1);
507
+
508
+ const bool is_prefill = decoded.empty();
509
+ const double t0 = now_s();
510
+ const int rc = llama_decode(ctx, b);
511
+ const double dt = now_s() - t0;
512
+ if (rc != 0) {
513
+ fprintf(stderr, "\n[sdq] llama_decode 失敗 rc=%d(記憶體不足?)\n", rc);
514
+ exit(3);
515
+ }
516
+ decoded.push_back(tok);
517
+
518
+ // 每個 ubatch 結束 → 更新路由歷史並觸發下一輪預取
519
+ P.end_token();
520
+
521
+ if (is_prefill) {
522
+ n_prefill_tok++;
523
+ t_prefill += dt;
524
+ } else {
525
+ n_decode_tok++;
526
+ t_decode += dt;
527
+ P.get_stats().tokens.fetch_add(1, std::memory_order_relaxed);
528
+ }
529
+
530
+ if (logits_fp) {
531
+ const int32_t n_logits = llama_vocab_n_tokens(vocab);
532
+ const float * lg = llama_get_logits_ith(ctx, -1);
533
+ if (lg) {
534
+ logits_store.assign(lg, lg + n_logits);
535
+ fwrite(&n_logits, sizeof(int32_t), 1, logits_fp);
536
+ fwrite(logits_store.data(), sizeof(float), n_logits, logits_fp);
537
+ fflush(logits_fp);
538
+ }
539
+ }
540
+
541
+ if (last) {
542
+ return -1;
543
+ }
544
+ const llama_token id = llama_sampler_sample(smpl, ctx, -1);
545
+ // EOG 就停:這個模型生成完答案會吐 <|role_end|> / <|endoftext|>,
546
+ // 不停下來的話會一直重複同一組 special token。
547
+ // SDQ_BENCH_NO_EOG=1:benchmark 時固定生成 n 個 token(不同提示詞的生成長度
548
+ // 不同會讓記憶體實驗的 tok/s 失去可比性)。
549
+ if (llama_vocab_is_eog(vocab, id) && !getenv("SDQ_BENCH_NO_EOG")) {
550
+ fprintf(stderr, "\n[sdq] 生成 EOG token %d,停止\n", (int) id);
551
+ return -1;
552
+ }
553
+ return (int) id;
554
+ };
555
+
556
+ // ---- 產生 prompt
557
+ auto run_prompt = [&](const std::string & user_text, bool add_to_history) {
558
+ if (opt.chat) {
559
+ st.msgs.push_back({"user", user_text.c_str()});
560
+ }
561
+ std::string text;
562
+ if (opt.chat) {
563
+ const char * tmpl = llama_model_chat_template(model, nullptr);
564
+ size_t n_chars = 0;
565
+ for (const auto & m : st.msgs) {
566
+ n_chars += strlen(m.role) + strlen(m.content) + 32;
567
+ }
568
+ std::vector<char> buf(2 * n_chars + 1024);
569
+ int32_t len = llama_chat_apply_template(tmpl, st.msgs.data(), st.msgs.size(),
570
+ true, buf.data(), (int32_t) buf.size());
571
+ if (len > (int32_t) buf.size()) {
572
+ buf.resize((size_t) len + 1);
573
+ llama_chat_apply_template(tmpl, st.msgs.data(), st.msgs.size(), true,
574
+ buf.data(), (int32_t) buf.size());
575
+ }
576
+ text.assign(buf.data());
577
+ } else {
578
+ text = user_text;
579
+ }
580
+
581
+ if (getenv("SDQ_PRINT_PROMPT")) {
582
+ fprintf(stderr, "[sdq] prompt (%zu bytes): <<%s>>\n", text.size(), text.c_str());
583
+ }
584
+ auto toks = tokenize(text);
585
+ const int n_prompt = (int) toks.size();
586
+ if (getenv("SDQ_PRINT_PROMPT")) {
587
+ fprintf(stderr, "[sdq] prompt tokens (%d):", n_prompt);
588
+ for (int i = 0; i < n_prompt; i++) {
589
+ fprintf(stderr, " %d", (int) toks[i]);
590
+ }
591
+ fprintf(stderr, "\n");
592
+ }
593
+ const uint64_t io0 = read_bytes_io();
594
+
595
+ // 分批送進去(每批 = 一個 ubatch,控制記憶體)
596
+ const int32_t ub = (int32_t) cparams.n_ubatch;
597
+ llama_sampler_reset(smpl);
598
+ int32_t pos = 0;
599
+ for (int32_t i = 0; i < n_prompt; i += ub) {
600
+ const int32_t n = std::min(ub, n_prompt - i);
601
+ llama_batch b = llama_batch_init((uint32_t) n, 0, 1);
602
+ for (int32_t j = 0; j < n; j++) {
603
+ b.token[j] = toks[i + j];
604
+ b.pos[j] = pos++;
605
+ b.n_seq_id[j] = 1;
606
+ b.seq_id[j][0] = 0;
607
+ }
608
+ b.n_tokens = (uint32_t) n;
609
+ const double t0 = now_s();
610
+ const int rc = llama_decode(ctx, b);
611
+ t_prefill += now_s() - t0;
612
+ n_prefill_tok += (uint64_t) n;
613
+ if (rc != 0) {
614
+ fprintf(stderr, "\n[sdq] prefill 失敗 rc=%d(%d tokens)\n", rc, n);
615
+ exit(3);
616
+ }
617
+ P.end_token();
618
+ llama_batch_free(b);
619
+ }
620
+ const double t_p = now_s();
621
+ (void) t_p;
622
+ (void) io0;
623
+
624
+ if (add_to_history && opt.chat) {
625
+ st.msgs.push_back({"assistant", ""});
626
+ }
627
+ return n_prompt;
628
+ };
629
+
630
+ // ---- 產生
631
+ auto generate = [&](int32_t n_max) {
632
+ const uint64_t io0 = read_bytes_io();
633
+ const uint64_t io_read_before = io0;
634
+ (void) io_read_before;
635
+ const double t0 = now_s();
636
+ int32_t n_gen = 0;
637
+ llama_token id = llama_sampler_sample(smpl, ctx, -1);
638
+ fputs("assistant: ", stdout);
639
+ fflush(stdout);
640
+ while (n_gen < n_max) {
641
+ const int rc = submit(id, /*last=*/n_gen + 1 >= n_max);
642
+ if (rc < 0) {
643
+ break;
644
+ }
645
+ id = (llama_token) rc;
646
+ char piece_buf[256];
647
+ const int n_piece = llama_token_to_piece(vocab, id, piece_buf, sizeof(piece_buf), 0, true);
648
+ const std::string piece(piece_buf, (size_t) std::max(0, n_piece));
649
+ fputs(piece.c_str(), stdout);
650
+ fflush(stdout);
651
+ n_gen++;
652
+ }
653
+ fputs("\n", stdout);
654
+ const double dt = now_s() - t0;
655
+ const uint64_t io1 = read_bytes_io();
656
+ fprintf(stderr, "[sdq] decode: %d tok / %.2fs = %.2f tok/s | SSD 讀 %.1f MB (%.1f MB/tok) | RSS %zu MB\n",
657
+ n_gen, dt, dt > 0 ? n_gen / dt : 0.0,
658
+ (io1 - io0) / 1048576.0, n_gen ? (io1 - io0) / 1048576.0 / n_gen : 0.0, rss_mb());
659
+ if (opt.chat) {
660
+ for (auto & m : st.msgs) {
661
+ if (strcmp(m.role, "assistant") == 0) {
662
+ m.content = "";
663
+ }
664
+ }
665
+ }
666
+ P.dump_stats("decode");
667
+ return n_gen;
668
+ };
669
+
670
+ // ---- 合併讀取自我測試:--span-test IL:E0:E1:...
671
+ if (!opt.span_test.empty()) {
672
+ std::vector<int> parts;
673
+ std::string cur_s;
674
+ for (char ch : opt.span_test + ":") {
675
+ if (ch == ':') { parts.push_back(atoi(cur_s.c_str())); cur_s.clear(); }
676
+ else cur_s += ch;
677
+ }
678
+ if (parts.size() < 3) {
679
+ fprintf(stderr, "用法:--span-test IL:E0:E1[:E2...](E 要連續)\n");
680
+ return 1;
681
+ }
682
+ const int il = parts[0];
683
+ std::vector<int32_t> ex;
684
+ for (size_t i = 1; i < parts.size(); i++) {
685
+ ex.push_back(parts[i]);
686
+ }
687
+ std::vector<int32_t> slots(ex.size(), -1);
688
+ for (size_t i = 0; i < ex.size(); i++) {
689
+ slots[i] = P.reserve_slot(il, ex[i]);
690
+ }
691
+ const uint64_t got = P.read_expert_span(il, ex.data(), (int) ex.size(), slots.data());
692
+ printf("[sdq-span-test] L%d experts=%zu 讀取 %.2f MB\n", il, ex.size(), got / 1048576.0);
693
+ int bad = 0;
694
+ for (size_t i = 0; i < ex.size(); i++) {
695
+ std::vector<uint8_t> one((size_t) P.layer(il)->experts[ex[i]].bytes, 0);
696
+ if (!P.read_expert(il, ex[i], one.data())) {
697
+ printf(" 單獨讀取失敗\n");
698
+ return 1;
699
+ }
700
+ const uint8_t * span_ptr = P.slot_ptr(slots[i]);
701
+ size_t diff_at = (size_t) -1;
702
+ for (size_t b = 0; b < one.size(); b++) {
703
+ if (one[b] != span_ptr[b]) { diff_at = b; break; }
704
+ }
705
+ printf(" expert %4d: %s", ex[i], diff_at == (size_t) -1 ? "一致 ✅" : "不一致 ❌");
706
+ if (diff_at != (size_t) -1) {
707
+ printf("(第一個不同的 byte 在 %zu / %zu,slot=%d)", diff_at, one.size(), slots[i]);
708
+ bad++;
709
+ }
710
+ printf("\n");
711
+ }
712
+ sdq_shutdown();
713
+ llama_free(ctx);
714
+ llama_model_free(model);
715
+ llama_backend_free();
716
+ return bad ? 1 : 0;
717
+ }
718
+
719
+ // ---- 自我測試模式:--selftest IL:IE:SEED:OUT.bin
720
+ if (!opt.selftest.empty()) {
721
+ int il = 0, ie = 0, seed = 1234, n_tok = 1;
722
+ std::string outpath;
723
+ {
724
+ std::vector<std::string> parts;
725
+ std::string cur_s;
726
+ for (char c : opt.selftest + ":") {
727
+ if (c == ':') { parts.push_back(cur_s); cur_s.clear(); }
728
+ else cur_s += c;
729
+ }
730
+ if (parts.size() >= 3) {
731
+ il = atoi(parts[0].c_str());
732
+ ie = atoi(parts[1].c_str());
733
+ seed = atoi(parts[2].c_str());
734
+ }
735
+ if (parts.size() >= 4) {
736
+ outpath = parts[3];
737
+ }
738
+ if (parts.size() >= 5) {
739
+ n_tok = atoi(parts[4].c_str());
740
+ }
741
+ }
742
+ const int64_t n_embd = P.info().n_embd;
743
+ const int64_t n_ff = P.info().n_ff_exp;
744
+ std::vector<float> x((size_t) n_embd * n_tok);
745
+ uint32_t rng = (uint32_t) seed * 2654435761u + 1u;
746
+ for (auto & v : x) {
747
+ rng = rng * 1664525u + 1013904223u;
748
+ v = ((float) (rng >> 8) / (float) (1 << 24)) * 2.0f - 1.0f;
749
+ }
750
+ std::vector<float> io((size_t) n_embd * n_tok * 2);
751
+ const int rc = sdq_moe_selftest(il, ie, x.data(), io.data(), n_embd, n_ff, n_tok);
752
+ if (rc == 0 && !outpath.empty()) {
753
+ FILE * f = fopen(outpath.c_str(), "wb");
754
+ fwrite(io.data(), sizeof(float), io.size(), f);
755
+ fclose(f);
756
+ printf("[sdq] selftest 結果已寫到 %s(layer=%d expert=%d seed=%d n_embd=%lld n_ff=%lld)\n",
757
+ outpath.c_str(), il, ie, seed, (long long) n_embd, (long long) n_ff);
758
+ }
759
+ sdq_shutdown();
760
+ llama_free(ctx);
761
+ llama_model_free(model);
762
+ llama_backend_free();
763
+ return rc;
764
+ }
765
+
766
+ // ---- 主流程
767
+ if (!opt.prompt.empty()) {
768
+ run_prompt(opt.prompt, true);
769
+ generate(opt.n_predict);
770
+ }
771
+
772
+ if (opt.bench) {
773
+ // benchmark:固定提示詞 ×3,量 prefill / decode / 分頁指標
774
+ const std::vector<std::string> prompts = {
775
+ "請用中文簡短介紹 SSD、RAM、CPU 之間的資料分層是什麼,以及為什麼快取很重要。",
776
+ "Write a short paragraph explaining how a mixture-of-experts model routes tokens to experts.",
777
+ "請列出十種日常會用到電腦的工作,並各用一句話說明。",
778
+ };
779
+ for (int round = 0; round < 3; round++) {
780
+ fprintf(stderr, "[bench] round %d\n", round);
781
+ llama_memory_clear(llama_get_memory(ctx), true);
782
+ decoded.clear();
783
+ P.get_stats().tokens.store(0);
784
+ run_prompt(prompts[round % prompts.size()], false);
785
+ generate(48);
786
+ P.dump_stats(round == 0 ? "bench_warm" : "bench_hot");
787
+ }
788
+ }
789
+
790
+ if (opt.interactive) {
791
+ printf("\n(輸入文字開始對話;/exit 離開、/reset 重設、/stats 顯示指標)\n");
792
+ std::string line;
793
+ while (true) {
794
+ printf("\nuser> ");
795
+ fflush(stdout);
796
+ if (!std::getline(std::cin, line)) {
797
+ break;
798
+ }
799
+ if (line == "/exit" || line == "/quit") {
800
+ break;
801
+ }
802
+ if (line == "/reset") {
803
+ llama_memory_clear(llama_get_memory(ctx), true);
804
+ decoded.clear();
805
+ st.msgs.clear();
806
+ if (!opt.system.empty()) {
807
+ st.msgs.push_back({"system", opt.system.c_str()});
808
+ }
809
+ printf("(已重設)\n");
810
+ continue;
811
+ }
812
+ if (line == "/stats") {
813
+ P.dump_stats("manual");
814
+ fprintf(stderr, "hit_rate=%.1f%% arena=%zu MB RSS=%zu MB\n",
815
+ P.hit_rate() * 100, P.used_bytes() / 1048576, rss_mb());
816
+ continue;
817
+ }
818
+ if (line.empty()) {
819
+ continue;
820
+ }
821
+ run_prompt(line, true);
822
+ generate(opt.n_predict);
823
+ }
824
+ }
825
+
826
+ P.get_stats().moe_calls.fetch_add(0);
827
+ sdq_shutdown();
828
+
829
+ fprintf(stderr, "\n[sdq] ===== 總結 =====\n");
830
+ fprintf(stderr, "[sdq] prefill: %" PRIu64 " tok / %.2fs = %.1f tok/s\n",
831
+ n_prefill_tok, t_prefill, t_prefill > 0 ? n_prefill_tok / t_prefill : 0.0);
832
+ const uint64_t io_end = read_bytes_io();
833
+ fprintf(stderr, "[sdq] 全程 SSD 實際讀取: %.2f GB\n", (io_end - io_read0) / 1073741824.0);
834
+ print_mem_report("結束時");
835
+
836
+ if (logits_fp) {
837
+ fclose(logits_fp);
838
+ }
839
+ llama_sampler_free(smpl);
840
+ llama_free(ctx);
841
+ llama_model_free(model);
842
+ llama_backend_free();
843
+ return 0;
844
+ }
sdq-ling/sdq_moe.cpp ADDED
@@ -0,0 +1,735 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ // SDQ 被分頁的 MoE FFN(自訂 ggml op)
2
+ //
3
+ // 為什麼要自己算:上游 build_moe_ffn() 用 ggml_mul_mat_id 直接對
4
+ // blk.N.ffn_{gate,up,down}_exps.weight 這個「三維大張量」做矩陣乘,
5
+ // 權重必須整個在 RAM 裡。20.6 GiB 的專家權重在 8 GB 機器上不可能。
6
+ //
7
+ // 這裡改成:
8
+ // 1. 路由(gate_inp matmul + softmax + top-k + 權重正規化)仍由 ggml 圖算完,
9
+ // 我們只拿 selected_experts [n_expert_used, n_tokens] 與 weights。
10
+ // 2. 每個 (layer, token) 需要的 8 個 expert 由 pager 保證在 RAM:
11
+ // 缺頁 → 從 SSD pread 進 arena(背景執行緒的預取可先後到)。
12
+ // 3. 用 ggml 自己最佳化的量化點積(ggml_vec_dot_*,AVX2/AVX512)逐 expert 算
13
+ // gate/up → SiLU → down → 加權累加。
14
+ // 4. 多執行緒:ith/nth 由 ggml 分配;分頁階段各執行緒各自 pread,餵飽 SSD 佇列深度。
15
+
16
+ #include "sdq_pager.h"
17
+
18
+ #include "ggml-cpu.h"
19
+ #include "ggml-cpu/quants.h"
20
+
21
+ #include <algorithm>
22
+ #include <atomic>
23
+ #include <chrono>
24
+ #include <cmath>
25
+ #include <condition_variable>
26
+ #include <cstdio>
27
+ #include <cstring>
28
+ #include <mutex>
29
+ #include <utility>
30
+ #include <string>
31
+ #include <vector>
32
+
33
+ namespace sdq {
34
+
35
+ // ---------------------------------------------------------------- 效能計時探針
36
+ // 用途:找出「自訂 MoE op」裡面時間到底花在哪裡。開 SDQ_PROFILE_TIME=1 才啟用,
37
+ // 關閉時是零成本(幾個 if constexpr 級別的布林檢查)。
38
+ struct moe_prof {
39
+ std::atomic<uint64_t> op_ns{0}; // 整個 op
40
+ std::atomic<uint64_t> page_ns{0}; // probe / reserve_slot(含鎖)
41
+ std::atomic<uint64_t> setup_ns{0}; // op 前置:unpin_all + 建 tok_of_expert + 前兩道 barrier
42
+ std::atomic<uint64_t> bar_ns{0}; // 所有 barrier 的等待時間
43
+ std::atomic<uint64_t> read_ns{0}; // read_expert_span / acquire 的 I/O
44
+ std::atomic<uint64_t> fwd_ns{0}; // expert_forward 合計
45
+ std::atomic<uint64_t> qact_ns{0}; // 其中:量化 activation
46
+ std::atomic<uint64_t> dot_ns{0}; // 其中:gate/up/down 三次點積
47
+ std::atomic<uint64_t> act_ns{0}; // 其中:SiLU + 加權
48
+ std::atomic<uint64_t> calls{0};
49
+ std::atomic<uint64_t> experts{0};
50
+ };
51
+ static moe_prof g_prof;
52
+
53
+ static inline bool prof_on() {
54
+ static const bool on = [] { const char * e = getenv("SDQ_PROFILE_TIME"); return e && *e && *e != '0'; }();
55
+ return on;
56
+ }
57
+ static inline uint64_t now_ns() {
58
+ return (uint64_t) std::chrono::duration_cast<std::chrono::nanoseconds>(
59
+ std::chrono::steady_clock::now().time_since_epoch()).count();
60
+ }
61
+ #define PROF_T0_x(v) const uint64_t v = prof_on() ? now_ns() : 0
62
+ #define PROF_ADD(field, tv) do { if (prof_on()) g_prof.field += now_ns() - (tv); } while (0)
63
+
64
+ // ---------------------------------------------------------------- 執行緒柵欄
65
+
66
+ class gen_barrier {
67
+ public:
68
+ void sync(int n) {
69
+ PROF_T0_x(_tbar);
70
+ std::unique_lock<std::mutex> lk(mu_);
71
+ const unsigned long g = gen_;
72
+ if (++count_ == n) {
73
+ count_ = 0;
74
+ gen_++;
75
+ cv_.notify_all();
76
+ PROF_ADD(bar_ns, _tbar);
77
+ return;
78
+ }
79
+ cv_.wait(lk, [this, g] { return gen_ != g; });
80
+ PROF_ADD(bar_ns, _tbar);
81
+ }
82
+
83
+ private:
84
+ int count_ = 0;
85
+ unsigned long gen_ = 0;
86
+ std::mutex mu_;
87
+ std::condition_variable cv_;
88
+ };
89
+
90
+ // ---------------------------------------------------------------- 量化點積派送
91
+
92
+ // ggml-cpu 的 K-quant kernel 把 fp16 scale 交給 ggml_table_f32_f16 查表換算
93
+ // (simd-mappings.h 的 fallback 分支,x86 走的就是這條),而那張表只在
94
+ // ggml_cpu_init() 裡填。用 ggml_vec_dot_* 直接呼叫 kernel 時若 ggml_cpu_init()
95
+ // 還沒跑過,每個點積都會安靜地回傳 0 —— 不會 crash,也不會報錯。
96
+ // llama.cpp 自己的 mul_mat 不會踩到這個坑,因為 backend 建立時一定呼叫過
97
+ // ggml_cpu_init();自訂 op 走的是 ggml graph,時序上不一定保證。
98
+ static void sdq_ensure_cpu_init() {
99
+ static std::once_flag once;
100
+ std::call_once(once, [] { ggml_cpu_init(); });
101
+ }
102
+
103
+ using vec_dot_fn = void (*)(int, float *, size_t, const void *, size_t, const void *, size_t, int);
104
+
105
+ static vec_dot_fn dot_for(ggml_type t) {
106
+ switch (t) {
107
+ case GGML_TYPE_Q4_K: return ggml_vec_dot_q4_K_q8_K;
108
+ case GGML_TYPE_Q5_K: return ggml_vec_dot_q5_K_q8_K;
109
+ case GGML_TYPE_Q6_K: return ggml_vec_dot_q6_K_q8_K;
110
+ case GGML_TYPE_Q3_K: return ggml_vec_dot_q3_K_q8_K;
111
+ case GGML_TYPE_Q2_K: return ggml_vec_dot_q2_K_q8_K;
112
+ case GGML_TYPE_Q8_0: return ggml_vec_dot_q8_0_q8_0;
113
+ case GGML_TYPE_Q4_0: return ggml_vec_dot_q4_0_q8_0;
114
+ case GGML_TYPE_Q5_0: return ggml_vec_dot_q5_0_q8_0;
115
+ // Qwen3.8-Flash-Next 的 down_exps 用 q5_1(unsloth UD-Q4_K_XL)。
116
+ // q5_1 在 ggml 裡配的是 q8_1 activation(見 type_traits:q5_1 的
117
+ // vec_dot_type 是 q8_1),不是 q8_0 —— 配錯會整層數值錯掉。
118
+ case GGML_TYPE_Q5_1: return ggml_vec_dot_q5_1_q8_1;
119
+ case GGML_TYPE_IQ4_NL: return ggml_vec_dot_iq4_nl_q8_0;
120
+ case GGML_TYPE_IQ4_XS: return ggml_vec_dot_iq4_xs_q8_K;
121
+ default: return nullptr;
122
+ }
123
+ }
124
+
125
+ // K-quant 的權重配 Q8_K activation;其餘配 Q8_0(與 ggml 的 vec_dot_type 一致)
126
+ static ggml_type act_type_for(ggml_type w) {
127
+ switch (w) {
128
+ case GGML_TYPE_Q4_K:
129
+ case GGML_TYPE_Q5_K:
130
+ case GGML_TYPE_Q6_K:
131
+ case GGML_TYPE_Q3_K:
132
+ case GGML_TYPE_Q2_K:
133
+ case GGML_TYPE_IQ4_XS:
134
+ return GGML_TYPE_Q8_K;
135
+ // q5_1 配 q8_1(q5_1 的 min 值是 fp16,不是 q8_0 的 0)
136
+ case GGML_TYPE_Q5_1:
137
+ return GGML_TYPE_Q8_1;
138
+ default:
139
+ return GGML_TYPE_Q8_0;
140
+ }
141
+ }
142
+
143
+ // 把 activation 量化成對應的 dot 型別
144
+ // 注意:ggml_quantize_chunk() 不支援 Q8_K(Q8_K 只在 mul_mat kernel 內部產生),
145
+ // 所以這裡直接呼叫 ggml-cpu 的 quantize_row_q8_K。
146
+ static void quantize_act(ggml_type t, const float * x, void * dst, int64_t n) {
147
+ switch (t) {
148
+ case GGML_TYPE_Q8_K: quantize_row_q8_K(x, dst, n); break;
149
+ case GGML_TYPE_Q8_1: quantize_row_q8_1(x, dst, n); break;
150
+ case GGML_TYPE_Q8_0: quantize_row_q8_0(x, dst, n); break;
151
+ default: ggml_quantize_chunk(t, x, dst, 0, 1, n, nullptr); break;
152
+ }
153
+ }
154
+
155
+ // ⚠️ 不要在這裡「批次化」點積。
156
+ // ggml 的 x86 版 K-quant kernel(ggml/src/ggml-cpu/arch/x86/quants.c)開頭就是
157
+ // assert(nrc == 1); UNUSED(nrc); UNUSED(bx); UNUSED(by); UNUSED(bs);
158
+ // 只有 ARM 的 SVE+MATMUL 版本才支援 nrc==2。而 Release 編譯(-DNDEBUG)會把 assert
159
+ // 編譯掉,於是「傳 nrc=16」不會報錯、只會安靜地算第 1 列,其餘 15 列留在記憶體裡
160
+ // 是舊資料 —— 產出看起來合理但數值全錯。
161
+ // (本專案真的踩過:微基準測量到「nrc=16 快 16 倍」,其實是「只做了 1/16 的工作」,
162
+ // 而且權重填 0x5A 讓 checksum 兩邊都是 0,完美掩蓋了錯誤。教訓見 docs/04。)
163
+ // 想要真的多列 SIMD 效能,只能自己寫 AVX512-VNNI 的 kernel,不是靠改參數。
164
+
165
+ static inline void dot_row(vec_dot_fn fn, const uint8_t * row, const void * xq,
166
+ size_t stride_row, int n, float & acc) {
167
+ float s = 0.0f;
168
+ fn(n, &s, 0, row, stride_row, xq, 0, 1);
169
+ acc += s;
170
+ }
171
+
172
+ // 一次算 SDQ_DOT_BATCH 個輸出列。
173
+ // 為什麼需要:ggml 的 K-quant 點積 kernel 是為「多列」設計的,nrc>1 會走完全不同的
174
+ // SIMD 路徑。實測(tools 裡的 microdot):nrc=1 只有 24~28 GMAC/s,nrc=16 有
175
+ // 347~438 GMAC/s(快 14~16 倍),而且 **checksum 完全相同**(逐位元組等價)。
176
+ // 舊程式逐列呼叫 nrc=1,等於把 14 倍的效能丟掉。
177
+ static const int SDQ_DOT_BATCH = 16;
178
+
179
+ static inline void dot_rows(vec_dot_fn fn, const uint8_t * rows, size_t row_stride,
180
+ const void * xq, int n, int64_t n_rows, float * out) {
181
+ int64_t r = 0;
182
+ for (; r + SDQ_DOT_BATCH <= n_rows; r += SDQ_DOT_BATCH) {
183
+ fn(n, out + r, 1, rows + (size_t) r * row_stride, row_stride, xq, 0, SDQ_DOT_BATCH);
184
+ }
185
+ for (; r < n_rows; r++) {
186
+ fn(n, out + r, 0, rows + (size_t) r * row_stride, row_stride, xq, 0, 1);
187
+ }
188
+ }
189
+
190
+ // ---------------------------------------------------------------- 路由紀錄(SDQ_ROUTE_LOG)
191
+ // 用途:量測「連續 k 個 token 的 expert 集合重疊多少」。
192
+ // 這決定多 token 批次(投機解碼 / 批次解碼)能不能省下 SSD 讀取:
193
+ // 每層的關鍵路徑是「有沒有任何一個 expert 沒命中」,所以 U(k) < 8k 就有賺。
194
+ // 格式(每行一次 op 呼叫,空白分欄):
195
+ // <layer> <n_tok> <t0:e0,e1,...> <t1:...> ...
196
+ static void sdq_route_log(int il, int64_t n_tok, const ggml_tensor * ids,
197
+ size_t ids_nb0, size_t ids_nb1, int n_used) {
198
+ static const char * path = getenv("SDQ_ROUTE_LOG");
199
+ if (!path || !*path) {
200
+ return;
201
+ }
202
+ static std::mutex mtx;
203
+ static FILE * f = nullptr;
204
+ {
205
+ std::lock_guard<std::mutex> lk(mtx);
206
+ if (!f) {
207
+ f = fopen(path, "w");
208
+ }
209
+ if (!f) {
210
+ return;
211
+ }
212
+ fprintf(f, "%d %lld", il, (long long) n_tok);
213
+ const uint8_t * base = (const uint8_t *) ids->data;
214
+ for (int64_t t = 0; t < n_tok; t++) {
215
+ fprintf(f, " %lld:", (long long) t);
216
+ for (int k = 0; k < n_used; k++) {
217
+ const int32_t e = *(const int32_t *) (base + (size_t) k * ids_nb0 + (size_t) t * ids_nb1);
218
+ fprintf(f, "%s%d", k ? "," : "", e);
219
+ }
220
+ }
221
+ fputc('\n', f);
222
+ }
223
+ }
224
+
225
+ // ---------------------------------------------------------------- 每層執行紀錄
226
+
227
+ // 重要:input 張量的指標**不能**在建圖時暫存起來 —— llama.cpp 每次 decode 都會
228
+ // 重新建圖,graph context 的配置器可能把張量搬到不同位址(形狀不同時尤其明顯)。
229
+ // 所以 op 執行時一律從 dst->src[] 取得,這才是唯一可靠的來源。
230
+ struct layer_rec {
231
+ int il = -1;
232
+ int64_t n_expert = 0;
233
+ int64_t n_expert_used = 0;
234
+ float swiglu_limit = 0.0f;
235
+
236
+ // 每個 expert 需要處理的 (token, topk 位置) 配對
237
+ std::vector<std::vector<std::pair<int32_t, int32_t>>> tok_of_expert;
238
+ // 每個 token 用到的 (expert, topk 位置) —— token 切分時用(避免輸出競爭)
239
+ std::vector<std::vector<std::pair<int32_t, int32_t>>> exp_of_tok;
240
+ std::vector<int32_t> need; // 本層要用的 expert 編號
241
+ std::vector<int32_t> need_sorted; // 依編號排序(合併讀取用)
242
+ std::vector<int32_t> need_slot; // 對應的 arena 槽位
243
+ std::vector<std::vector<float>> partial; // mode A:每個執行緒一份累加緩衝
244
+ std::vector<int32_t> slot_of_expert; // [expert] -> arena 槽位(-1 = 沒分頁)
245
+ std::vector<int32_t> need_slot_of_expert; // 合併讀取時暫存槽位
246
+ std::vector<int32_t> miss_sorted; // 需要從 SSD 讀的 expert(已排序)
247
+ gen_barrier bar;
248
+ std::mutex mtx;
249
+ bool warned = false;
250
+ };
251
+
252
+ static std::vector<layer_rec *> g_recs;
253
+
254
+ static layer_rec * rec_for(int il) {
255
+ static std::mutex m;
256
+ std::lock_guard<std::mutex> lk(m);
257
+ if ((int) g_recs.size() <= il) {
258
+ g_recs.resize(il + 1, nullptr);
259
+ }
260
+ if (!g_recs[il]) {
261
+ g_recs[il] = new layer_rec();
262
+ g_recs[il]->il = il;
263
+ }
264
+ return g_recs[il];
265
+ }
266
+
267
+ // ---------------------------------------------------------------- 自訂 op 本體
268
+
269
+ // mode C(預設,partial 緩衝 <= 上限):依 expert 流水線,取→算交錯、無中途 barrier。
270
+ // mode B(partial 緩衝太大時的退路):先分頁完再一起算,會有 barrier 等待。
271
+ // 上限給一個很大的值=永遠走 mode C;給 0=強制 mode B(除錯用)。
272
+ static size_t env_partial_cap_mb() {
273
+ const char * e = getenv("SDQ_PARTIAL_CAP_MB");
274
+ return e ? (size_t) atoi(e) : 4096; // MB,預設 4096 MB → 一律 mode C
275
+ }
276
+
277
+ // ---------------------------------------------------------------- 單一 expert 的數學
278
+ // 從 arena 的槽位記憶體算出一個 expert 對多個 token 的輸出(加權後直接累加到 out)
279
+ static void expert_forward(const expert_desc & ed, const uint8_t * base,
280
+ const float * cur, int64_t n_embd,
281
+ const std::vector<std::pair<int32_t, int32_t>> & toks,
282
+ const float * wts, size_t wts_nb0, size_t wts_nb1, int /*n_used*/,
283
+ float swiglu_limit,
284
+ float * out, std::vector<uint8_t> * scratch) {
285
+ sdq_ensure_cpu_init();
286
+
287
+ int64_t gate_rows = ed.ne1;
288
+ int64_t up_rows = ed.ne1;
289
+ size_t gate_row = ggml_row_size(ed.type[part::GATE], ed.ne0);
290
+ size_t up_row = gate_row;
291
+ if (ed.fused_gate_up) {
292
+ gate_rows = ed.ne1 / 2;
293
+ up_rows = gate_rows;
294
+ } else {
295
+ up_row = ggml_row_size(ed.type[part::UP], ed.ne0);
296
+ }
297
+
298
+ const vec_dot_fn f_gate = dot_for(ed.type[part::GATE]);
299
+ const vec_dot_fn f_up = ed.fused_gate_up ? f_gate : dot_for(ed.type[part::UP]);
300
+ const vec_dot_fn f_down = dot_for(ed.type[part::DOWN]);
301
+ if (!f_gate || !f_up || !f_down) {
302
+ fprintf(stderr, "[sdq] 不支援的專家量化型別:gate=%s up=%s down=%s\n",
303
+ ggml_type_name(ed.type[part::GATE]),
304
+ ggml_type_name(ed.type[part::UP]),
305
+ ggml_type_name(ed.type[part::DOWN]));
306
+ return;
307
+ }
308
+
309
+ const uint8_t * wg = base + ed.slot_off[part::GATE];
310
+ const uint8_t * wu = base + ed.slot_off[part::UP];
311
+ const uint8_t * wd = base + ed.slot_off[part::DOWN];
312
+ const size_t down_row = ggml_row_size(ed.type[part::DOWN], ed.down_ne0);
313
+
314
+ const ggml_type act_gate = act_type_for(ed.type[part::GATE]);
315
+ const ggml_type act_down = act_type_for(ed.type[part::DOWN]);
316
+ const size_t xq_sz = ggml_row_size(act_gate, ed.ne0);
317
+ const size_t hq_sz = ggml_row_size(act_down, ed.down_ne0);
318
+ if (scratch->size() < xq_sz + hq_sz) {
319
+ scratch->assign(xq_sz + hq_sz, 0);
320
+ }
321
+ uint8_t * xq = scratch->data();
322
+ uint8_t * hq = xq + xq_sz;
323
+
324
+ std::vector<float> gate((size_t) gate_rows);
325
+ std::vector<float> upv((size_t) up_rows);
326
+ std::vector<float> h((size_t) ed.down_ne0);
327
+
328
+ PROF_T0_x(_ta); // expert_forward 總計時
329
+ for (const auto & tk : toks) {
330
+ const int32_t t = tk.first;
331
+ const int32_t k = tk.second;
332
+
333
+ PROF_T0_x(_tb); // 量化 activation
334
+ quantize_act(act_gate, cur + (size_t) t * n_embd, xq, ed.ne0);
335
+ PROF_ADD(qact_ns, _tb);
336
+
337
+ PROF_T0_x(_tc); // gate / up / down 三次點積
338
+ for (int64_t r = 0; r < gate_rows; r++) {
339
+ float acc = 0.0f;
340
+ dot_row(f_gate, wg + r * gate_row, xq, gate_row, (int) ed.ne0, acc);
341
+ gate[(size_t) r] = acc;
342
+ }
343
+ for (int64_t r = 0; r < up_rows; r++) {
344
+ float acc = 0.0f;
345
+ const uint8_t * src = ed.fused_gate_up ? (wg + (gate_rows + r) * gate_row)
346
+ : (wu + r * up_row);
347
+ dot_row(f_up, src, xq, ed.fused_gate_up ? gate_row : up_row, (int) ed.ne0, acc);
348
+ upv[(size_t) r] = acc;
349
+ }
350
+
351
+ PROF_ADD(dot_ns, _tc);
352
+
353
+ PROF_T0_x(_td); // SiLU + 加權求和
354
+ for (int64_t r = 0; r < ed.down_ne0; r++) {
355
+ const float g = gate[(size_t) r];
356
+ const float u = upv[(size_t) r];
357
+ float act;
358
+ if (swiglu_limit > 1e-6f) {
359
+ const float uc = std::max(-swiglu_limit, std::min(swiglu_limit, u));
360
+ float gc = g / (1.0f + std::exp(-g));
361
+ gc = std::max(-swiglu_limit, std::min(swiglu_limit, gc));
362
+ act = gc * uc;
363
+ } else {
364
+ act = (g / (1.0f + std::exp(-g))) * u;
365
+ }
366
+ h[(size_t) r] = act;
367
+ }
368
+ quantize_act(act_down, h.data(), hq, ed.down_ne0);
369
+
370
+ PROF_ADD(act_ns, _td);
371
+
372
+ const float w = *(const float *) ((const uint8_t *) wts + k * wts_nb0 + (size_t) t * wts_nb1);
373
+ float * o = out + (size_t) t * n_embd;
374
+ for (int64_t r = 0; r < ed.down_ne1; r++) {
375
+ float acc = 0.0f;
376
+ dot_row(f_down, wd + r * down_row, hq, down_row, (int) ed.down_ne0, acc);
377
+ o[r] += acc * w;
378
+ }
379
+ }
380
+ PROF_ADD(fwd_ns, _ta);
381
+ }
382
+
383
+
384
+ static void moe_custom_op(ggml_tensor * dst, int ith, int nth, void * userdata) {
385
+ layer_rec * R = (layer_rec *) userdata;
386
+ pager & P = pager::instance();
387
+
388
+ // 從 op 的 src 取得輸入(保證是「這一次」真的被計算的張量)
389
+ const ggml_tensor * src_cur = dst->src[0];
390
+ const ggml_tensor * src_ids = dst->src[1];
391
+ const ggml_tensor * src_wts = dst->src[2];
392
+ if (!src_cur || !src_ids || !src_wts) {
393
+ if (ith == 0) {
394
+ fprintf(stderr, "[sdq] op 缺少 src(%d)\n", R->il);
395
+ }
396
+ return;
397
+ }
398
+
399
+ const layer_desc * L = P.layer(R->il);
400
+ PROF_T0_x(_tg); // op 總計時
401
+ g_prof.calls++;
402
+ const int64_t n_embd = src_cur->ne[0];
403
+ const int64_t n_tok = src_cur->ne[2];
404
+ const int64_t n_used = R->n_expert_used;
405
+ // ids 是 2D [n_expert_used, n_tokens] → 用 nb[0]/nb[1]
406
+ // wts 是 3D [1, n_expert_used, n_tokens] → 用 nb[1]/nb[2](nb[0] 會是 4 而不是 n_used*4!)
407
+ const size_t ids_nb0 = src_ids->nb[0], ids_nb1 = src_ids->nb[1];
408
+ const size_t wts_nb0 = src_wts->nb[src_wts->ne[0] == 1 ? 1 : 0];
409
+ const size_t wts_nb1 = src_wts->nb[src_wts->ne[0] == 1 ? 2 : 1];
410
+ const float * cur = (const float *) src_cur->data;
411
+ const float * wts = (const float *) src_wts->data;
412
+
413
+ if (!L || L->experts.empty() || dst->type != GGML_TYPE_F32) {
414
+ if (ith == 0 && !R->warned) {
415
+ R->warned = true;
416
+ fprintf(stderr, "[sdq] 層 %d 沒有可用的分頁描述\n", R->il);
417
+ }
418
+ return;
419
+ }
420
+
421
+ if (R->il == 0 && getenv("SDQ_DBG_OP")) {
422
+ fprintf(stderr, "[sdq-op] L%d n_tok=%lld nth=%d need=%d ids_nb1=%zu wts_nb1=%zu\n",
423
+ R->il, (long long) n_tok, nth, (int) R->need.size(), ids_nb1, wts_nb1);
424
+ }
425
+ // ---- 1) 解除上一層的 pin(mode B 直接累加輸出,需要先清空)
426
+ PROF_T0_x(_tsetup);
427
+ if (ith == 0) {
428
+ P.unpin_all();
429
+ }
430
+ if (nth > 1) { R->bar.sync(nth); }
431
+
432
+ // ---- 2) 依 expert 分組 token(只有執行緒 0 做,其餘在此等待)
433
+ if (ith == 0) {
434
+ std::lock_guard<std::mutex> lk(R->mtx);
435
+ R->tok_of_expert.assign((size_t) R->n_expert, {});
436
+ R->exp_of_tok.assign((size_t) n_tok, {});
437
+ const uint8_t * ids_base = (const uint8_t *) src_ids->data;
438
+ for (int64_t t = 0; t < n_tok; t++) {
439
+ for (int64_t k = 0; k < n_used; k++) {
440
+ const int32_t e = *(const int32_t *) (ids_base + k * ids_nb0 + (size_t) t * ids_nb1);
441
+ if (e >= 0 && e < R->n_expert) {
442
+ auto & v = R->tok_of_expert[(size_t) e];
443
+ bool seen = false;
444
+ for (const auto & p : v) {
445
+ if (p.first == (int32_t) t) { seen = true; break; }
446
+ }
447
+ if (!seen) {
448
+ v.emplace_back((int32_t) t, (int32_t) k);
449
+ }
450
+ R->exp_of_tok[(size_t) t].emplace_back(e, (int32_t) k);
451
+ }
452
+ }
453
+ }
454
+ R->need.clear();
455
+ for (int64_t e = 0; e < R->n_expert; e++) {
456
+ if (!R->tok_of_expert[(size_t) e].empty()) {
457
+ R->need.push_back((int32_t) e);
458
+ }
459
+ }
460
+ R->need_sorted = R->need;
461
+ std::sort(R->need_sorted.begin(), R->need_sorted.end());
462
+ R->need_slot.assign(R->need.size(), -1); // 只能由 ith==0 配置,其餘執行緒唯讀
463
+ R->slot_of_expert.assign((size_t) R->n_expert, -1);
464
+ R->need_slot_of_expert.assign((size_t) R->n_expert, -1);
465
+ R->miss_sorted.clear();
466
+ R->miss_sorted.reserve((size_t) R->n_expert);
467
+ if (R->partial.size() < 64) {
468
+ R->partial.resize(64);
469
+ }
470
+ sdq_route_log(R->il, n_tok, src_ids, ids_nb0, ids_nb1, (int) n_used);
471
+ }
472
+ if (nth > 1) { R->bar.sync(nth); }
473
+ PROF_ADD(setup_ns, _tsetup);
474
+
475
+ const int n_need = (int) R->need.size();
476
+
477
+ // ---- 3+4) 分頁與計算融合成一條 pipeline
478
+ //
479
+ // 關鍵效能點:不要「先把整層的 expert 都讀完再開始算」。做法是每個執行緒
480
+ // 自己依序處理它負責的 expert:讀一個 → 立刻算它 → 再讀下一個。
481
+ // 這樣 SSD 讀取和 CPU 計算自然重疊,而且 8 條執行緒各自發出 pread,
482
+ // SSD 佇列深度也夠(不需要再靠 queue depth)。
483
+ //
484
+ // 平行切分:mode A 依 expert 切(每個執行緒有自己的累加緩衝,最後相加);
485
+ // mode B 依 token 切(同一個 token 的所有 expert 由同一執行緒算,
486
+ // 但必須先把該 token 的 expert 都備齊,所以先分頁再計算)。
487
+ const int64_t out_elems = n_embd * n_tok;
488
+ const size_t partial_bytes = (size_t) out_elems * 4 * (size_t) nth;
489
+ const bool mode_a = partial_bytes <= (size_t) env_partial_cap_mb() << 20;
490
+ float * out = (float *) dst->data;
491
+
492
+ if (!mode_a) {
493
+ memset(dst->data, 0, (size_t) ggml_nbytes(dst));
494
+ }
495
+
496
+ if (mode_a) {
497
+ // ===================== mode C(預設):依 expert 流水線 =====================
498
+ // 三個設計重點,每個都來自實測:
499
+ //
500
+ // (1) **取→算交錯,中間沒有 barrier**。
501
+ // 舊的 mode B 是「先把整層 8 個 expert 全部讀完 → barrier → 才開始算」。
502
+ // 量測顯示 decode 有 66% 的時間卡在 barrier 上等最慢的執行緒
503
+ // (SDQ_PROFILE_TIME 的 barrier 等待 = op 時間的 70%)。改成交錯後
504
+ // 讀取與計算自然重疊,實測 2.79 → 3.49 tok/s。
505
+ //
506
+ // (2) **分頁走 acquire(),不用 reserve_slot + read_expert_span**。
507
+ // reserve_slot() 會在資料還沒讀進來之前就把 (layer,expert) publish 到
508
+ // page_ 表;一旦該層的 pread 失敗或被中斷,這個槽就會留在 page_ 裡
509
+ // 卻是舊資料,之後的 probe() 會把它當成命中 → 靜默算錯。
510
+ // acquire() 是「先讀完再 publish」,成本相同但不會有這個失效模式。
511
+ //
512
+ // (3) **i += nth 交錯分配**,不是切成連續區塊。
513
+ // 每個 expert 的 SSD 延遲差異很大,交錯分配讓快執行緒自然多幫忙。
514
+ //
515
+ // 輸出不直接寫 dst(同一個 token 的不同 expert 會撞在一起),
516
+ // 所以仍然用「每執行緒一份累加緩衝 + 最後歸約」。
517
+ std::vector<float> & mine = R->partial[(size_t) ith];
518
+ mine.assign((size_t) out_elems, 0.0f);
519
+ std::vector<uint8_t> scratch;
520
+
521
+ // 在取本層的 expert **之前**先替下一層排預取。
522
+ //
523
+ // 實測結論(S7):**預設關閉,而且量到開啟反而慢 8%。**
524
+ // 預測來自 prev_used_[il+1](上一個 token 在該層選的專家)。想用它躲在
525
+ // 「本層的讀取+計算」後面,但量到的結果是:
526
+ // 開(5.68 tok/s,預取使用率 0.48 %)vs 關(6.20 tok/s,0.28 %)
527
+ // 命中率幾乎一樣(70.5 % vs 70.7 %),但 SSD 位元組從 201.5 升到 216.0 MB/token
528
+ // → 預測不夠準,多讀的全是白讀,而且還要和本層的 demand 讀搶頻寬。
529
+ // 另外開啟時同一提示詞的輸出不一致 → 預取佔用的槽位(busy)會讓
530
+ // acquire() 偶爾拿不到槽位而**靜默跳過一個 expert**,這是正確性問題。
531
+ //
532
+ // 保留程式碼是為了讓後人不用重新發明;文件見 docs/06_slots.md §4。
533
+ // SDQ_PREFETCH_LAYER_AHEAD=1 可以打開來重現那個負結果。
534
+ static const bool layer_ahead = [] {
535
+ const char * e = getenv("SDQ_PREFETCH_LAYER_AHEAD");
536
+ return e ? atoi(e) != 0 : false;
537
+ }();
538
+ if (layer_ahead && ith == 0) {
539
+ P.prefetch_layer_ahead(R->il + 1);
540
+ }
541
+
542
+ for (int i = ith; i < n_need; i += nth) {
543
+ const int32_t e = R->need[(size_t) i];
544
+ const int32_t sl = P.acquire(R->il, e, true);
545
+ if (sl < 0) {
546
+ continue;
547
+ }
548
+ P.note_used(R->il, e);
549
+ expert_forward(L->experts[(size_t) e], P.slot_ptr(sl), cur, n_embd,
550
+ R->tok_of_expert[(size_t) e], wts, wts_nb0, wts_nb1,
551
+ (int) n_used, R->swiglu_limit, mine.data(), &scratch);
552
+ // expert_forward 回傳後這個槽位就沒人讀了(最後的歸約只讀 mine),
553
+ // 所以立刻放掉 pin,不要讓它被釘到下一層開頭。
554
+ // 不放的話 prefill 一層會把整層用到的 expert 全部釘住(最多 256 個),
555
+ // 大槽位不夠時就會中止(見 pager::release 的說明)。
556
+ P.release(sl);
557
+ }
558
+
559
+ if (nth > 1) { R->bar.sync(nth); }
560
+ for (int64_t idx = ith; idx < out_elems; idx += nth) {
561
+ float acc = 0.0f;
562
+ for (int k = 0; k < nth; k++) {
563
+ acc += R->partial[(size_t) k][(size_t) idx];
564
+ }
565
+ out[idx] = acc;
566
+ }
567
+ } else {
568
+ // mode B:token 多到 partial 緩衝太大時,依 token 切分
569
+ // 先把所有需要的 expert 分頁(各執行緒平行 pread)
570
+ {
571
+ PROF_T0_x(_th); // 分頁 / SSD 讀取
572
+ for (int i = ith; i < n_need; i += nth) {
573
+ const int32_t e = R->need[(size_t) i];
574
+ const int s = P.acquire(R->il, e, true);
575
+ g_prof.experts++;
576
+ R->need_slot[(size_t) i] = s;
577
+ if (s >= 0) {
578
+ R->slot_of_expert[(size_t) e] = s;
579
+ P.note_used(R->il, e);
580
+ }
581
+ }
582
+ PROF_ADD(read_ns, _th);
583
+ }
584
+ if (nth > 1) { R->bar.sync(nth); }
585
+ const float * cur_p = (const float *) src_cur->data;
586
+ const float * wts_p = (const float *) src_wts->data;
587
+ for (int64_t t = ith; t < n_tok; t += nth) {
588
+ std::vector<uint8_t> scratch;
589
+ for (const auto & ek : R->exp_of_tok[(size_t) t]) {
590
+ const int slot = R->slot_of_expert[(size_t) ek.first];
591
+ if (slot < 0) {
592
+ continue;
593
+ }
594
+ std::vector<std::pair<int32_t, int32_t>> one = {{(int32_t) t, ek.second}};
595
+ expert_forward(L->experts[(size_t) ek.first], P.slot_ptr(slot), cur_p, n_embd,
596
+ one, wts_p, wts_nb0, wts_nb1, (int) n_used,
597
+ R->swiglu_limit, out, &scratch);
598
+ }
599
+ }
600
+ }
601
+
602
+ // 除錯:把某一層的實際輸入/輸出 dump 出來,用 NumPy 對照。
603
+ // 必須先 barrier:上面的歸約是由各執行緒分片寫 out 的,沒有這一道會讀到半成品。
604
+ if (nth > 1) { R->bar.sync(nth); }
605
+ if (const char * dump_path = getenv("SDQ_MOE_DUMP")) {
606
+ const int dump_layer = getenv("SDQ_MOE_DUMP_LAYER") ? atoi(getenv("SDQ_MOE_DUMP_LAYER")) : -1;
607
+ if ((dump_layer < 0 || R->il == dump_layer) && ith == 0) {
608
+ std::string path = std::string(dump_path) + "." + std::to_string(R->il);
609
+ FILE * f = fopen(path.c_str(), "wb");
610
+ if (f) {
611
+ const int32_t meta[6] = {(int32_t) n_embd, (int32_t) n_tok, (int32_t) n_used,
612
+ (int32_t) R->n_expert, 1, 2};
613
+ fwrite(meta, sizeof(int32_t), 6, f);
614
+ fwrite(cur, sizeof(float), (size_t) (n_embd * n_tok), f);
615
+ const uint8_t * ids_base = (const uint8_t *) src_ids->data;
616
+ for (int64_t t = 0; t < n_tok; t++) {
617
+ for (int64_t k = 0; k < n_used; k++) {
618
+ const int32_t e = *(const int32_t *) (ids_base + k * ids_nb0 + (size_t) t * ids_nb1);
619
+ fwrite(&e, sizeof(int32_t), 1, f);
620
+ }
621
+ }
622
+ fwrite(wts, sizeof(float), (size_t) (n_used * n_tok), f);
623
+ fwrite(out, sizeof(float), (size_t) (n_embd * n_tok), f);
624
+ fclose(f);
625
+ fprintf(stderr, "[sdq] dumped layer %d -> %s\n", R->il, path.c_str());
626
+ }
627
+ }
628
+ }
629
+
630
+ if (nth > 1) { R->bar.sync(nth); }
631
+ PROF_ADD(op_ns, _tg);
632
+ }
633
+
634
+ // 輸出計時結果(sdq_shutdown 呼叫)
635
+ void sdq_print_time_profile() {
636
+ if (!prof_on()) {
637
+ return;
638
+ }
639
+ const double ms = 1e6;
640
+ auto v = [ms](const std::atomic<uint64_t> & a) { return a.load() / ms; };
641
+ const double op = v(g_prof.op_ns);
642
+ const double fwd = v(g_prof.fwd_ns);
643
+ fprintf(stderr, "[sdq-prof] 自訂 MoE op(%llu 次呼叫、%llu 個 expert 分頁)\n",
644
+ (unsigned long long) g_prof.calls.load(), (unsigned long long) g_prof.experts.load());
645
+ fprintf(stderr, "[sdq-prof] op 合計 %9.1f ms\n", op);
646
+ fprintf(stderr, "[sdq-prof] expert_forward %9.1f ms (%5.1f%%) 其中:\n", fwd, op > 0 ? 100.0 * fwd / op : 0.0);
647
+ fprintf(stderr, "[sdq-prof] 量化 activation%9.1f ms (%5.1f%% of fwd)\n", v(g_prof.qact_ns), fwd > 0 ? 100.0 * v(g_prof.qact_ns) / fwd : 0.0);
648
+ fprintf(stderr, "[sdq-prof] 三次點積 %9.1f ms (%5.1f%% of fwd)\n", v(g_prof.dot_ns), fwd > 0 ? 100.0 * v(g_prof.dot_ns) / fwd : 0.0);
649
+ fprintf(stderr, "[sdq-prof] SiLU+加權 %9.1f ms (%5.1f%% of fwd)\n", v(g_prof.act_ns), fwd > 0 ? 100.0 * v(g_prof.act_ns) / fwd : 0.0);
650
+ fprintf(stderr, "[sdq-prof] SSD 讀取/分頁 %9.1f ms (%5.1f%%)\n", v(g_prof.read_ns), op > 0 ? 100.0 * v(g_prof.read_ns) / op : 0.0);
651
+ fprintf(stderr, "[sdq-prof] 前置 setup %9.1f ms (%5.1f%%)\n", v(g_prof.setup_ns), op > 0 ? 100.0 * v(g_prof.setup_ns) / op : 0.0);
652
+ fprintf(stderr, "[sdq-prof] barrier 等待 %9.1f ms (%5.1f%%)\n", v(g_prof.bar_ns), op > 0 ? 100.0 * v(g_prof.bar_ns) / op : 0.0);
653
+ }
654
+
655
+ // ---------------------------------------------------------------- graph 建構
656
+
657
+ ggml_tensor * sdq_build_moe(ggml_cgraph * gf, ggml_context * ctx, ggml_tensor * cur,
658
+ ggml_tensor * ids, ggml_tensor * weights, int il, int64_t n_expert,
659
+ int64_t n_expert_used, float swiglu_limit) {
660
+ layer_rec * R = rec_for(il);
661
+ R->n_expert = n_expert;
662
+ R->n_expert_used = n_expert_used;
663
+ R->swiglu_limit = swiglu_limit;
664
+
665
+ // 呼叫端給的是 2D [n_embd, n_tokens],這裡跟上游一樣先 reshape 成 3D
666
+ // [n_embd, 1, n_tokens](上游 mul_mat_id 路徑也是這樣拿 token 數的)
667
+ ggml_tensor * cur3 = (cur->ne[1] == 1) ? cur : ggml_reshape_3d(ctx, cur, cur->ne[0], 1, cur->ne[1]);
668
+ const int64_t n_tok = cur3->ne[2];
669
+
670
+ ggml_tensor * args[3] = { cur3, ids, weights };
671
+ ggml_tensor * out = ggml_custom_4d(ctx, GGML_TYPE_F32, cur3->ne[0], n_tok, 1, 1,
672
+ args, 3, moe_custom_op, GGML_N_TASKS_MAX, R);
673
+ ggml_set_name(out, "sdq_moe_out");
674
+ // 關鍵:把這個節點註冊進圖,ggml 才會為它配置獨立的 compute buffer。
675
+ // 少了這一行,layer>=1 的輸出會被其他張量覆蓋(layer 0 僥倖正確)。
676
+ ggml_build_forward_expand(gf, out);
677
+ return out;
678
+ }
679
+
680
+ } // namespace sdq
681
+ namespace sdq {
682
+ // 自我測試:對指定 (layer, expert) 用 pager 的資料算一次 MoE 輸出,
683
+ // 讓 Python 用純 NumPy 從同一段 GGUF 位元組重算並比對。
684
+ // 輸出:out[0..n_embd) = 輸出,out[n_embd..2*n_embd) = 輸入 activation
685
+ int sdq_moe_selftest(int il, int ie, const float * x, float * out_io,
686
+ int64_t n_embd, int64_t n_ff, int n_tok) {
687
+ pager & P = pager::instance();
688
+ const layer_desc * L = P.layer(il);
689
+ if (!L || ie < 0 || ie >= (int) L->experts.size()) {
690
+ fprintf(stderr, "[sdq-selftest] 層/expert 不存在:%d/%d\n", il, ie);
691
+ return 1;
692
+ }
693
+ const int slot = P.acquire(il, ie, true);
694
+ if (slot < 0) {
695
+ fprintf(stderr, "[sdq-selftest] 分頁失敗\n");
696
+ return 2;
697
+ }
698
+ const expert_desc & ed = L->experts[(size_t) ie];
699
+ if (ed.ne0 != n_embd || ed.ne1 != n_ff) {
700
+ fprintf(stderr, "[sdq-selftest] 維度不符:ne0=%lld ne1=%lld(預期 %lld/%lld)\n",
701
+ (long long) ed.ne0, (long long) ed.ne1, (long long) n_embd, (long long) n_ff);
702
+ return 3;
703
+ }
704
+ std::vector<std::pair<int32_t, int32_t>> toks;
705
+ std::vector<float> wts((size_t) n_tok, 1.0f);
706
+ for (int t = 0; t < n_tok; t++) {
707
+ toks.emplace_back(t, 0);
708
+ }
709
+ std::vector<float> out((size_t) n_embd * n_tok, 0.0f);
710
+ std::vector<uint8_t> scratch;
711
+ // SDQ_SELFTEST_DUMP: 把 arena 槽位的原始位元組存檔,驗證 pager 讀到的是正確的 expert
712
+ if (const char * dp = getenv("SDQ_SELFTEST_DUMP")) {
713
+ std::string path = std::string(dp) + "." + std::to_string(il) + "." + std::to_string(ie);
714
+ FILE * f = fopen(path.c_str(), "wb");
715
+ if (f) {
716
+ fwrite(P.slot_ptr(slot), 1, (size_t) ed.bytes, f);
717
+ fclose(f);
718
+ fprintf(stderr, "[sdq-selftest] slot dump -> %s (%lld B)\n", path.c_str(), (long long) ed.bytes);
719
+ }
720
+ }
721
+ expert_forward(ed, P.slot_ptr(slot), x, n_embd, toks, wts.data(), 4, 4, 1, 0.0f,
722
+ out.data(), &scratch);
723
+ for (int64_t t = 0; t < n_tok; t++) {
724
+ for (int64_t i = 0; i < n_embd; i++) {
725
+ out_io[t * n_embd + i] = out[(size_t) t * n_embd + i];
726
+ out_io[n_tok * n_embd + t * n_embd + i] = x[(size_t) t * n_embd + i];
727
+ }
728
+ }
729
+ P.unpin_all();
730
+ fprintf(stderr, "[sdq-selftest] L%d E%d n_tok=%d 計算完成(gate %s / up %s / down %s)\n",
731
+ il, ie, n_tok, ggml_type_name(ed.type[part::GATE]), ggml_type_name(ed.type[part::UP]),
732
+ ggml_type_name(ed.type[part::DOWN]));
733
+ return 0;
734
+ }
735
+ } // namespace sdq
sdq-ling/sdq_pager.cpp ADDED
@@ -0,0 +1,2292 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ // SDQ expert pager 實作:見 sdq_pager.h 的設計說明
2
+ #include "sdq_pager.h"
3
+
4
+ #include "llama-mmap.h"
5
+
6
+ #include <algorithm>
7
+ #include <thread>
8
+ #include <cinttypes>
9
+ #include <cmath>
10
+ #include <cstdio>
11
+ #include <cstdlib>
12
+ #include <cstring>
13
+ #include <fcntl.h>
14
+ #include <ctime>
15
+ #include <fstream>
16
+ #include <map>
17
+ #include <set>
18
+ #include <sstream>
19
+ #include <stdexcept>
20
+ #include <sys/mman.h>
21
+ #include <sys/stat.h>
22
+ #include <sys/types.h>
23
+ #include <unistd.h>
24
+
25
+ namespace sdq {
26
+
27
+ // ============================================================ 小工具
28
+
29
+ uint64_t now_us() {
30
+ struct timespec ts;
31
+ clock_gettime(CLOCK_MONOTONIC, &ts);
32
+ return (uint64_t) ts.tv_sec * 1000000ull + (uint64_t) ts.tv_nsec / 1000ull;
33
+ }
34
+
35
+ size_t process_rss_bytes() {
36
+ std::ifstream f("/proc/self/statm");
37
+ size_t total = 0, resident = 0;
38
+ if (f >> total >> resident) {
39
+ return resident * (size_t) sysconf(_SC_PAGESIZE);
40
+ }
41
+ return 0;
42
+ }
43
+
44
+ size_t physical_ram_bytes() {
45
+ std::ifstream f("/proc/meminfo");
46
+ std::string key;
47
+ size_t value = 0;
48
+ std::string unit;
49
+ while (f >> key >> value >> unit) {
50
+ if (key == "MemTotal:") {
51
+ return value * 1024;
52
+ }
53
+ }
54
+ return 0;
55
+ }
56
+
57
+ static size_t env_size(const char * name, size_t dflt) {
58
+ const char * v = getenv(name);
59
+ if (!v || !*v) {
60
+ return dflt;
61
+ }
62
+ return (size_t) strtoull(v, nullptr, 10);
63
+ }
64
+
65
+ static bool env_flag(const char * name, bool dflt) {
66
+ const char * v = getenv(name);
67
+ if (!v || !*v) {
68
+ return dflt;
69
+ }
70
+ return !(v[0] == '0' && v[1] == '\0');
71
+ }
72
+
73
+ // ============================================================ GGUF 檔頭解析
74
+
75
+ namespace {
76
+
77
+ struct gguf_reader {
78
+ const uint8_t * p = nullptr;
79
+ size_t size = 0;
80
+ size_t o = 0;
81
+
82
+ uint8_t u8() { return p[o++]; }
83
+ uint32_t u32() { uint32_t v; memcpy(&v, p + o, 4); o += 4; return v; }
84
+ uint64_t u64() { uint64_t v; memcpy(&v, p + o, 8); o += 8; return v; }
85
+ int64_t i64() { int64_t v; memcpy(&v, p + o, 8); o += 8; return v; }
86
+ float f32() { float v; memcpy(&v, p + o, 4); o += 4; return v; }
87
+ std::string str() {
88
+ uint64_t n = u64();
89
+ std::string s((const char *) p + o, (size_t) n);
90
+ o += n;
91
+ return s;
92
+ }
93
+ void skip(uint64_t n) { o += (size_t) n; }
94
+ uint16_t u16() { uint16_t v; memcpy(&v, p + o, 2); o += 2; return v; }
95
+ int16_t i16() { int16_t v; memcpy(&v, p + o, 2); o += 2; return v; }
96
+ double f64() { double v; memcpy(&v, p + o, 8); o += 8; return v; }
97
+ };
98
+
99
+ struct kv_value {
100
+ enum kind { NONE, INT, REAL, STR, ARR } k = NONE;
101
+ int64_t i = 0;
102
+ double f = 0;
103
+ std::string s;
104
+ std::vector<int64_t> arr;
105
+ };
106
+
107
+ struct tensor_info {
108
+ std::string name;
109
+ std::vector<int64_t> dims;
110
+ ggml_type type = GGML_TYPE_COUNT;
111
+ uint64_t offset = 0; // 在所屬 shard 檔案內、相對 data_start 的偏移
112
+ int shard = 0; // 屬於哪一個 shard(sharded GGUF)
113
+ };
114
+
115
+ size_t ggml_type_size_of(ggml_type t) {
116
+ return ggml_row_size(t, 1);
117
+ }
118
+
119
+ } // namespace
120
+
121
+ // ============================================================ pager
122
+
123
+ pager & pager::instance() {
124
+ static pager p;
125
+ return p;
126
+ }
127
+
128
+ // 依 GGUF split 命名規則把第一個 shard 路徑展開成全部 shard。
129
+ // 規則是 "<prefix>-00001-of-00004.gguf";llama.cpp 用 llama_split_prefix/llama_split_path
130
+ // 做同一件事,這裡獨立實作以免 pager 依賴 loader 的內部符號。
131
+ std::vector<std::string> pager::list_splits(const std::string & first_path, int n_split) {
132
+ std::vector<std::string> out;
133
+ if (n_split <= 1) {
134
+ out.push_back(first_path);
135
+ return out;
136
+ }
137
+ // 用 llama.cpp 自己的命名規則(%s-%05d-of-%05d.gguf,且 split_no 由 1 起算)。
138
+ // 自己刻字串很容易差一位數字 —— 實測就是這樣變成 "...XL--00001-of-..."。
139
+ std::vector<char> pre(llama_path_max(), 0);
140
+ const int pre_len = llama_split_prefix(pre.data(), pre.size(), first_path.c_str(), 0, n_split);
141
+ if (pre_len <= 0) {
142
+ throw std::runtime_error("無法解析 split 檔名:" + first_path);
143
+ }
144
+ std::vector<char> buf(llama_path_max(), 0);
145
+ for (int i = 0; i < n_split; i++) {
146
+ const int len = llama_split_path(buf.data(), buf.size(), pre.data(), i, n_split);
147
+ if (len <= 0) {
148
+ throw std::runtime_error("無法產生 split 路徑:" + first_path);
149
+ }
150
+ out.emplace_back(buf.data(), (size_t) len);
151
+ }
152
+ return out;
153
+ }
154
+
155
+ void pager::load_gguf_header(const std::string & path) {
156
+ const int fd = open(path.c_str(), O_RDONLY);
157
+ if (fd < 0) {
158
+ throw std::runtime_error("無法開啟 GGUF:" + path);
159
+ }
160
+
161
+ struct stat st;
162
+ fstat(fd, &st);
163
+
164
+ // 只讀檔頭:metadata + tensor 資訊大約 11 MB(shard 1 還含整份 tokenizer,
165
+ // 這裡的 64 MiB 上限對它足夠)
166
+ const size_t want = (size_t) std::min<int64_t>(st.st_size, 64ull << 20);
167
+ std::vector<uint8_t> buf(want);
168
+ ssize_t got = 0;
169
+ while (got < (ssize_t) want) {
170
+ ssize_t n = pread(fd, buf.data() + got, want - got, got);
171
+ if (n <= 0) {
172
+ break;
173
+ }
174
+ got += n;
175
+ }
176
+ buf.resize((size_t) std::max<ssize_t>(got, 0));
177
+
178
+ gguf_reader r{buf.data(), buf.size(), 0};
179
+ if (memcmp(buf.data(), "GGUF", 4) != 0) {
180
+ close(fd);
181
+ throw std::runtime_error("不是 GGUF 檔:" + path);
182
+ }
183
+ r.o = 4;
184
+ const uint32_t version = r.u32();
185
+ if (version < 2 || version > 3) {
186
+ close(fd);
187
+ throw std::runtime_error("不支援的 GGUF 版本");
188
+ }
189
+ const uint64_t n_tensors = r.u64();
190
+ const uint64_t n_kv = r.u64();
191
+
192
+ std::map<std::string, kv_value> kv;
193
+ for (uint64_t i = 0; i < n_kv; i++) {
194
+ std::string key = r.str();
195
+ const uint32_t t = r.u32();
196
+ kv_value v;
197
+ switch (t) {
198
+ // GGUF 型別寬度:U8/I8/BOOL=1、U16/I16=2、U32/I32/F32=4、U64/I64/F64=8。
199
+ // 舊版把 case 0/1/2/3 都讀成 u32,遇到 tokenizer 或 U16 metadata 就整個
200
+ // header 錯位(症狀是 std::length_error: basic_string::_M_create)。
201
+ case 0: v.k = kv_value::INT; v.i = (int64_t) r.u8(); v.f = (double) v.i; break;
202
+ case 1: v.k = kv_value::INT; v.i = (int64_t) (int8_t) r.u8(); v.f = (double) v.i; break;
203
+ case 2: v.k = kv_value::INT; v.i = (int64_t) r.u16(); v.f = (double) v.i; break;
204
+ case 3: v.k = kv_value::INT; v.i = (int64_t) r.i16(); v.f = (double) v.i; break;
205
+ case 4: v.k = kv_value::INT; v.i = (int64_t) (uint32_t) r.u32(); v.f = (double) v.i; break;
206
+ case 5: v.k = kv_value::INT; v.i = (int64_t) (int32_t) r.u32(); v.f = (double) v.i; break;
207
+ case 6: v.k = kv_value::REAL; v.f = r.f32(); v.i = (int64_t) v.f; break;
208
+ case 7: v.k = kv_value::INT; v.i = (int64_t) r.u8(); v.f = (double) v.i; break; // BOOL = 1 byte
209
+ case 8: v.k = kv_value::STR; v.s = r.str(); break;
210
+ case 9: {
211
+ v.k = kv_value::ARR;
212
+ const uint32_t et = r.u32();
213
+ const uint64_t n = r.u64();
214
+ for (uint64_t j = 0; j < n; j++) {
215
+ switch (et) {
216
+ // 元素寬度必須跟 kv 一致(U8/I8/BOOL=1、U16/I16=2 …)。
217
+ // F64 一定要消費 8 bytes,舊版沒讀會讓 header 往後全部錯位。
218
+ case 0: v.arr.push_back(r.u8()); break;
219
+ case 1: v.arr.push_back((int8_t) r.u8()); break;
220
+ case 2: v.arr.push_back(r.u16()); break;
221
+ case 3: v.arr.push_back(r.i16()); break;
222
+ case 4: v.arr.push_back((uint32_t) r.u32()); break;
223
+ case 5: v.arr.push_back((int32_t) r.u32()); break;
224
+ case 6: v.arr.push_back((int64_t) r.f32()); break;
225
+ case 7: v.arr.push_back(r.u8()); break;
226
+ case 10: v.arr.push_back((int64_t) r.u64()); break;
227
+ case 11: v.arr.push_back(r.i64()); break;
228
+ case 12: r.f64(); v.arr.push_back(0); break;
229
+ case 8: r.skip(r.u64()); v.arr.push_back(0); break;
230
+ default:
231
+ close(fd);
232
+ throw std::runtime_error("GGUF 陣列元素型別不支援: " + std::to_string(et));
233
+ }
234
+ }
235
+ } break;
236
+ case 10: v.k = kv_value::INT; v.i = (int64_t) r.u64(); v.f = (double) v.i; break;
237
+ case 11: v.k = kv_value::INT; v.i = r.i64(); v.f = (double) v.i; break;
238
+ case 12: v.k = kv_value::REAL; { double d; memcpy(&d, buf.data() + r.o, 8); r.o += 8; v.f = d; v.i = (int64_t) d; } break;
239
+ default: close(fd); throw std::runtime_error("GGUF kv 型別不支援: " + std::to_string(t));
240
+ }
241
+ kv[key] = v;
242
+ }
243
+
244
+ std::vector<tensor_info> tensors;
245
+ tensors.reserve(n_tensors);
246
+ for (uint64_t i = 0; i < n_tensors; i++) {
247
+ tensor_info ti;
248
+ ti.name = r.str();
249
+ const uint32_t nd = r.u32();
250
+ for (uint32_t d = 0; d < nd; d++) {
251
+ ti.dims.push_back((int64_t) r.u64());
252
+ }
253
+ ti.type = (ggml_type) r.u32();
254
+ ti.offset = r.u64();
255
+ tensors.push_back(std::move(ti));
256
+ }
257
+ const uint64_t align = kv.count("general.alignment") ? (uint64_t) kv["general.alignment"].i : 32;
258
+ const uint64_t data_start = (r.o + align - 1) / align * align;
259
+
260
+ close(fd);
261
+
262
+ // ---- sharded GGUF:shard 1 只帶 metadata(n_tensors=0),真正的權重分散在
263
+ // 其他 shard,而且每個 shard 有自己的 data_start。舊版只讀第一個檔,
264
+ // 於是 page map 會是空的(症狀:pager 起來但沒有任何 expert 可分頁)。
265
+ // 這裡把每個 shard 的 header 都讀一遍,張量名稱為全域鍵,offset 記成 (shard, off)。
266
+ {
267
+ const int64_t n_split = kv.count("split.count") ? kv["split.count"].i : 1;
268
+ shard_paths_ = list_splits(path, (int) n_split);
269
+ shard_data_start_.assign(shard_paths_.size(), 0);
270
+ shard_data_start_[0] = data_start;
271
+
272
+ if (n_split > 1) {
273
+ std::vector<tensor_info> merged;
274
+ merged.reserve((size_t) n_tensors + 1024);
275
+ for (size_t si = 1; si < shard_paths_.size(); si++) {
276
+ const int sfd = open(shard_paths_[si].c_str(), O_RDONLY);
277
+ if (sfd < 0) {
278
+ throw std::runtime_error("無法開啟 GGUF shard:" + shard_paths_[si]);
279
+ }
280
+ struct stat sst;
281
+ fstat(sfd, &sst);
282
+ // 後續 shard 的 header 很小(沒有 tokenizer),讀 16 MiB 綽綽有餘
283
+ const size_t swant = (size_t) std::min<int64_t>(sst.st_size, 16ull << 20);
284
+ std::vector<uint8_t> sbuf(swant);
285
+ ssize_t sgot = 0;
286
+ while (sgot < (ssize_t) swant) {
287
+ const ssize_t n = pread(sfd, sbuf.data() + sgot, swant - sgot, sgot);
288
+ if (n <= 0) {
289
+ break;
290
+ }
291
+ sgot += n;
292
+ }
293
+ sbuf.resize((size_t) std::max<ssize_t>(sgot, 0));
294
+
295
+ gguf_reader sr{sbuf.data(), sbuf.size(), 0};
296
+ if (sbuf.size() < 16 || memcmp(sbuf.data(), "GGUF", 4) != 0) {
297
+ close(sfd);
298
+ throw std::runtime_error("shard 不是 GGUF:" + shard_paths_[si]);
299
+ }
300
+ sr.o = 4;
301
+ sr.u32(); // version
302
+ const uint64_t s_ntensors = sr.u64();
303
+ const uint64_t s_nkv = sr.u64();
304
+ for (uint64_t i = 0; i < s_nkv; i++) {
305
+ sr.str(); // key
306
+ const uint32_t t = sr.u32();
307
+ switch (t) {
308
+ case 0: case 1: case 7: sr.skip(1); break;
309
+ case 2: case 3: sr.skip(2); break;
310
+ case 4: case 5: case 6: sr.skip(4); break;
311
+ case 10: case 11: case 12: sr.skip(8); break;
312
+ case 8: sr.skip(sr.u64()); break;
313
+ case 9: {
314
+ const uint32_t et = sr.u32();
315
+ const uint64_t n = sr.u64();
316
+ if (et == 8) {
317
+ for (uint64_t j = 0; j < n; j++) { sr.skip(sr.u64()); }
318
+ } else {
319
+ size_t esz = 0;
320
+ switch (et) {
321
+ case 0: case 1: case 7: esz = 1; break;
322
+ case 2: case 3: esz = 2; break;
323
+ case 4: case 5: case 6: esz = 4; break;
324
+ case 10: case 11: case 12: esz = 8; break;
325
+ default: esz = 0; break;
326
+ }
327
+ sr.skip((uint64_t) n * esz);
328
+ }
329
+ } break;
330
+ default:
331
+ close(sfd);
332
+ throw std::runtime_error("shard kv 型別不支援: " + std::to_string(t));
333
+ }
334
+ }
335
+ for (uint64_t i = 0; i < s_ntensors; i++) {
336
+ tensor_info ti;
337
+ ti.name = sr.str();
338
+ const uint32_t nd = sr.u32();
339
+ for (uint32_t d = 0; d < nd; d++) {
340
+ ti.dims.push_back((int64_t) sr.u64());
341
+ }
342
+ ti.type = (ggml_type) sr.u32();
343
+ ti.offset = sr.u64();
344
+ ti.shard = (int) si;
345
+ merged.push_back(std::move(ti));
346
+ }
347
+ shard_data_start_[si] = (sr.o + align - 1) / align * align;
348
+ close(sfd);
349
+ }
350
+ for (auto & ti : tensors) {
351
+ ti.shard = 0;
352
+ merged.push_back(std::move(ti));
353
+ }
354
+ tensors.swap(merged);
355
+ fprintf(stderr, "[sdq] sharded GGUF:%d 個 shard,合計 %zu 個張量(shard1 為 metadata-only)\n",
356
+ (int) shard_paths_.size(), tensors.size());
357
+ }
358
+ }
359
+
360
+ // ---- hparams
361
+ auto gi = [&](const std::string & k, int64_t dflt) -> int64_t {
362
+ auto it = kv.find(k);
363
+ return it == kv.end() ? dflt : it->second.i;
364
+ };
365
+ info_.arch = kv.count("general.architecture") ? kv["general.architecture"].s : "";
366
+ info_.name = kv.count("general.name") ? kv["general.name"].s : "";
367
+ const std::string pfx = info_.arch + ".";
368
+ info_.n_layer = gi(pfx + "block_count", 0);
369
+ info_.n_expert = gi(pfx + "expert_count", 0);
370
+ info_.n_expert_used= gi(pfx + "expert_used_count", 0);
371
+ info_.n_embd = gi(pfx + "embedding_length", 0);
372
+ info_.n_ff_exp = gi(pfx + "expert_feed_forward_length", 0);
373
+ info_.n_ff_shexp = gi(pfx + "expert_shared_feed_forward_length", 0);
374
+ info_.n_head = gi(pfx + "attention.head_count", 0);
375
+ info_.n_head_kv = gi(pfx + "attention.head_count_kv", 0);
376
+ info_.n_embd_head_k= gi(pfx + "attention.key_length", 0);
377
+ info_.n_embd_head_v= gi(pfx + "attention.value_length", info_.n_embd_head_k);
378
+ info_.context_length = gi(pfx + "context_length", 0);
379
+ info_.full_attention_interval = gi(pfx + "full_attention_interval", 4);
380
+ // bailingmoe3(Ling-3.0-flash):前 n_layer_dense_lead 層是 dense FFN,
381
+ // 它們沒有 ffn_*_exps 張量,必須讓 page map 保持「空」——
382
+ // sdq_pager_active_for_layer() 看到 layer() == nullptr 就會走原生路徑。
383
+ info_.n_layer_dense_lead = gi(pfx + "leading_dense_block_count", 0);
384
+ info_.n_expert_groups = gi(pfx + "expert_group_count", 0);
385
+ info_.n_expert_group_used = gi(pfx + "expert_group_used_count", 0);
386
+
387
+ if (info_.n_expert <= 0 || info_.n_layer <= 0) {
388
+ throw std::runtime_error("這不是 MoE 模型(expert_count=" + std::to_string(info_.n_expert) + ")");
389
+ }
390
+
391
+ // ---- 頁面表:每個 (layer, expert) 的三段
392
+ layers_.assign((size_t) info_.n_layer, {});
393
+ std::map<std::string, const tensor_info *> by_name;
394
+ for (const auto & ti : tensors) {
395
+ by_name[ti.name] = &ti;
396
+ }
397
+
398
+ size_t max_expert_bytes = 0;
399
+ int n_moe_layers = 0;
400
+ for (int64_t il = 0; il < info_.n_layer; il++) {
401
+ const std::string b = "blk." + std::to_string(il) + ".";
402
+ layer_desc & L = layers_[(size_t) il];
403
+
404
+ auto get = [&](const std::string & suffix) -> const tensor_info * {
405
+ auto it = by_name.find(b + suffix);
406
+ return it == by_name.end() ? nullptr : it->second;
407
+ };
408
+
409
+ const tensor_info * t_gate = get("ffn_gate_exps.weight");
410
+ const tensor_info * t_up = get("ffn_up_exps.weight");
411
+ const tensor_info * t_gu = get("ffn_gate_up_exps.weight");
412
+ const tensor_info * t_down = get("ffn_down_exps.weight");
413
+
414
+ if (!t_down || (t_gate && !t_up && !t_gu)) {
415
+ // 這層不是 MoE(bailingmoe3 的 leading dense layers 就是這樣)。
416
+ // 保持 experts 為空 → layer() 回 nullptr → 這層走原生 mul_mat_id。
417
+ continue;
418
+ }
419
+ L.experts.assign((size_t) info_.n_expert, {});
420
+ n_moe_layers++;
421
+
422
+ // sharded GGUF:每段的絕對偏移要用「該張量自己 shard 的 data_start」,
423
+ // 不能用第一個 shard 的(兩者差了近 10 MB,直接用會讀到別的權重)。
424
+ auto abs_off = [&](const tensor_info * t, uint64_t rel) -> uint64_t {
425
+ const size_t s = (size_t) t->shard;
426
+ const uint64_t base = (s < shard_data_start_.size()) ? shard_data_start_[s] : 0;
427
+ return base + rel;
428
+ };
429
+
430
+ for (int64_t e = 0; e < info_.n_expert; e++) {
431
+ expert_desc & ed = L.experts[(size_t) e];
432
+ if (t_gate || t_gu) {
433
+ const tensor_info * t = t_gate ? t_gate : t_gu;
434
+ const uint64_t nb = (uint64_t) ggml_row_size(t->type, t->dims[0]) * (uint64_t) t->dims[1];
435
+ ed.seg[part::GATE].off = abs_off(t, t->offset + (uint64_t) e * nb);
436
+ ed.seg[part::GATE].size = nb;
437
+ ed.seg[part::GATE].shard = t->shard;
438
+ ed.ne0 = t->dims[0];
439
+ ed.ne1 = t->dims[1];
440
+ ed.type[part::GATE] = t->type;
441
+ }
442
+ if (t_up) {
443
+ const uint64_t nb = (uint64_t) ggml_row_size(t_up->type, t_up->dims[0]) * (uint64_t) t_up->dims[1];
444
+ ed.seg[part::UP].off = abs_off(t_up, t_up->offset + (uint64_t) e * nb);
445
+ ed.seg[part::UP].size = nb;
446
+ ed.seg[part::UP].shard = t_up->shard;
447
+ ed.type[part::UP] = t_up->type; // gate/up 都是 Q4_K,型別一定要各自填
448
+ if (!t_gate) {
449
+ ed.ne0 = t_up->dims[0];
450
+ ed.ne1 = t_up->dims[1];
451
+ ed.type[part::GATE] = t_up->type;
452
+ }
453
+ }
454
+ {
455
+ const uint64_t nb = (uint64_t) ggml_row_size(t_down->type, t_down->dims[0]) * (uint64_t) t_down->dims[1];
456
+ ed.seg[part::DOWN].off = abs_off(t_down, t_down->offset + (uint64_t) e * nb);
457
+ ed.seg[part::DOWN].size = nb;
458
+ ed.seg[part::DOWN].shard = t_down->shard;
459
+ ed.type[part::DOWN] = t_down->type;
460
+ ed.down_ne0 = t_down->dims[0];
461
+ ed.down_ne1 = t_down->dims[1];
462
+ }
463
+ ed.fused_gate_up = (t_gate == nullptr && t_gu != nullptr);
464
+ ed.bytes = 0;
465
+ int64_t acc = 0;
466
+ for (int p = 0; p < 3; p++) {
467
+ if (ed.seg[p].size == 0) {
468
+ continue;
469
+ }
470
+ acc = (acc + 63) & ~(int64_t) 63; // 每段 64B 對齊
471
+ ed.slot_off[p] = acc;
472
+ acc += (int64_t) ed.seg[p].size;
473
+ }
474
+ ed.bytes = acc;
475
+ max_expert_bytes = std::max(max_expert_bytes, (size_t) ed.bytes);
476
+ }
477
+ L.bytes_per_expert = L.experts.empty() ? 0 : L.experts[0].bytes;
478
+ }
479
+
480
+ slot_bytes_ = (max_expert_bytes + 4095) & ~(size_t) 4095;
481
+ if (slot_bytes_ < (256ull << 10)) {
482
+ slot_bytes_ = 256ull << 10;
483
+ }
484
+
485
+ // ---- 大小兩種槽位 ----------------------------------------------------
486
+ // 這個模型有 37 層的 expert 是 1900544 B(Q4_K + Q5_K),但層 34/38/39 是
487
+ // 2039808 B(down 換成 Q6_K)。若所有槽位都照最大的配,arena 有 6.8% 被浪費
488
+ // ——而 arena 就是 8 GB 預算裡最寶貴的資源(實測 +5% 槽位 ≈ +9% tok/s)。
489
+ //
490
+ // 解法:槽位分成兩區。「小槽位」給 37 層用,「大槽位」只給那 3 層用,
491
+ // 大槽位的數量按「那 3 層佔請求的比例」分配(3/40 = 7.5%)。
492
+ // 選「大多數層」當小的:用眾數而不是最大值,這樣即使某個模型只有一種大小,
493
+ // 也不會誤判。
494
+ {
495
+ std::map<size_t, int> hist;
496
+ for (const auto & L : layers_) {
497
+ if (L.experts.empty()) {
498
+ continue; // dense layer:沒有 expert,不參與槽位尺寸統計
499
+ }
500
+ hist[(size_t) L.bytes_per_expert]++;
501
+ }
502
+ size_t small = 0;
503
+ int best_n = -1;
504
+ for (const auto & kv : hist) {
505
+ if (kv.second > best_n) {
506
+ best_n = kv.second;
507
+ small = kv.first;
508
+ }
509
+ }
510
+ size_t big = 0;
511
+ for (const auto & L : layers_) {
512
+ if (!L.experts.empty() && (size_t) L.bytes_per_expert > small) {
513
+ big = std::max(big, (size_t) L.bytes_per_expert);
514
+ }
515
+ }
516
+ small_bytes_ = (small + 4095) & ~(size_t) 4095;
517
+ big_bytes_ = big ? ((big + 4095) & ~(size_t) 4095) : 0;
518
+ if (big_bytes_ && big_bytes_ <= small_bytes_) {
519
+ big_bytes_ = 0; // 其實一樣大,別分兩區
520
+ }
521
+ slot_bytes_ = big_bytes_ ? big_bytes_ : small_bytes_;
522
+ int n_big_layers = 0;
523
+ for (auto & L : layers_) {
524
+ L.needs_big_slot = big_bytes_ && !L.experts.empty() &&
525
+ (size_t) L.bytes_per_expert > small_bytes_;
526
+ if (L.needs_big_slot) {
527
+ n_big_layers++;
528
+ }
529
+ }
530
+ // 統計給 README / 實驗報告用:expert 權重在檔案裡到底佔多少
531
+ uint64_t expert_bytes = 0;
532
+ for (const auto & L : layers_) {
533
+ expert_bytes += (uint64_t) L.bytes_per_expert * (uint64_t) L.experts.size();
534
+ }
535
+ n_moe_layers_ = n_moe_layers;
536
+ expert_bytes_total_ = expert_bytes;
537
+ fprintf(stderr,
538
+ "[sdq] MoE 層 %d / 共 %lld 層(dense lead %lld)"
539
+ ":expert 權重合計 %.2f GiB\n",
540
+ n_moe_layers, (long long) info_.n_layer,
541
+ (long long) info_.n_layer_dense_lead,
542
+ expert_bytes / 1073741824.0);
543
+ fprintf(stderr,
544
+ "[sdq] 槽位大小:小的 %zu B(%zu 層)/大的 %zu B(%d 層)\n",
545
+ small_bytes_, n_moe_layers - n_big_layers, big_bytes_, n_big_layers);
546
+ }
547
+ }
548
+
549
+ bool pager::need_big_slot(int il, int ie) const {
550
+ if (!big_bytes_) {
551
+ return false;
552
+ }
553
+ const layer_desc * L = layer(il);
554
+ if (!L) {
555
+ return false;
556
+ }
557
+ if (L->needs_big_slot) {
558
+ return true;
559
+ }
560
+ return ie >= 0 && ie < (int) L->experts.size() &&
561
+ (size_t) L->experts[(size_t) ie].bytes > small_bytes_;
562
+ }
563
+
564
+ void pager::alloc_arena() {
565
+ const size_t slot = cfg_.slot_bytes ? cfg_.slot_bytes : slot_bytes_;
566
+ slot_bytes_ = slot;
567
+
568
+ // arena 大小決定:RAM 是「總預算」,不是「還有多少可用」
569
+ // 總預算 = 非專家權重 + ggml 模型緩衝 + KV cache + 計算緩衝 + 詞表 + arena
570
+ // 所以有指定 ram_budget 時,用「目前 RSS(不含 arena)」當作已用額度扣掉。
571
+ //
572
+ // 重要:pager 在 llama.cpp 載入權重**之前**就做 sizing,此時 rss_now 只有
573
+ // 幾十 MB(實測 0.09 GiB),所以後續所有匿名配置都不在 rss_now 裡,
574
+ // 必須由 reserve 手動涵蓋。這些事實上是三塊:
575
+ //
576
+ // 1. ggml 的模型緩衝 —— 主機有 AMX 時(-march=native 會定義 __AMX_INT8__
577
+ // 使 llama.cpp 啟用 AMX repack buffer)Ling-3.0-flash Q4_K_M 要
578
+ // 1876.34 MiB;關掉 AMX(GGML_NATIVE=OFF)則不需要這塊。
579
+ // 2. CPU compute buffer:83.5 MiB @ubatch=128,ubatch 越大越多。
580
+ // 3. KV/SSM state + 執行緒堆疊:ctx 越大越多。
581
+ //
582
+ // ��的預設值 900 MB 只涵蓋 (2)+(3),漏掉 (1),在 AMX 主機上會讓實際
583
+ // 匿名用量超出預算約 1.9 GiB —— 嚴格 RLIMIT_DATA 下直接 OOM。
584
+ // 實測(本機,ctx=2048、ubatch=128、AMX build):
585
+ // reserve 1024 MB → 預算 8 GiB 實測匿名峰值 10.96 GiB(超出,OOM)
586
+ // reserve 3600 MB → 預算 8 GiB 通過,arena 4.40 GiB
587
+ // reserve 2700 MB → 預算 4 GiB 通過,arena 1.27 GiB(無 AMX build)
588
+ // 寧可少一些槽位,也不要在目標機器的 8 GiB 上 OOM —— 這是本專案的硬約束。
589
+ size_t reserve = (size_t) env_size("SDQ_RESERVE_MB", 3600) << 20;
590
+ const size_t rss_now = process_rss_bytes();
591
+
592
+ size_t arena;
593
+ if (cfg_.ram_budget_bytes) {
594
+ const size_t used = rss_now; // 這時候 arena 還沒配置
595
+ size_t avail = cfg_.ram_budget_bytes > used + reserve
596
+ ? cfg_.ram_budget_bytes - used - reserve
597
+ : 0;
598
+ arena = avail;
599
+ if (cfg_.arena_bytes) {
600
+ arena = std::min(arena, cfg_.arena_bytes);
601
+ }
602
+ fprintf(stderr, "[sdq] RAM 預算 %.2f GiB:已用(RSS) %.2f GiB、預留 %.2f GiB → arena %.2f GiB\n",
603
+ cfg_.ram_budget_bytes / 1073741824.0, used / 1073741824.0,
604
+ reserve / 1073741824.0, arena / 1073741824.0);
605
+ } else {
606
+ const size_t phys = physical_ram_bytes();
607
+ const size_t budget = (size_t) (phys * 0.70);
608
+ arena = cfg_.arena_bytes ? cfg_.arena_bytes : std::min(budget, (size_t) (4ull << 30));
609
+ }
610
+ if (arena < slot_bytes_ * 16) {
611
+ throw std::runtime_error("expert arena 太小");
612
+ }
613
+ // 切成兩區:大槽位只分給「需要大槽位的層」,比例 = 那些層佔全部層的比例。
614
+ const bool force_uniform = env_flag("SDQ_FORCE_UNIFORM_SLOT", false);
615
+ if (big_bytes_ && small_bytes_ && !layers_.empty() && !force_uniform) {
616
+ int n_big_layers = 0;
617
+ for (const auto & L : layers_) {
618
+ if (L.needs_big_slot) {
619
+ n_big_layers++;
620
+ }
621
+ }
622
+ const double denom = (double) (n_moe_layers_ ? n_moe_layers_ : layers_.size());
623
+ const double f = (double) n_big_layers / denom;
624
+ const double avg = f * (double) big_bytes_ + (1.0 - f) * (double) small_bytes_;
625
+ size_t n_total = (size_t)((double) arena / avg);
626
+ size_t nb = (size_t)(n_total * f);
627
+ // 大槽位的下限不是「有點就好」,而是**工作集的下限**:
628
+ // 一個 token 在每個大槽位層要 8 個大 expert,這個模型有 3 個大槽位層
629
+ // → 至少要 3 × 8 = 24 個大槽位,而且還要留給預取的空間。
630
+ // 只給 16 個時量到過「找不到能放 il=34 ie=114 的槽位」而中止
631
+ // (這是在 verify 新增的預取壓力測試 [2b] 抓到的)。
632
+ const size_t big_need = (size_t) (n_big_layers * info_.n_expert_used);
633
+ if (nb < big_need) {
634
+ nb = big_need;
635
+ }
636
+ if (nb > n_total / 4) {
637
+ nb = n_total / 4;
638
+ }
639
+ // 連工作集下限都放不進去時,寧可讓小槽位少一點(n_total 保持不變),
640
+ // 否則會算出一個「必定會中止」的配置。
641
+ n_big_ = nb;
642
+ n_small_ = n_total > nb ? n_total - nb : 0;
643
+ arena = n_small_ * small_bytes_ + n_big_ * big_bytes_;
644
+ } else {
645
+ // 單一尺寸:全部槽位都是「大的那一種」(slot_bytes_)。
646
+ // 這是 SDQ_FORCE_UNIFORM_SLOT=1 與「模型只有一種 expert 大小」時的路徑。
647
+ //
648
+ // 這裡必須記成 n_small_=0 / n_big_=全部,不能記成「n_small_=全部、單位用大的」:
649
+ // slot_ptr()/slot_size_of() 是用「索引 < n_small_ → 小槽位」來分支的,
650
+ // 把它記成全部都是小槽位的話,大 expert 會拿到小槽位的 stride
651
+ // (寫出槽位邊界、算錯),而且大 expert 永遠配不到槽位 → 推論卡死。
652
+ // (第一版就是這樣,實驗跑到一半 hang 才找出來。)
653
+ const size_t unit = slot_bytes_ ? slot_bytes_ : small_bytes_;
654
+ n_small_ = 0;
655
+ n_big_ = arena / unit;
656
+ arena = n_big_ * unit;
657
+ }
658
+ if (n_small_ == 0 && n_big_ == 0) {
659
+ throw std::runtime_error("expert arena 太小(算不出槽位)");
660
+ }
661
+ n_slots_ = n_small_ + n_big_;
662
+ if (n_slots_ < 16) {
663
+ throw std::runtime_error("expert arena 太小(算不出 16 個槽位)");
664
+ }
665
+
666
+ void * p = nullptr;
667
+ const size_t alloc = arena + (2ull << 20);
668
+ if (posix_memalign(&p, 2ull << 20, alloc) != 0 || !p) {
669
+ throw std::runtime_error("無法配置 expert arena(" + std::to_string(arena >> 20) + " MB)");
670
+ }
671
+ // 讓 arena 全部變成實體 RAM,並提示核心使用 huge page(減少 TLB 壓力)
672
+ memset(p, 0, alloc);
673
+ madvise(p, alloc, MADV_HUGEPAGE);
674
+ arena_ = (uint8_t *) p;
675
+ arena_bytes_ = arena;
676
+
677
+ // OPT-1:剖面計數表��n_layer × n_expert,40×256 = 10240 筆 ≈ 40 KB,可忽略)
678
+ prof_.assign((size_t) info_.n_layer * (size_t) info_.n_expert, 0);
679
+ ensure_layer_stats((int) info_.n_layer); // SDQ-0004
680
+ if (!cfg_.profile_in.empty()) {
681
+ load_profile(cfg_.profile_in);
682
+ }
683
+ profile_out_ = cfg_.profile_out;
684
+
685
+ slots_.assign(n_slots_, {});
686
+ free_slots_.reserve(n_small_);
687
+ for (size_t i = n_small_; i-- > 0;) {
688
+ free_slots_.push_back((int32_t) i);
689
+ }
690
+ free_big_slots_.reserve(n_big_);
691
+ for (size_t i = n_slots_; i-- > n_small_;) {
692
+ free_big_slots_.push_back((int32_t) i);
693
+ }
694
+ used_.store(0);
695
+
696
+ hist_.assign((size_t) info_.n_layer, {});
697
+ for (auto & per_layer : hist_) {
698
+ per_layer.assign((size_t) info_.n_expert, {});
699
+ }
700
+ prev_used_.assign((size_t) info_.n_layer, {});
701
+ cur_used_.assign((size_t) info_.n_layer, {});
702
+ }
703
+
704
+ void pager::init(const config & cfg, ggml_context * ctx) {
705
+ (void) ctx;
706
+ if (ready_) {
707
+ return;
708
+ }
709
+ cfg_ = cfg;
710
+ if (cfg_.model_path.empty()) {
711
+ const char * v = getenv("SDQ_MODEL_PATH");
712
+ cfg_.model_path = v ? v : "";
713
+ }
714
+ if (cfg_.model_path.empty()) {
715
+ throw std::runtime_error("SDQ 缺少模型路徑(設定 SDQ_MODEL_PATH)");
716
+ }
717
+
718
+ // 先讀 header(順便列舉 shard),再開 fd:shard 清單只有在解析
719
+ // split.count 之後才知道,而 open_model_fd() 需要它。
720
+ load_gguf_header(cfg_.model_path);
721
+ open_model_fd();
722
+ if (shard_fd_.empty() || shard_fd_[0] < 0) {
723
+ throw std::runtime_error("無法開啟模型檔:" + shard_path(0));
724
+ }
725
+ // 順序有意義:I/O 池的對齊緩衝必須在 alloc_arena() 之前配置,
726
+ // 否則 arena 會把它當成「不存在的 RAM」,實際上就偷用了 8 GB 預算。
727
+ io_pool_start();
728
+ alloc_arena();
729
+
730
+ if (cfg_.prefetch) {
731
+ pf_ = std::thread([this] { pf_worker(); });
732
+ }
733
+ ready_ = true;
734
+ print_summary();
735
+ }
736
+
737
+ const layer_desc * pager::layer(int il) const {
738
+ if (il < 0 || il >= (int) layers_.size()) {
739
+ return nullptr;
740
+ }
741
+ const layer_desc & L = layers_[(size_t) il];
742
+ if (L.experts.empty()) {
743
+ return nullptr;
744
+ }
745
+ return &L;
746
+ }
747
+
748
+ void pager::open_model_fd() {
749
+ if (shard_paths_.empty()) {
750
+ shard_paths_.push_back(cfg_.model_path);
751
+ }
752
+ shard_fd_.assign(shard_paths_.size(), -1);
753
+ shard_fd_direct_.assign(shard_paths_.size(), -1);
754
+ for (size_t i = 0; i < shard_paths_.size(); i++) {
755
+ shard_fd_[i] = open(shard_paths_[i].c_str(), O_RDONLY);
756
+ if (shard_fd_[i] < 0) {
757
+ fprintf(stderr, "[sdq] 無法開啟 shard %zu: %s\n", i, shard_paths_[i].c_str());
758
+ return;
759
+ }
760
+ }
761
+ // O_DIRECT:讀資料完全不進 kernel page cache。
762
+ // 這是「RAM 帳目誠實」的關鍵 —— 否則核心預讀會讓 mmap 檔案的頁算進 RSS
763
+ // (實測 4K prefill 時檔案對映 RSS 可以多到 6 GB 以上),而且量到的
764
+ // 「SSD 讀取量」會被上一輪的 page cache 灌水。
765
+ const char * no_direct = getenv("SDQ_NO_DIRECT_IO");
766
+ if (no_direct && *no_direct && *no_direct != '0') {
767
+ return;
768
+ }
769
+ #ifdef O_DIRECT
770
+ // 每個 shard 各自開一個 O_DIRECT fd:O_DIRECT 是 fd 層級的屬性,不能共用。
771
+ for (size_t i = 0; i < shard_paths_.size(); i++) {
772
+ shard_fd_direct_[i] = open(shard_paths_[i].c_str(), O_RDONLY | O_DIRECT);
773
+ if (shard_fd_direct_[i] < 0) {
774
+ break;
775
+ }
776
+ }
777
+ use_direct_ = shard_fd_direct_[0] >= 0;
778
+ if (use_direct_) {
779
+ fprintf(stderr, "[sdq] 使用 O_DIRECT 讀取模型檔(%zu 個 shard,不佔用 kernel page cache)\n",
780
+ shard_paths_.size());
781
+ } else {
782
+ fprintf(stderr, "[sdq] O_DIRECT 不可用,改用 buffered + fadvise(DONTNEED)\n");
783
+ }
784
+ #endif
785
+ }
786
+
787
+ // 有對齊的緩衝區(O_DIRECT 要求),thread_local 避免競爭
788
+ static thread_local uint8_t * g_direct_buf = nullptr;
789
+ static thread_local size_t g_direct_sz = 0;
790
+
791
+ // 與 read_range_direct 相同,但用呼叫端提供的緩衝區(已配置好、已算進 RSS)。
792
+ // I/O 執行緒池走這條:每條 worker 在啟動時就配置好對齊緩衝,於是 arena sizing 時
793
+ // 這筆記憶體已經在 RSS 裡,不會變成「額外偷用的 RAM」。
794
+ // sharded GGUF:把「哪個 shard」一路帶到 pread。
795
+ static inline int sdq_fd_for(const std::vector<int> & fds, int shard) {
796
+ if (shard < 0 || shard >= (int) fds.size()) {
797
+ return -1;
798
+ }
799
+ return fds[(size_t) shard];
800
+ }
801
+
802
+ bool pager::read_range_staged(int shard, uint64_t off, uint8_t * dst, size_t size,
803
+ uint8_t * stage, size_t stage_sz) {
804
+ if (!use_direct_) {
805
+ return read_range(shard, off, dst, size);
806
+ }
807
+ const int fd = sdq_fd_for(shard_fd_direct_, shard);
808
+ if (fd < 0) {
809
+ return false;
810
+ }
811
+ const size_t blk = 4096;
812
+ const uint64_t start = off & ~(uint64_t)(blk - 1);
813
+ const uint64_t end = (off + size + blk - 1) & ~(uint64_t)(blk - 1);
814
+ const size_t bytes = (size_t)(end - start);
815
+ if (stage_sz < bytes) {
816
+ // 預先配置不足(不該發生):退回 thread_local 的通用路徑
817
+ return read_range_direct(shard, off, dst, size);
818
+ }
819
+ size_t done = 0;
820
+ while (done < bytes) {
821
+ const ssize_t n = pread(fd, stage + done, bytes - done,
822
+ (off_t)(start + done));
823
+ if (n <= 0) {
824
+ return false;
825
+ }
826
+ done += (size_t) n;
827
+ }
828
+ memcpy(dst, stage + (off - start), size);
829
+ return true;
830
+ }
831
+
832
+ bool pager::read_range_direct(int shard, uint64_t off, uint8_t * dst, size_t size) {
833
+ if (!use_direct_) {
834
+ return read_range(shard, off, dst, size);
835
+ }
836
+ const int fd = sdq_fd_for(shard_fd_direct_, shard);
837
+ if (fd < 0) {
838
+ return false;
839
+ }
840
+ const size_t blk = 4096;
841
+ const uint64_t start = off & ~(uint64_t)(blk - 1);
842
+ const uint64_t end = (off + size + blk - 1) & ~(uint64_t)(blk - 1);
843
+ const size_t bytes = (size_t)(end - start);
844
+ if (g_direct_sz < bytes) {
845
+ if (g_direct_buf) {
846
+ free(g_direct_buf);
847
+ }
848
+ if (posix_memalign((void **) &g_direct_buf, blk, bytes) != 0) {
849
+ g_direct_buf = nullptr;
850
+ g_direct_sz = 0;
851
+ use_direct_ = false;
852
+ return read_range(shard, off, dst, size);
853
+ }
854
+ g_direct_sz = bytes;
855
+ }
856
+ size_t done = 0;
857
+ while (done < bytes) {
858
+ const ssize_t n = pread(fd, g_direct_buf + done, bytes - done,
859
+ (off_t)(start + done));
860
+ if (n <= 0) {
861
+ return false;
862
+ }
863
+ done += (size_t) n;
864
+ }
865
+ memcpy(dst, g_direct_buf + (off - start), size);
866
+ return true;
867
+ }
868
+
869
+ bool pager::read_range(int shard, uint64_t off, uint8_t * dst, size_t size) {
870
+ const int fd = sdq_fd_for(shard_fd_, shard);
871
+ if (fd < 0) {
872
+ return false;
873
+ }
874
+ size_t done = 0;
875
+ while (done < size) {
876
+ const ssize_t n = pread(fd, dst + done, size - done, (off_t) (off + done));
877
+ if (n <= 0) {
878
+ return false;
879
+ }
880
+ done += (size_t) n;
881
+ }
882
+ // 讓 RAM 帳目誠實:把 kernel 對這個檔案的 page cache 丟掉,避免「藏在核心裡的 RAM」
883
+ if (cfg_.fadvise_dontneed) {
884
+ posix_fadvise(fd, (off_t) off, (off_t) size, POSIX_FADV_DONTNEED);
885
+ }
886
+ return true;
887
+ }
888
+
889
+ bool pager::read_expert(int il, int ie, uint8_t * dst) {
890
+ const layer_desc * L = layer(il);
891
+ if (!L || ie < 0 || ie >= (int) L->experts.size()) {
892
+ return false;
893
+ }
894
+ const expert_desc & ed = L->experts[(size_t) ie];
895
+ for (int p = 0; p < 3; p++) {
896
+ if (ed.seg[p].size == 0) {
897
+ continue;
898
+ }
899
+ if (!read_range_direct(ed.seg[p].shard, ed.seg[p].off,
900
+ dst + ed.slot_off[p], (size_t) ed.seg[p].size)) {
901
+ return false;
902
+ }
903
+ }
904
+ return true;
905
+ }
906
+
907
+ // ---------------------------------------------------------------- I/O 執行緒池
908
+ //
909
+ // 問題:一個 expert = gate / up / down 三段,檔案上是三段不相鄰的資料。
910
+ // 循序讀就是 3 次 pread 排在同一條執行緒上,而單次 1.9 MB 的 pread 在本機量到
911
+ // ~7 ms(≈275 MB/s 的單流頻寬)。因此「一個 expert 要多久」= 3 × 單次延遲,
912
+ // 這條路徑的長度跟「有多少條執行緒」無關 —— mode C 再怎麼排程都救不了。
913
+ //
914
+ // 做法:把三段交給三條執行緒同時 pread,臨界路徑降到約「單段」的時間。
915
+ // 量測(8 個 expert 全部 miss 的一層,各 9 次取中位數):
916
+ // 8 執行緒 × 三段序列讀 : 12.60 ms
917
+ // 16 執行緒 × 三段平行讀 : 7.57 ms
918
+ // 24 執行緒 × 三段平行讀 : 7.58 ms ← 16 就飽和,所以預設 3 × n_threads
919
+ //
920
+ // 這個池子對 prefill 更有用:一層要 256 個 expert = 768 個讀取工作,
921
+ // 序列讀時 16 執行緒只能拿到 16 路的頻寬 utilization。
922
+ void pager::io_pool_start() {
923
+ if (!cfg_.io_parallel) {
924
+ fprintf(stderr, "[sdq] I/O 段平行:關(每個 expert 的 gate/up/down 循序讀)\n");
925
+ return;
926
+ }
927
+ int n = cfg_.io_threads > 0 ? cfg_.io_threads : cfg_.n_threads * 3;
928
+ if (n < 4) {
929
+ n = 4;
930
+ }
931
+ if (n > 48) {
932
+ n = 48;
933
+ }
934
+ // 先算「一段權重最大有多少位元組」,每條 worker 配一塊對齊緩衝。
935
+ // 這必须在 alloc_arena() 之前做完,否則這筆 RAM 會變成帳外支出。
936
+ size_t max_seg = 0;
937
+ for (const auto & L : layers_) {
938
+ for (const auto & ed : L.experts) {
939
+ for (int p = 0; p < 3; p++) {
940
+ max_seg = std::max(max_seg, (size_t) ed.seg[p].size);
941
+ }
942
+ }
943
+ }
944
+ const size_t stage_bytes = ((max_seg + 2 * 4096) + 4095) & ~(size_t) 4095;
945
+ io_stage_.assign((size_t) n, io_stage{});
946
+ for (int i = 0; i < n; i++) {
947
+ void * p = nullptr;
948
+ if (posix_memalign(&p, 4096, stage_bytes) == 0 && p) {
949
+ memset(p, 0, stage_bytes); // 實際碰頁,讓它出現在 RSS 裡
950
+ io_stage_[(size_t) i].buf = (uint8_t *) p;
951
+ io_stage_[(size_t) i].sz = stage_bytes;
952
+ }
953
+ }
954
+ io_split_ = cfg_.io_split > 0 ? cfg_.io_split : 1;
955
+ io_stop_ = false;
956
+ io_pool_.reserve((size_t) n);
957
+ for (int i = 0; i < n; i++) {
958
+ const size_t wid = (size_t) i;
959
+ try {
960
+ io_pool_.emplace_back([this, wid] { io_worker(wid); });
961
+ } catch (...) {
962
+ break;
963
+ }
964
+ }
965
+ io_threads_ = (int) io_pool_.size();
966
+ io_pool_on_ = io_threads_ > 0;
967
+ if (io_pool_on_) {
968
+ fprintf(stderr, "[sdq] I/O 段平行:%d 條讀取執行緒 × %zu KB 對齊緩衝"
969
+ "(gate/up/down 同時讀,每段再切 %d 塊)\n",
970
+ io_threads_, stage_bytes >> 10, io_split_);
971
+ }
972
+ }
973
+
974
+ void pager::io_pool_stop() {
975
+ if (io_pool_.empty()) {
976
+ for (auto & s : io_stage_) {
977
+ free(s.buf);
978
+ s.buf = nullptr;
979
+ }
980
+ io_stage_.clear();
981
+ return;
982
+ }
983
+ {
984
+ std::lock_guard<std::mutex> lk(io_mu_);
985
+ io_stop_ = true;
986
+ }
987
+ io_cv_.notify_all();
988
+ for (auto & t : io_pool_) {
989
+ if (t.joinable()) {
990
+ t.join();
991
+ }
992
+ }
993
+ io_pool_.clear();
994
+ io_pool_on_ = false;
995
+ for (auto & s : io_stage_) {
996
+ free(s.buf);
997
+ s.buf = nullptr;
998
+ s.sz = 0;
999
+ }
1000
+ io_stage_.clear();
1001
+ }
1002
+
1003
+ void pager::io_worker(size_t wid) {
1004
+ for (;;) {
1005
+ io_job job{};
1006
+ {
1007
+ std::unique_lock<std::mutex> lk(io_mu_);
1008
+ io_cv_.wait(lk, [this] { return io_stop_ || !io_q_.empty(); });
1009
+ if (io_q_.empty()) {
1010
+ if (io_stop_) {
1011
+ return;
1012
+ }
1013
+ continue;
1014
+ }
1015
+ job = io_q_.front();
1016
+ io_q_.pop_front();
1017
+ }
1018
+ io_jobs_.fetch_add(1, std::memory_order_relaxed);
1019
+ // 讀取本身不持鎖:io_mu_ 只保護佇列與完成計數
1020
+ const bool ok = (wid < io_stage_.size() && io_stage_[wid].buf)
1021
+ ? read_range_staged(job.shard, job.off, job.dst, job.size,
1022
+ io_stage_[wid].buf, io_stage_[wid].sz)
1023
+ : read_range_direct(job.shard, job.off, job.dst, job.size);
1024
+ {
1025
+ std::lock_guard<std::mutex> lk(io_mu_);
1026
+ if (!ok) {
1027
+ job.ctx->ok = false;
1028
+ }
1029
+ if (job.ctx->remaining.fetch_sub(1) == 1) {
1030
+ io_done_cv_.notify_all();
1031
+ }
1032
+ }
1033
+ }
1034
+ }
1035
+
1036
+ // 把 (il,ie) 的三段交給執行緒池並行讀,呼叫端等到整批完成才返回。
1037
+ // ctx 放在呼叫端 stack:worker 只會在 remaining 歸零之後才再碰它,
1038
+ // 而我們正是等到 remaining==0 才離開函式,所以沒有 use-after-return。
1039
+ bool pager::read_expert_par(int il, int ie, uint8_t * dst) {
1040
+ const layer_desc * L = layer(il);
1041
+ if (!L || ie < 0 || ie >= (int) L->experts.size()) {
1042
+ return false;
1043
+ }
1044
+ if (!io_pool_on_) {
1045
+ return read_expert(il, ie, dst);
1046
+ }
1047
+ const expert_desc & ed = L->experts[(size_t) ie];
1048
+ io_ctx ctx;
1049
+ int n = 0;
1050
+ // 把每一段再切成一樣大的好幾塊,交給不同的執行緒同時讀。
1051
+ //
1052
+ // 為什麼:單一 pread 的時間 ≈ 固定延遲 + 位元組數 / 單流頻寬。
1053
+ // 本機實測單流約 275 MB/s、4K 隨機讀延遲約 0.57 ms,所以
1054
+ // 590 KB 讀一次 ≈ 0.57 + 2.1 = 2.7 ms
1055
+ // 590 KB 切成 2 塊 ≈ 0.57 + 1.05 = 1.6 ms
1056
+ // 而「讀完一個 expert」的時間正是每層的關鍵路徑,切塊就是直接砍關鍵路徑。
1057
+ // 設定檔 SDQ_IO_SPLIT。**實測預設是 1(不切塊)**:切塊在 24 條讀取執行緒
1058
+ // 之下沒有可重現的改善(split=1..6 的中位數在 4.9~5.9 之間跳動,純噪音),
1059
+ // 因為 8 個 expert × 3 段 = 24 個並行請求已經把裝置的佇列灌飽,
1060
+ // 這時瓶頸是裝置總頻寬而不是單一請求的延遲。保留這個開關是為了其他裝置。
1061
+ const int split = io_split_;
1062
+ const size_t blk = 4096;
1063
+ {
1064
+ std::lock_guard<std::mutex> lk(io_mu_);
1065
+ for (int p = 0; p < 3; p++) {
1066
+ const size_t sz = (size_t) ed.seg[p].size;
1067
+ if (sz == 0) {
1068
+ continue;
1069
+ }
1070
+ uint8_t * base = dst + ed.slot_off[p];
1071
+ const int sh = ed.seg[p].shard;
1072
+ if (split <= 1 || sz <= 2 * blk) {
1073
+ io_q_.push_back(io_job{sh, ed.seg[p].off, base, sz, &ctx});
1074
+ n++;
1075
+ continue;
1076
+ }
1077
+ size_t chunk = ((sz + (size_t) split - 1) / (size_t) split + blk - 1) & ~(blk - 1);
1078
+ for (size_t off = 0; off < sz; off += chunk) {
1079
+ const size_t take = sz - off < chunk ? sz - off : chunk;
1080
+ io_q_.push_back(io_job{sh, ed.seg[p].off + off, base + off, take, &ctx});
1081
+ n++;
1082
+ }
1083
+ }
1084
+ ctx.remaining.store(n);
1085
+ }
1086
+ if (n == 0) {
1087
+ return true;
1088
+ }
1089
+ io_cv_.notify_all();
1090
+ const uint64_t t0 = now_us();
1091
+ {
1092
+ std::unique_lock<std::mutex> lk(io_mu_);
1093
+ io_done_cv_.wait(lk, [&ctx] { return ctx.remaining.load() == 0; });
1094
+ }
1095
+ io_wait_ns_.fetch_add((now_us() - t0) * 1000ull, std::memory_order_relaxed);
1096
+ return ctx.ok;
1097
+ }
1098
+
1099
+ uint64_t pager::read_expert_span(int il, const int32_t * experts, int n, const int32_t * slots) {
1100
+ const layer_desc * L = layer(il);
1101
+ if (!L || n <= 0) {
1102
+ return 0;
1103
+ }
1104
+ const uint64_t t_start = now_us();
1105
+ // 每個執行緒一份暫存區(多執行緒會同時呼叫,不能共用一個 buffer)
1106
+ static thread_local std::vector<uint8_t> scratch_buf;
1107
+ uint64_t total = 0;
1108
+
1109
+ // 把 experts 切成「檔案編號連續」的區段
1110
+ int i = 0;
1111
+ while (i < n) {
1112
+ int j = i + 1;
1113
+ while (j < n && experts[j] == experts[j - 1] + 1) {
1114
+ j++;
1115
+ }
1116
+ const int cnt = j - i;
1117
+ const expert_desc & first = L->experts[(size_t) experts[i]];
1118
+ const expert_desc & last = L->experts[(size_t) experts[j - 1]];
1119
+
1120
+ for (int p = 0; p < 3; p++) {
1121
+ if (first.seg[p].size == 0 || last.seg[p].size == 0) {
1122
+ continue;
1123
+ }
1124
+ // 連續區段必須在同一個 shard 內;跨 shard 就切斷這個區段。
1125
+ if (last.seg[p].shard != first.seg[p].shard) {
1126
+ continue;
1127
+ }
1128
+ const uint64_t off = first.seg[p].off;
1129
+ const uint64_t end = last.seg[p].off + last.seg[p].size;
1130
+ const size_t bytes = (size_t) (end - off);
1131
+ if (scratch_buf.size() < bytes) {
1132
+ scratch_buf.assign(bytes, 0);
1133
+ }
1134
+ if (!read_range_direct(first.seg[p].shard, off, scratch_buf.data(), bytes)) {
1135
+ return total;
1136
+ }
1137
+ total += bytes;
1138
+ // 散到各自的槽位:scratch 從 off 開始,所以偏移是「相對於 off」
1139
+ for (int k = i; k < j; k++) {
1140
+ const expert_desc & ed = L->experts[(size_t) experts[k]];
1141
+ const size_t seg_bytes = (size_t) ed.seg[p].size;
1142
+ const size_t delta = (size_t) (ed.seg[p].off - off);
1143
+ if (slots[k] < 0) {
1144
+ continue;
1145
+ }
1146
+ if ((size_t) slots[k] >= n_slots_ || (uint64_t) ed.slot_off[p] + seg_bytes > slot_bytes_
1147
+ || delta + seg_bytes > scratch_buf.size()) {
1148
+ fprintf(stderr, "[sdq-pager] span 越界:il=%d slot=%d/%zu slot_off=%lld seg=%zu delta=%zu scratch=%zu p=%d e=%d\n",
1149
+ il, slots[k], n_slots_, (long long) ed.slot_off[p], seg_bytes, delta, scratch_buf.size(), p, experts[k]);
1150
+ return total;
1151
+ }
1152
+ memcpy(slot_ptr(slots[k]) + ed.slot_off[p],
1153
+ scratch_buf.data() + delta, seg_bytes);
1154
+ }
1155
+ }
1156
+ i = j;
1157
+ }
1158
+ stats_.demand_wait_us.fetch_add(now_us() - t_start, std::memory_order_relaxed);
1159
+ return total;
1160
+ }
1161
+
1162
+ int pager::reserve_slot(int il, int ie) {
1163
+ if (!ready_) {
1164
+ return -1;
1165
+ }
1166
+ count_request(il, ie);
1167
+ std::lock_guard<std::mutex> lk(mu_);
1168
+ const int32_t exist = find_slot_locked(il, ie);
1169
+ if (exist >= 0) {
1170
+ slots_[(size_t) exist].last_use = ++clock_;
1171
+ slots_[(size_t) exist].pin++;
1172
+ touch_freq_locked(slots_[(size_t) exist]);
1173
+ return exist;
1174
+ }
1175
+ const int32_t s = victim_locked();
1176
+ if (s < 0) {
1177
+ return -1;
1178
+ }
1179
+ slot_state & st = slots_[(size_t) s];
1180
+ if (st.layer >= 0) {
1181
+ page_.erase(((int64_t) st.layer << 20) | (int64_t) st.expert);
1182
+ stats_.evictions.fetch_add(1, std::memory_order_relaxed);
1183
+ used_.fetch_sub(1, std::memory_order_relaxed);
1184
+ }
1185
+ st.layer = il;
1186
+ st.expert = ie;
1187
+ st.last_use = ++clock_;
1188
+ st.freq = 1;
1189
+ st.pin = 1;
1190
+ st.is_hot = hot_lookup(il, ie);
1191
+ page_[((int64_t) il << 20) | (int64_t) ie] = s;
1192
+ used_.fetch_add(1, std::memory_order_relaxed);
1193
+ stats_.expert_misses.fetch_add(1, std::memory_order_relaxed);
1194
+ return s;
1195
+ }
1196
+
1197
+ int pager::probe(int il, int ie) {
1198
+ if (!ready_) {
1199
+ return -1;
1200
+ }
1201
+ stats_.expert_requests.fetch_add(1, std::memory_order_relaxed);
1202
+ count_request(il, ie);
1203
+ std::lock_guard<std::mutex> lk(mu_);
1204
+ const int32_t s = find_slot_locked(il, ie);
1205
+ if (s < 0) {
1206
+ stats_.expert_misses.fetch_add(1, std::memory_order_relaxed);
1207
+ return -1;
1208
+ }
1209
+ slots_[(size_t) s].last_use = ++clock_;
1210
+ slots_[(size_t) s].pin++;
1211
+ touch_freq_locked(slots_[(size_t) s]);
1212
+ stats_.expert_hits.fetch_add(1, std::memory_order_relaxed);
1213
+ return s;
1214
+ }
1215
+
1216
+ void pager::note_span_read(uint64_t bytes) {
1217
+ stats_.ssb_read_bytes_demand.fetch_add(bytes, std::memory_order_relaxed);
1218
+ stats_.demand_wait_us.fetch_add(0, std::memory_order_relaxed);
1219
+ }
1220
+
1221
+ // 衰減式頻率:每 8192 次使用就把所有計數減半,讓「過去熱門、現在變冷」的 expert 能被淘汰
1222
+ void pager::touch_freq_locked(slot_state & st) {
1223
+ if (++freq_tick_ >= 8192) {
1224
+ freq_tick_ = 0;
1225
+ for (auto & s : slots_) {
1226
+ s.freq /= 2;
1227
+ }
1228
+ }
1229
+ if (st.freq < 0xFFFFFFFFu) {
1230
+ st.freq++;
1231
+ }
1232
+ }
1233
+
1234
+ // ============================== OPT-1:熱門 expert 剖面 ==============================
1235
+ // 問題實測(docs/03):arena 3.88 GiB 只放得下 2044 個 expert,總共卻有 10240 個,
1236
+ // prefill 的幾百個「只用一次」的專家會把熱區整個洗掉,命中率卡在 ~70%。
1237
+ // 解法借用 Strata 的做法(src/core/expert_cache.cpp):**離線剖面 + 靜態排行榜 +
1238
+ // 熱頁永不淘汰**。這裡不動檔案格式,用純文字存排行榜。
1239
+
1240
+ // ---- SDQ-0004
1241
+ void pager::ensure_layer_stats(int n_layer) {
1242
+ if (lstats_ || n_layer <= 0) {
1243
+ return;
1244
+ }
1245
+ lstats_.reset(new layer_stat[(size_t) n_layer]);
1246
+ }
1247
+
1248
+ void pager::count_request(int il, int ie) {
1249
+ if (prof_.empty() || il < 0 || ie < 0) {
1250
+ return;
1251
+ }
1252
+ const size_t idx = (size_t) il * (size_t) info_.n_expert + (size_t) ie;
1253
+ if (idx < prof_.size()) {
1254
+ // 單執行緒 ++ 即可:請求次數是統計用途,不參與任何正確性判斷
1255
+ if (prof_[idx] < 0xFFFFFFFFu) {
1256
+ prof_[idx]++;
1257
+ }
1258
+ }
1259
+ }
1260
+
1261
+ bool pager::hot_lookup(int il, int ie) const {
1262
+ if (hot_.empty() || il < 0 || ie < 0) {
1263
+ return false;
1264
+ }
1265
+ const size_t idx = (size_t) il * (size_t) info_.n_expert + (size_t) ie;
1266
+ return idx < hot_.size() && hot_[idx] != 0;
1267
+ }
1268
+
1269
+ // 挑一個容得下 (il,ie) 的槽。與 victim_locked 的差別:**先只在冷頁裡找**,
1270
+ // 找不到才退讓去動熱頁。這就是「熱區不會被 prefill 洗掉」的機制。
1271
+ // 這個槽位放得下「需要 big 的 expert」嗎?
1272
+ // 小槽位:只能放小 expert
1273
+ // 大槽位:小expert 也放得下(只是浪費一點空間)
1274
+ static inline bool slot_accepts(bool need_big, bool slot_is_big) {
1275
+ return !need_big || slot_is_big;
1276
+ }
1277
+
1278
+ int pager::admit_slot_locked(int il, int ie) {
1279
+ // il/ie 都 < 0 代表「不指定大小」→ 兩種槽位都可以(舊的呼叫路徑)。
1280
+ const bool need_big = (il >= 0 && ie >= 0) ? need_big_slot(il, ie) : false;
1281
+ // 完全沒用過的空槽永遠優先(它不是熱頁,佔用等於浪費)
1282
+ // 大 expert 先看大槽位;小 expert 先看小槽位。
1283
+ std::vector<int32_t> & first = need_big ? free_big_slots_ : free_slots_;
1284
+ std::vector<int32_t> & second = need_big ? free_slots_ : free_big_slots_;
1285
+ while (!first.empty()) {
1286
+ const int32_t s = first.back();
1287
+ if (!slots_[(size_t) s].busy) {
1288
+ first.pop_back();
1289
+ return s;
1290
+ }
1291
+ }
1292
+ // 第一輪:只在「冷頁」裡挑最久未用的。
1293
+ // want_hot=true → 熱門 expert 進來時,把冷頁換掉(不動其他熱頁)
1294
+ // want_hot=false → 冷門 expert 進來時,優先吃冷頁
1295
+ // 兩種情況的規則相同,所以 hot_lookup 的結果不影響選擇順序,只影響標記。
1296
+ for (int pass = 0; pass < 2; pass++) {
1297
+ int64_t best = -1;
1298
+ uint64_t best_clock = UINT64_MAX;
1299
+ for (size_t s = 0; s < n_slots_; s++) {
1300
+ const slot_state & st = slots_[s];
1301
+ if (st.pin > 0 || st.busy) {
1302
+ continue;
1303
+ }
1304
+ if (pass == 0 && st.is_hot) {
1305
+ continue; // 第一輪不動熱頁
1306
+ }
1307
+ if (!slot_accepts(need_big, is_big_slot(s))) {
1308
+ continue;
1309
+ }
1310
+ if (st.last_use < best_clock) {
1311
+ best_clock = st.last_use;
1312
+ best = (int64_t) s;
1313
+ }
1314
+ }
1315
+ if (best >= 0) {
1316
+ return (int) best;
1317
+ }
1318
+ }
1319
+ // 退讓:另一種大小的空槽。**一定要檢查放不放得下** —— 小槽位放不下大 expert,
1320
+ // 硬塞會讓 read_expert 寫出槽位邊界(靜默把別的 expert 寫壞)。
1321
+ while (!second.empty()) {
1322
+ const int32_t s = second.back();
1323
+ if (!slots_[(size_t) s].busy &&
1324
+ slot_accepts(need_big, is_big_slot(s))) {
1325
+ second.pop_back();
1326
+ return s;
1327
+ }
1328
+ if (!slots_[(size_t) s].busy) {
1329
+ second.pop_back();
1330
+ } else {
1331
+ break;
1332
+ }
1333
+ }
1334
+ return -1;
1335
+ }
1336
+
1337
+ bool pager::load_profile(const std::string & path) {
1338
+ std::FILE * f = std::fopen(path.c_str(), "rb");
1339
+ if (!f) {
1340
+ fprintf(stderr, "[sdq] 找不到 expert 剖面 %s(將以動態 LRU/LFU 運作)\n", path.c_str());
1341
+ return false;
1342
+ }
1343
+ char magic[8] = {0};
1344
+ if (std::fread(magic, 1, 4, f) != 4 || std::memcmp(magic, "SDQP", 4) != 0) {
1345
+ std::fclose(f);
1346
+ fprintf(stderr, "[sdq] %s ��是 SDQ expert 剖面(缺少 SDQP 標頭)\n", path.c_str());
1347
+ return false;
1348
+ }
1349
+ uint32_t hdr[3] = {0, 0, 0};
1350
+ if (std::fread(hdr, sizeof(uint32_t), 3, f) != 3) {
1351
+ std::fclose(f);
1352
+ return false;
1353
+ }
1354
+ const uint32_t n = hdr[2];
1355
+ std::vector<std::pair<uint32_t, std::pair<int32_t, int32_t>>> ranked;
1356
+ ranked.reserve(n);
1357
+ for (uint32_t i = 0; i < n; i++) {
1358
+ int32_t l = 0, e = 0;
1359
+ uint32_t c = 0;
1360
+ if (std::fread(&l, sizeof(int32_t), 1, f) != 1) break;
1361
+ if (std::fread(&e, sizeof(int32_t), 1, f) != 1) break;
1362
+ if (std::fread(&c, sizeof(uint32_t), 1, f) != 1) break;
1363
+ ranked.push_back({c, {l, e}});
1364
+ }
1365
+ std::fclose(f);
1366
+ std::sort(ranked.begin(), ranked.end(),
1367
+ [](const auto & a, const auto & b) { return a.first > b.first; });
1368
+
1369
+ hot_.assign((size_t) info_.n_layer * (size_t) info_.n_expert, 0);
1370
+ const int want = cfg_.hot_pin > 0 ? cfg_.hot_pin : (int) ranked.size();
1371
+ n_hot_ = 0;
1372
+ for (const auto & r : ranked) {
1373
+ if (n_hot_ >= want) {
1374
+ break;
1375
+ }
1376
+ const int l = r.second.first, e = r.second.second;
1377
+ if (l < 0 || l >= info_.n_layer || e < 0 || e >= info_.n_expert) {
1378
+ continue;
1379
+ }
1380
+ const size_t idx = (size_t) l * (size_t) info_.n_expert + (size_t) e;
1381
+ if (!hot_[idx]) {
1382
+ hot_[idx] = 1;
1383
+ n_hot_++;
1384
+ }
1385
+ }
1386
+ fprintf(stderr, "[sdq] expert 剖面:%s → 釘選 %d / %zu 個熱門 expert(永不淘汰)\n",
1387
+ path.c_str(), n_hot_, ranked.size());
1388
+ return n_hot_ > 0;
1389
+ }
1390
+
1391
+ void pager::dump_profile(const std::string & path) {
1392
+ if (prof_.empty()) {
1393
+ return;
1394
+ }
1395
+ std::vector<std::pair<uint32_t, std::pair<int32_t, int32_t>>> ranked;
1396
+ ranked.reserve(prof_.size());
1397
+ uint64_t total = 0;
1398
+ for (int l = 0; l < info_.n_layer; l++) {
1399
+ for (int e = 0; e < info_.n_expert; e++) {
1400
+ const uint32_t c = prof_[(size_t) l * (size_t) info_.n_expert + (size_t) e];
1401
+ if (c > 0) {
1402
+ ranked.push_back({c, {l, e}});
1403
+ total += c;
1404
+ }
1405
+ }
1406
+ }
1407
+ std::sort(ranked.begin(), ranked.end(),
1408
+ [](const auto & a, const auto & b) { return a.first > b.first; });
1409
+ std::FILE * f = std::fopen(path.c_str(), "wb");
1410
+ if (!f) {
1411
+ fprintf(stderr, "[sdq] 無法寫出 expert 剖面 %s\n", path.c_str());
1412
+ return;
1413
+ }
1414
+ std::fwrite("SDQP", 1, 4, f);
1415
+ const uint32_t hdr[3] = {1, (uint32_t) info_.n_layer, (uint32_t) ranked.size()};
1416
+ std::fwrite(hdr, sizeof(uint32_t), 3, f);
1417
+ for (const auto & r : ranked) {
1418
+ const int32_t l = r.second.first, e = r.second.second;
1419
+ const uint32_t c = r.first;
1420
+ std::fwrite(&l, sizeof(int32_t), 1, f);
1421
+ std::fwrite(&e, sizeof(int32_t), 1, f);
1422
+ std::fwrite(&c, sizeof(uint32_t), 1, f);
1423
+ }
1424
+ std::fclose(f);
1425
+ // 覆盖率報告:前 K 名可以蓋掉多少請求 → 直接決定 hot_pin 該設多少
1426
+ uint64_t acc = 0;
1427
+ const size_t ks[6] = {512, 1024, 1536, 2048, 3072, 4096};
1428
+ fprintf(stderr, "[sdq] expert 剖面已寫到 %s(%zu 個有被用到,總請求 %llu)\n",
1429
+ path.c_str(), ranked.size(), (unsigned long long) total);
1430
+ size_t ki = 0;
1431
+ for (size_t i = 0; i < ranked.size(); i++) {
1432
+ acc += ranked[i].first;
1433
+ if (ki < 6 && i + 1 == ks[ki]) {
1434
+ fprintf(stderr, "[sdq] 前 %5zu 名涵蓋 %6.2f%% 的請求\n",
1435
+ ks[ki], 100.0 * (double) acc / (double) (total ? total : 1));
1436
+ ki++;
1437
+ }
1438
+ }
1439
+ }
1440
+
1441
+ int pager::find_slot_locked(int il, int ie) {
1442
+ const int64_t key = ((int64_t) il << 20) | (int64_t) ie;
1443
+ auto it = page_.find(key);
1444
+ return it == page_.end() ? -1 : it->second;
1445
+ }
1446
+
1447
+ // 找一個「可以放 (il,ie) 而且 last_use <= stale_before」的槽位。
1448
+ //
1449
+ // 為什麼不能直接用 victim_stale_locked():大小兩種槽位之後,那個函式會掃過**所有**
1450
+ // 槽位(包含小槽位),於是一個需要大槽位的 expert 可能拿到小槽位,
1451
+ // 預取時 read_expert 會寫出槽位邊界 → 靜默把鄰居槽位的資料寫壞。
1452
+ // 這是引進大小兩種槽位時真正會咬人的地方。
1453
+ //
1454
+ // 順序:先看對應尺寸的空槽(從沒用過,最該拿來放預取),
1455
+ // 再掃全部槽位挑最久未用的(放得下的那種尺寸)。
1456
+ int pager::victim_stale_for(int il, int ie, uint64_t stale_before) {
1457
+ const bool need_big = need_big_slot(il, ie);
1458
+ std::vector<int32_t> & first = need_big ? free_big_slots_ : free_slots_;
1459
+ std::vector<int32_t> & second = need_big ? free_slots_ : free_big_slots_;
1460
+ for (std::vector<int32_t> * lst : { &first, &second }) {
1461
+ while (!lst->empty()) {
1462
+ const int32_t s = lst->back();
1463
+ if (slots_[(size_t) s].busy) {
1464
+ break;
1465
+ }
1466
+ if (slots_[(size_t) s].last_use <= stale_before &&
1467
+ slot_accepts(need_big, is_big_slot(s))) {
1468
+ lst->pop_back();
1469
+ return s;
1470
+ }
1471
+ lst->pop_back();
1472
+ }
1473
+ }
1474
+ int64_t best = -1;
1475
+ uint64_t best_clock = UINT64_MAX;
1476
+ for (size_t i = 0; i < n_slots_; i++) {
1477
+ const slot_state & st = slots_[i];
1478
+ if (st.pin > 0 || st.busy || st.last_use > stale_before) {
1479
+ continue;
1480
+ }
1481
+ if (!slot_accepts(need_big, is_big_slot(i))) {
1482
+ continue;
1483
+ }
1484
+ if (st.last_use < best_clock) {
1485
+ best_clock = st.last_use;
1486
+ best = (int64_t) i;
1487
+ }
1488
+ }
1489
+ return (int) best;
1490
+ }
1491
+
1492
+ int pager::victim_stale_locked(uint64_t stale_before) {
1493
+ // 只挑「last_use 很舊」而且沒被 pin 的槽:保護最近的工作集
1494
+ while (!free_slots_.empty()) {
1495
+ const int32_t s = free_slots_.back();
1496
+ if (!slots_[(size_t) s].busy && slots_[(size_t) s].last_use <= stale_before) {
1497
+ free_slots_.pop_back();
1498
+ return s;
1499
+ }
1500
+ if (free_slots_.size() * 4 < n_slots_) {
1501
+ break;
1502
+ }
1503
+ free_slots_.pop_back();
1504
+ }
1505
+ int64_t best = -1;
1506
+ uint64_t best_clock = UINT64_MAX;
1507
+ for (size_t i = 0; i < n_slots_; i++) {
1508
+ const slot_state & st = slots_[i];
1509
+ if (st.pin > 0 || st.busy || st.last_use > stale_before) {
1510
+ continue;
1511
+ }
1512
+ if (st.last_use < best_clock) {
1513
+ best_clock = st.last_use;
1514
+ best = (int64_t) i;
1515
+ }
1516
+ }
1517
+ return (int) best;
1518
+ }
1519
+
1520
+ int pager::acquire(int il, int ie, bool blocking) {
1521
+ if (!ready_) {
1522
+ return -1;
1523
+ }
1524
+ std::unique_lock<std::mutex> lk(mu_);
1525
+ int32_t s = find_slot_locked(il, ie);
1526
+ if (s >= 0 && slots_[s].busy) {
1527
+ // 這個槽位**正在被填入**:另一條執行緒剛 admit 它、資料還沒讀完。
1528
+ //
1529
+ // 這裡一定要等,不能當成命中直接用。否則讀到的是「寫到一半的權重」,
1530
+ // 而且那個錯誤是**非決定性**的 —— 每次執行結果可能不同(實測在
1531
+ // prefill 時會讓同一個提示詞跑出不同的文字)。
1532
+ // busy 的槽位在 admit/victim 那邊本來就會被跳過,所以這裡是唯一
1533
+ // 還能拿到它的路徑,也因此必須在這裡等。
1534
+ const uint64_t t0 = now_us();
1535
+ while (slots_[s].busy) {
1536
+ lk.unlock();
1537
+ std::this_thread::sleep_for(std::chrono::microseconds(200));
1538
+ lk.lock();
1539
+ if (now_us() - t0 > (uint64_t) env_size("SDQ_BUSY_WAIT_MS", 30000) * 1000) {
1540
+ fprintf(stderr,
1541
+ "[sdq] 致命:等待槽位 il=%d ie=%d 讀入超過 %u ms(busy 一直沒解除)\n",
1542
+ il, ie, (unsigned) env_size("SDQ_BUSY_WAIT_MS", 30000));
1543
+ std::fflush(stderr);
1544
+ abort();
1545
+ }
1546
+ }
1547
+ // 讀取可能失敗(那時槽位已被清空)→ 重新走一次正常的 miss 路徑
1548
+ if (find_slot_locked(il, ie) != s) {
1549
+ s = -1;
1550
+ }
1551
+ }
1552
+ if (s >= 0) {
1553
+ slots_[s].last_use = ++clock_;
1554
+ slots_[s].pin++;
1555
+ touch_freq_locked(slots_[(size_t) s]);
1556
+ stats_.expert_requests.fetch_add(1, std::memory_order_relaxed);
1557
+ stats_.expert_hits.fetch_add(1, std::memory_order_relaxed);
1558
+ if (lstats_ && il >= 0 && il < (int) info_.n_layer) {
1559
+ lstats_[il].req.fetch_add(1, std::memory_order_relaxed);
1560
+ lstats_[il].hit.fetch_add(1, std::memory_order_relaxed);
1561
+ }
1562
+ // A prefetch that already finished counts as "used" here. Without this
1563
+ // the finished-before-request case was invisible and prefetch precision
1564
+ // was understated (docs/10 §f).
1565
+ if (slots_[s].from_prefetch) {
1566
+ slots_[s].from_prefetch = false;
1567
+ stats_.prefetch_used.fetch_add(1, std::memory_order_relaxed);
1568
+ stats_.prefetch_bytes_used.fetch_add(slot_size_of(s), std::memory_order_relaxed);
1569
+ }
1570
+ return s;
1571
+ }
1572
+
1573
+ stats_.expert_requests.fetch_add(1, std::memory_order_relaxed);
1574
+ count_request(il, ie);
1575
+ stats_.expert_misses.fetch_add(1, std::memory_order_relaxed);
1576
+ if (lstats_ && il >= 0 && il < (int) info_.n_layer) {
1577
+ lstats_[il].req.fetch_add(1, std::memory_order_relaxed);
1578
+ lstats_[il].miss.fetch_add(1, std::memory_order_relaxed);
1579
+ }
1580
+
1581
+ // 先找一個已經在背景讀入這頁的槽
1582
+ for (size_t i = 0; i < n_slots_; i++) {
1583
+ slot_state & st = slots_[i];
1584
+ if (st.busy && st.layer == il && st.expert == ie) {
1585
+ // 等背景執行緒讀完
1586
+ lk.unlock();
1587
+ const uint64_t t0 = now_us();
1588
+ while (true) {
1589
+ lk.lock();
1590
+ if (!slots_[i].busy) {
1591
+ break;
1592
+ }
1593
+ lk.unlock();
1594
+ std::this_thread::sleep_for(std::chrono::microseconds(200));
1595
+ }
1596
+ s = (int32_t) i;
1597
+ lk.unlock();
1598
+ stats_.prefetch_wait_us.fetch_add(now_us() - t0, std::memory_order_relaxed);
1599
+ std::unique_lock<std::mutex> lk2(mu_);
1600
+ slots_[s].last_use = ++clock_;
1601
+ slots_[s].pin++;
1602
+ stats_.expert_hits.fetch_add(1, std::memory_order_relaxed);
1603
+ stats_.prefetch_used.fetch_add(1, std::memory_order_relaxed);
1604
+ return s;
1605
+ }
1606
+ }
1607
+
1608
+ if (!blocking) {
1609
+ return -1;
1610
+ }
1611
+
1612
+ // 選槽位時必須知道「這個 expert 需要哪一種大小」,否則大 expert 可能被塞進小槽位
1613
+ // (寫出邊界 → 靜默算錯),或者小 expert 佔掉大槽位(浪費 6.8% 的 arena)。
1614
+ s = admit_slot_locked(il, ie);
1615
+ if (s < 0) {
1616
+ // 拿不到槽位:短暫等待別的執行緒解除 pin(每層開頭會 unpin_all)。
1617
+ // 這裡「重試」而不是直接回 -1,因為回 -1 會讓呼叫端**跳過這個 expert**,
1618
+ // 結果是靜默少算一條專家分支 —— 那是最嚴重的錯誤。
1619
+ // 2000 次 × 250 µs = 最多等 0.5 秒;真的撐不到那麼久,代表 arena 被
1620
+ // 預取的 busy 槽位佔住了(實測:跨層預取開啟時會發生)。
1621
+ for (int attempt = 0; attempt < 2000 && s < 0; attempt++) {
1622
+ lk.unlock();
1623
+ std::this_thread::sleep_for(std::chrono::microseconds(250));
1624
+ lk.lock();
1625
+ s = admit_slot_locked(il, ie);
1626
+ }
1627
+ if (s < 0) {
1628
+ fprintf(stderr,
1629
+ "[sdq] 嚴重:找不到能放 il=%d ie=%d(need_big=%d)的槽位,"
1630
+ "大槽 %zu/小槽 %zu\n",
1631
+ il, ie, (int) need_big_slot(il, ie), n_big_, n_small_);
1632
+ // 這裡「回 -1」會讓呼叫端**跳過這個 expert**,結果是靜默少算一條專家分支
1633
+ // —— 那是整個引擎最嚴重的錯誤型態(模型會安靜地給出錯誤答案,而且
1634
+ // 每次執行還不一定一樣)。寧可整個行程死掉,也不要產出「看起來正常」
1635
+ // 的錯誤結果。
1636
+ fprintf(stderr,
1637
+ "[sdq] 致命:無法取得 expert 槽位。為了不給出靜默錯誤的結果,"
1638
+ "直接中止。請回報這個情況(多半是預取把槽位佔成 busy 太久)。\n");
1639
+ std::fflush(stderr);
1640
+ abort();
1641
+ }
1642
+ }
1643
+ slot_state & st = slots_[s];
1644
+ if (st.layer >= 0) {
1645
+ page_.erase(((int64_t) st.layer << 20) | (int64_t) st.expert);
1646
+ stats_.evictions.fetch_add(1, std::memory_order_relaxed);
1647
+ used_.fetch_sub(1, std::memory_order_relaxed);
1648
+ }
1649
+ st.layer = il;
1650
+ st.expert = ie;
1651
+ st.last_use = ++clock_;
1652
+ st.pin = 1;
1653
+ st.is_hot = hot_lookup(il, ie);
1654
+ // **先標 busy,再 publish 到 page_。**
1655
+ // 順序反過來的話(先 publish 再標 busy 中間有一個窗口)另一條執行緒會把它
1656
+ // 當成命中,然後在資料還沒讀完時就拿去做矩陣乘法 → 非決定性的錯誤輸出。
1657
+ // acquire() 看到 busy 會等;admit/victim 本來就會跳過 busy 的槽位。
1658
+ st.busy = true;
1659
+ page_[((int64_t) il << 20) | (int64_t) ie] = s;
1660
+
1661
+ lk.unlock();
1662
+
1663
+ const uint64_t t0 = now_us();
1664
+ const bool ok = read_expert_par(il, ie, slot_ptr(s));
1665
+ const uint64_t dt = now_us() - t0;
1666
+
1667
+ std::unique_lock<std::mutex> lk2(mu_);
1668
+ st.busy = false;
1669
+ if (!ok) {
1670
+ st.layer = -1;
1671
+ st.expert = -1;
1672
+ st.pin = 0;
1673
+ page_.erase(((int64_t) il << 20) | (int64_t) ie);
1674
+ return -1;
1675
+ }
1676
+ used_.fetch_add(1, std::memory_order_relaxed);
1677
+ stats_.demand_wait_us.fetch_add(dt, std::memory_order_relaxed);
1678
+ const layer_desc * L = layer(il);
1679
+ if (L) {
1680
+ const uint64_t nb = (uint64_t) L->experts[(size_t) ie].bytes;
1681
+ stats_.ssb_read_bytes_demand.fetch_add(nb, std::memory_order_relaxed);
1682
+ if (lstats_ && il >= 0 && il < (int) info_.n_layer) { // SDQ-0004
1683
+ lstats_[il].demand_bytes.fetch_add(nb, std::memory_order_relaxed);
1684
+ }
1685
+ }
1686
+ return s;
1687
+ }
1688
+
1689
+ void pager::prefetch_expert(int il, int ie) {
1690
+ if (!ready_ || !cfg_.prefetch) {
1691
+ return;
1692
+ }
1693
+ {
1694
+ std::unique_lock<std::mutex> lk(mu_);
1695
+ if (find_slot_locked(il, ie) >= 0) {
1696
+ stats_.prefetch_duplicate.fetch_add(1, std::memory_order_relaxed);
1697
+ return;
1698
+ }
1699
+ // 預取只能占用「空槽」或「已經很久沒被用到的頁」。
1700
+ // (實測:無差別淘汰會讓命中率從 90% 掉到 27%,而預取使用率是 0%,純粹有害。)
1701
+ //
1702
+ // **槽位一定要選對尺寸**:need_big_slot() 為真的 expert 拿到小槽位會寫出邊界。
1703
+ const uint64_t stale_before = clock_ - (uint64_t) env_size("SDQ_PREFETCH_STALE_TOKENS", 2) * 320;
1704
+ // 預取佇列也不能太深:SSD 讀不夠快時,排在後面的預取早就過期了
1705
+ if (pf_q_.size() > (size_t) env_size("SDQ_PREFETCH_QUEUE", 64)) {
1706
+ return;
1707
+ }
1708
+ // **必須留夠空的槽位給 demand。**
1709
+ // 預取會把槽位標成 busy 並佔著它讀取;大槽位本來就少(工作集下限是
1710
+ // n_big_layers × n_expert_used),如果預取把最後幾個空槽都吃掉,
1711
+ // 下一層的 demand 就會「找不到能放 il=34 ie=114 的槽位」而中止。
1712
+ // (這是 verify 新增的預取壓力測試 [2b] 抓到的實際情況。)
1713
+ // 所以:對應尺寸的空槽少於 n_expert_used 個時,這一筆預取直接不做。
1714
+ const size_t reserve = (size_t) info_.n_expert_used;
1715
+ const size_t n_free_right = need_big_slot(il, ie) ? free_big_slots_.size() : free_slots_.size();
1716
+ if (n_free_right < reserve) {
1717
+ stats_.prefetch_duplicate.fetch_add(1, std::memory_order_relaxed);
1718
+ return;
1719
+ }
1720
+ const int32_t s = victim_stale_for(il, ie, stale_before);
1721
+ if (s < 0) {
1722
+ return;
1723
+ }
1724
+ slot_state & st = slots_[s];
1725
+ if (st.layer >= 0) {
1726
+ page_.erase(((int64_t) st.layer << 20) | (int64_t) st.expert);
1727
+ stats_.evictions.fetch_add(1, std::memory_order_relaxed);
1728
+ used_.fetch_sub(1, std::memory_order_relaxed);
1729
+ }
1730
+ st.layer = il;
1731
+ st.expert = ie;
1732
+ st.busy = true;
1733
+ st.pin = 0;
1734
+ pf_q_.push_back({s, il, ie, now_us()});
1735
+ stats_.prefetch_issued.fetch_add(1, std::memory_order_relaxed);
1736
+ }
1737
+ pf_cv_.notify_one();
1738
+ }
1739
+
1740
+ void pager::pf_worker() {
1741
+ std::unique_lock<std::mutex> lk(pf_mu_);
1742
+ for (;;) {
1743
+ pf_cv_.wait(lk, [this] { return pf_stop_ || !pf_q_.empty(); });
1744
+ if (pf_stop_ && pf_q_.empty()) {
1745
+ return;
1746
+ }
1747
+ const pf_item item = pf_q_.front();
1748
+ pf_q_.pop_front();
1749
+ lk.unlock();
1750
+
1751
+ // 這個槽可能已經被淘汰/重用 → 檢查 generation
1752
+ {
1753
+ std::unique_lock<std::mutex> g(mu_);
1754
+ const slot_state & st = slots_[item.slot];
1755
+ const bool still = st.busy && st.layer == item.il && st.expert == item.ie;
1756
+ if (!still) {
1757
+ lk.lock();
1758
+ continue;
1759
+ }
1760
+ }
1761
+
1762
+ // 預取也走 I/O 段平行:單獨一條執行緒循序讀三段,是這條路徑唯一的瓶頸。
1763
+ const bool ok = cfg_.io_parallel ? read_expert_par(item.il, item.ie, slot_ptr(item.slot))
1764
+ : read_expert(item.il, item.ie, slot_ptr(item.slot));
1765
+ const layer_desc * L = layer(item.il);
1766
+ {
1767
+ std::unique_lock<std::mutex> g(mu_);
1768
+ slot_state & st = slots_[item.slot];
1769
+ if (st.busy && st.layer == item.il && st.expert == item.ie) {
1770
+ st.busy = false;
1771
+ st.last_use = ++clock_;
1772
+ if (ok) {
1773
+ page_[((int64_t) item.il << 20) | (int64_t) item.ie] = item.slot;
1774
+ used_.fetch_add(1, std::memory_order_relaxed);
1775
+ stats_.prefetch_completed.fetch_add(1, std::memory_order_relaxed);
1776
+ // SDQ-0004: tag the page so the demand path can credit it,
1777
+ // and so unused prefetched bytes can be reported as waste.
1778
+ st.from_prefetch = true;
1779
+ if (lstats_ && item.il >= 0 && item.il < (int) info_.n_layer) {
1780
+ lstats_[item.il].prefetch_bytes.fetch_add(
1781
+ (uint64_t) slot_size_of(item.slot), std::memory_order_relaxed);
1782
+ }
1783
+ if (L) {
1784
+ stats_.ssb_read_bytes_prefetch.fetch_add(
1785
+ (uint64_t) L->experts[(size_t) item.ie].bytes, std::memory_order_relaxed);
1786
+ }
1787
+ } else {
1788
+ st.layer = -1;
1789
+ st.expert = -1;
1790
+ }
1791
+ }
1792
+ }
1793
+ lk.lock();
1794
+ }
1795
+ }
1796
+
1797
+ void pager::release(int s) {
1798
+ if (s < 0 || (size_t) s >= n_slots_) {
1799
+ return;
1800
+ }
1801
+ std::lock_guard<std::mutex> lk(mu_);
1802
+ slot_state & st = slots_[(size_t) s];
1803
+ if (st.pin > 0) {
1804
+ st.pin--;
1805
+ }
1806
+ }
1807
+
1808
+ void pager::unpin_all() {
1809
+ std::unique_lock<std::mutex> lk(mu_);
1810
+ for (auto & st : slots_) {
1811
+ st.pin = 0;
1812
+ }
1813
+ }
1814
+
1815
+ void pager::note_used(int il, int ie) {
1816
+ if (il < 0 || ie < 0) {
1817
+ return;
1818
+ }
1819
+ std::lock_guard<std::mutex> lk(hist_mu_);
1820
+ if (il >= (int) cur_used_.size()) {
1821
+ return;
1822
+ }
1823
+ auto & v = cur_used_[(size_t) il];
1824
+ if (std::find(v.begin(), v.end(), ie) == v.end()) {
1825
+ v.push_back(ie);
1826
+ }
1827
+ }
1828
+
1829
+ // 「下一層」預取:只在 layer L 的 op 開頭呼叫一次。
1830
+ //
1831
+ // 為什麼需要這個函式:end_token() 是**整個 token 的 40 層都算完之後**才被呼叫的
1832
+ // (sdq_cli.cpp 在 llama_decode 之後呼叫),所以它排的預取完全沒有和任何計算重疊
1833
+ // —— 量測上 prefetch 使用率只有 0~8 %,等於整條預取路徑是關的。
1834
+ // 在 layer L 開始時就替 layer L+1 排預取,才有一整層的時間(讀+算,約 4 ms)
1835
+ // 可以讓 SSD 讀取躲在後面。
1836
+ //
1837
+ // 預測用的是 prev_used_[il+1](上一個 token 在該層選的專家)—— MoE 的路由在
1838
+ // 鄰近 token 之間非常穩定,這是最便宜且不用訓練的預測器。
1839
+ void pager::prefetch_layer_ahead(int il) {
1840
+ if (!ready_ || !cfg_.prefetch || il < 0 || il >= (int) layers_.size()) {
1841
+ return;
1842
+ }
1843
+ int32_t ids[64];
1844
+ const int k = (int) info_.n_expert_used;
1845
+ if (k <= 0 || k > 64) {
1846
+ return;
1847
+ }
1848
+ predict(il, ids, k);
1849
+ for (int i = 0; i < k; i++) {
1850
+ prefetch_expert(il, ids[i]);
1851
+ }
1852
+ }
1853
+
1854
+ // SDQ-0005: called once per llama_decode() from any front end.
1855
+ //
1856
+ // sdq-chat drives this itself from main(); llama-server does not, which left
1857
+ // tokens at 0, ssd_bytes_per_token undefined and SDQ_STATS_FILE empty. This is
1858
+ // the single place that closes a decode step, so hook it here.
1859
+ //
1860
+ // n_tokens > 1 means prefill: a whole prompt was just paged in. Counting that as
1861
+ // one "token" would make ssd_bytes_per_token meaningless, and feeding it to the
1862
+ // Markov table would poison the transition counts with a whole-batch transition.
1863
+ // So prefill advances the byte accounting by n_tokens but leaves the routing
1864
+ // history alone -- it is not a single-token transition.
1865
+ void pager::note_decode(int n_tokens) {
1866
+ if (!ready_ || n_tokens <= 0) {
1867
+ return;
1868
+ }
1869
+ stats_.tokens.fetch_add((uint64_t) n_tokens, std::memory_order_relaxed);
1870
+ if (n_tokens == 1) {
1871
+ end_token();
1872
+ }
1873
+ dump_stats("decode");
1874
+ }
1875
+
1876
+ void pager::end_token() {
1877
+ if (!ready_) {
1878
+ return;
1879
+ }
1880
+ // 用「上個 token 的選擇」更新一階 Markov 轉移表
1881
+ {
1882
+ std::lock_guard<std::mutex> lk(hist_mu_);
1883
+ for (size_t il = 0; il < cur_used_.size(); il++) {
1884
+ auto & cur = cur_used_[il];
1885
+ if (cur.empty()) {
1886
+ continue;
1887
+ }
1888
+ auto & layer_hist = hist_[il];
1889
+ // 以每層最近一次(或平均)的前一次選擇為前驅
1890
+ if (!prev_used_.empty() && il < prev_used_.size() && !prev_used_[il].empty()) {
1891
+ for (int32_t from : prev_used_[il]) {
1892
+ trans & t = layer_hist[(size_t) from];
1893
+ for (int32_t to : cur) {
1894
+ int found = -1;
1895
+ for (int32_t i = 0; i < t.n; i++) {
1896
+ if (t.to[i] == to) {
1897
+ t.cnt[i]++;
1898
+ found = i;
1899
+ break;
1900
+ }
1901
+ }
1902
+ if (found < 0 && t.n < 64) {
1903
+ t.to[t.n] = to;
1904
+ t.cnt[t.n] = 1;
1905
+ t.n++;
1906
+ }
1907
+ }
1908
+ }
1909
+ }
1910
+ prev_used_[il] = cur;
1911
+ cur.clear();
1912
+ }
1913
+ }
1914
+
1915
+ // 依預測排預取
1916
+ if (!cfg_.prefetch) {
1917
+ return;
1918
+ }
1919
+ int32_t ids[64];
1920
+ for (size_t il = 0; il < layers_.size(); il++) {
1921
+ const int k = (int) info_.n_expert_used;
1922
+ predict((int) il, ids, k);
1923
+ for (int i = 0; i < k; i++) {
1924
+ prefetch_expert((int) il, ids[i]);
1925
+ }
1926
+ }
1927
+ }
1928
+
1929
+ void pager::predict(int il, int32_t * out, int k) {
1930
+ if (il < 0 || il >= (int) prev_used_.size()) {
1931
+ for (int i = 0; i < k; i++) {
1932
+ out[i] = i;
1933
+ }
1934
+ return;
1935
+ }
1936
+ // 以最近一次選擇為前驅;沒有歷史就直接沿用上一次的選擇
1937
+ const std::vector<int32_t> & prev = prev_used_[(size_t) il];
1938
+ std::vector<std::pair<uint32_t, int32_t>> scored;
1939
+ if (prev.empty()) {
1940
+ for (int i = 0; i < k; i++) {
1941
+ out[i] = i;
1942
+ }
1943
+ return;
1944
+ }
1945
+ for (int32_t from : prev) {
1946
+ const trans & t = hist_[(size_t) il][(size_t) from];
1947
+ for (int32_t i = 0; i < t.n; i++) {
1948
+ scored.push_back({t.cnt[i], t.to[i]});
1949
+ }
1950
+ }
1951
+ if (scored.empty()) {
1952
+ for (size_t i = 0; i < prev.size() && i < (size_t) k; i++) {
1953
+ out[i] = prev[i];
1954
+ }
1955
+ for (int i = (int) prev.size(); i < k; i++) {
1956
+ out[i] = i;
1957
+ }
1958
+ return;
1959
+ }
1960
+ std::sort(scored.begin(), scored.end(), [](auto & a, auto & b) {
1961
+ return a.first > b.first;
1962
+ });
1963
+ for (int i = 0; i < k && i < (int) scored.size(); i++) {
1964
+ out[i] = scored[(size_t) i].second;
1965
+ }
1966
+ }
1967
+
1968
+ double pager::hit_rate() const {
1969
+ const uint64_t req = stats_.expert_requests.load();
1970
+ const uint64_t hit = stats_.expert_hits.load();
1971
+ return req ? (double) hit / (double) req : 0.0;
1972
+ }
1973
+
1974
+ void pager::shutdown() {
1975
+ if (!ready_) {
1976
+ return;
1977
+ }
1978
+ if (!profile_out_.empty()) {
1979
+ std::lock_guard<std::mutex> lk(mu_);
1980
+ dump_profile(profile_out_);
1981
+ }
1982
+ if (pf_.joinable()) {
1983
+ {
1984
+ std::unique_lock<std::mutex> lk(pf_mu_);
1985
+ pf_stop_ = true;
1986
+ }
1987
+ pf_cv_.notify_all();
1988
+ pf_.join();
1989
+ }
1990
+ io_pool_stop();
1991
+ for (auto & fd : shard_fd_) {
1992
+ if (fd >= 0) {
1993
+ close(fd);
1994
+ }
1995
+ }
1996
+ for (auto & fd : shard_fd_direct_) {
1997
+ if (fd >= 0) {
1998
+ close(fd);
1999
+ }
2000
+ }
2001
+ shard_fd_.clear();
2002
+ shard_fd_direct_.clear();
2003
+ ready_ = false;
2004
+ }
2005
+
2006
+ pager::~pager() {
2007
+ shutdown();
2008
+ if (arena_) {
2009
+ // 注意:arena 是 posix_memalign 配置的,必須用 free() 釋放,
2010
+ // 用 munmap() 會破壞 glibc 的 heap(之後任何 free 都會報 double free)。
2011
+ free(arena_);
2012
+ arena_ = nullptr;
2013
+ }
2014
+ }
2015
+
2016
+ static double pct(uint64_t a, uint64_t b) {
2017
+ return b ? 100.0 * (double) a / (double) b : 0.0;
2018
+ }
2019
+
2020
+ void pager::print_summary() const {
2021
+ if (!ready_) {
2022
+ return;
2023
+ }
2024
+ // bailingmoe3 的第 0 層是 dense FFN(沒有 expert),要抓第一個真的有分頁的層
2025
+ const layer_desc * L0 = nullptr;
2026
+ int first_moe = -1;
2027
+ for (size_t i = 0; i < layers_.size(); i++) {
2028
+ if (!layers_[i].experts.empty()) {
2029
+ L0 = &layers_[i];
2030
+ first_moe = (int) i;
2031
+ break;
2032
+ }
2033
+ }
2034
+ if (L0 && !L0->experts.empty()) {
2035
+ const expert_desc & e = L0->experts[0];
2036
+ fprintf(stderr, "[sdq] page-map: n_layer=%lld n_expert=%lld n_expert_used=%lld n_embd=%lld n_ff_exp=%lld"
2037
+ " | L%dE0 gate=%llu/%llu up=%llu/%llu down=%llu/%llu (off/size) ne0=%lld ne1=%lld d_ne0=%lld d_ne1=%lld\n",
2038
+ (long long) info_.n_layer, (long long) info_.n_expert, (long long) info_.n_expert_used,
2039
+ (long long) info_.n_embd, (long long) info_.n_ff_exp, first_moe,
2040
+ (unsigned long long) e.seg[part::GATE].off, (unsigned long long) e.seg[part::GATE].size,
2041
+ (unsigned long long) e.seg[part::UP].off, (unsigned long long) e.seg[part::UP].size,
2042
+ (unsigned long long) e.seg[part::DOWN].off, (unsigned long long) e.seg[part::DOWN].size,
2043
+ (long long) e.ne0, (long long) e.ne1, (long long) e.down_ne0, (long long) e.down_ne1);
2044
+ }
2045
+ fprintf(stderr,
2046
+ "[sdq] expert pager ready: %s\n"
2047
+ "[sdq] 熱門 expert 釘選:%d 個%s\n"
2048
+ "[sdq] slot=%.2f MiB (%zu B) slots=%zu arena=%.2f GiB (RAM 實測 RSS=%.2f GiB)\n",
2049
+ cfg_.model_path.c_str(),
2050
+ n_hot_, n_hot_ > 0 ? "(永不淘汰)" : "(無剖面,用動態 LRU/LFU)",
2051
+ slot_bytes_ / 1048576.0, slot_bytes_, n_slots_, arena_bytes_ / 1073741824.0,
2052
+ process_rss_bytes() / 1073741824.0);
2053
+ if (n_big_) {
2054
+ // 大小兩種槽位:把配置結果講清楚,否則只看「slots=N」會誤以為每個槽一樣大。
2055
+ fprintf(stderr,
2056
+ "[sdq] 槽位分兩種:小 %.2f MiB × %zu + 大 %.2f MiB × %zu(平均 %.2f MiB)\n",
2057
+ small_bytes_ / 1048576.0, n_small_,
2058
+ big_bytes_ / 1048576.0, n_big_,
2059
+ (double) arena_bytes_ / (double) (n_slots_ ? n_slots_ : 1) / 1048576.0);
2060
+ }
2061
+ fprintf(stderr,
2062
+ "[sdq] prefetch=%s fadvise_dontneed=%d\n",
2063
+ cfg_.prefetch ? "on" : "off", (int) cfg_.fadvise_dontneed);
2064
+ }
2065
+
2066
+ void pager::dump_stats(const char * phase) {
2067
+ if (cfg_.stats_path.empty()) {
2068
+ return;
2069
+ }
2070
+ const uint64_t tokens = stats_.tokens.load();
2071
+ const uint64_t req = stats_.expert_requests.load();
2072
+ const uint64_t hit = stats_.expert_hits.load();
2073
+ const uint64_t issued = stats_.prefetch_issued.load();
2074
+ const uint64_t used = stats_.prefetch_used.load();
2075
+ const uint64_t b_dem = stats_.ssb_read_bytes_demand.load();
2076
+ const uint64_t b_pre = stats_.ssb_read_bytes_prefetch.load();
2077
+ // SDQ-0004: prefetched bytes that were consumed, so the rest is real waste.
2078
+ const uint64_t b_pre_used = stats_.prefetch_bytes_used.load();
2079
+
2080
+ std::ostringstream o;
2081
+ o << "{\"phase\":\"" << phase << "\""
2082
+ << ",\"tokens\":" << tokens
2083
+ << ",\"expert_requests\":" << req
2084
+ << ",\"expert_hits\":" << hit
2085
+ << ",\"expert_misses\":" << (req - hit)
2086
+ << ",\"hit_rate\":" << pct(hit, req)
2087
+ << ",\"ssb_demand_bytes\":" << b_dem
2088
+ << ",\"ssb_prefetch_bytes\":" << b_pre
2089
+ << ",\"ssb_total_bytes\":" << (b_dem + b_pre)
2090
+ << ",\"ssd_bytes_per_token\":" << (tokens ? (double) (b_dem + b_pre) / (double) tokens : 0.0)
2091
+ << ",\"prefetch_issued\":" << issued
2092
+ << ",\"prefetch_used\":" << used
2093
+ << ",\"prefetch_used_rate\":" << pct(used, issued)
2094
+ // SDQ-0004: prefetch waste. b_pre counts every byte the background reader
2095
+ // pulled in; b_pre_used counts the ones a demand request actually hit.
2096
+ // The difference is what made docs/07's negative result worse than "slower".
2097
+ << ",\"prefetch_bytes_used\":" << b_pre_used
2098
+ << ",\"prefetch_bytes_wasted\":" << (b_pre > b_pre_used ? b_pre - b_pre_used : 0)
2099
+ << ",\"prefetch_waste_rate\":" << pct(b_pre > b_pre_used ? b_pre - b_pre_used : 0, b_pre)
2100
+ << ",\"prefetch_duplicate\":" << stats_.prefetch_duplicate.load()
2101
+ << ",\"evictions\":" << stats_.evictions.load()
2102
+ << ",\"demand_wait_us\":" << stats_.demand_wait_us.load()
2103
+ << ",\"prefetch_wait_us\":" << stats_.prefetch_wait_us.load()
2104
+ << ",\"arena_bytes\":" << arena_bytes_
2105
+ // 大小兩種槽位之後,「用掉幾個槽位」不能直接乘 slot_bytes_(那是大的那種)。
2106
+ // dump_stats 只在階段轉換時呼叫一次,遍歷兩千幾個槽位的成本可以忽略。
2107
+ << ",\"arena_used_bytes\":" << used_bytes_snapshot()
2108
+ << ",\"rss_bytes\":" << process_rss_bytes()
2109
+ << ",\"n_slots\":" << n_slots_
2110
+ << ",\"n_small_slots\":" << n_small_
2111
+ << ",\"n_big_slots\":" << n_big_
2112
+ << ",\"small_slot_bytes\":" << small_bytes_
2113
+ << ",\"big_slot_bytes\":" << big_bytes_
2114
+ << ",\"slot_bytes\":" << slot_bytes_
2115
+ << ",\"n_moe_layers\":" << n_moe_layers_
2116
+ << ",\"expert_bytes_total\":" << expert_bytes_total_
2117
+ << ",\"arch\":\"" << info_.arch << "\""
2118
+ << ",\"name\":\"" << info_.name << "\""
2119
+ // SDQ-0004: per-layer breakdown. docs/10 needs this to tell "the device is
2120
+ // slow" apart from "one layer keeps evicting everyone else's experts".
2121
+ << ",\"per_layer\":[";
2122
+ for (int64_t il = 0; lstats_ && il < (int64_t) info_.n_layer; il++) {
2123
+ const uint64_t lreq = lstats_[il].req.load();
2124
+ const uint64_t lhit = lstats_[il].hit.load();
2125
+ o << (il ? "," : "")
2126
+ << "{\"layer\":" << il
2127
+ << ",\"req\":" << lreq
2128
+ << ",\"hit\":" << lhit
2129
+ << ",\"miss\":" << (lreq - lhit)
2130
+ << ",\"hit_rate\":" << pct(lhit, lreq)
2131
+ << ",\"demand_bytes\":" << lstats_[il].demand_bytes.load()
2132
+ << ",\"prefetch_bytes\":" << lstats_[il].prefetch_bytes.load()
2133
+ << "}";
2134
+ }
2135
+ o << "]" << "}";
2136
+ std::ofstream f(cfg_.stats_path, std::ios::app);
2137
+ if (f) {
2138
+ f << o.str() << "\n";
2139
+ }
2140
+ }
2141
+
2142
+ } // namespace sdq
2143
+ namespace sdq {
2144
+ // 由 CLI 呼叫:依環境變數/參數初始化 pager
2145
+ bool sdq_pager_active_for_layer(int il) {
2146
+ pager & P = pager::instance();
2147
+ if (!P.ready() || P.layer(il) == nullptr) {
2148
+ return false;
2149
+ }
2150
+ // 除錯用:SDQ_MOE_LO / SDQ_MOE_HI 可以只讓部分層走分頁路徑,
2151
+ // 用來二分搜尋第一個發生數值差異的層。
2152
+ static const int lo = [] { const char * e = getenv("SDQ_MOE_LO"); return e ? atoi(e) : 0; }();
2153
+ static const int hi = [] { const char * e = getenv("SDQ_MOE_HI"); return e ? atoi(e) : 1 << 30; }();
2154
+ if (il < lo || il >= hi) {
2155
+ return false;
2156
+ }
2157
+ // SDQ_MOE_SKIP=3,7:這些層即使在 [lo,hi) 內也走原生路徑。
2158
+ // 搭配 SDQ_BUFT_KEEP(讓該層的 expert 留在 RAM)就能做同一層的 A/B 比對。
2159
+ static const std::set<int> skip = [] {
2160
+ std::set<int> s;
2161
+ const char * e = getenv("SDQ_MOE_SKIP");
2162
+ if (e) {
2163
+ for (const char * p = e; *p; ) {
2164
+ const char * q = strchr(p, ',');
2165
+ const size_t n = q ? (size_t) (q - p) : strlen(p);
2166
+ if (n) {
2167
+ s.insert(atoi(std::string(p, n).c_str()));
2168
+ }
2169
+ if (!q) {
2170
+ break;
2171
+ }
2172
+ p = q + 1;
2173
+ }
2174
+ }
2175
+ return s;
2176
+ }();
2177
+ return skip.find(il) == skip.end();
2178
+ }
2179
+
2180
+ void sdq_on_decode(int n_tokens) {
2181
+ auto & inst = pager::instance();
2182
+ if (inst.ready()) {
2183
+ inst.note_decode(n_tokens);
2184
+ }
2185
+ }
2186
+
2187
+ bool sdq_init(const std::string & model_path) {
2188
+ config c;
2189
+ c.model_path = model_path;
2190
+ c.arena_bytes = env_size("SDQ_ARENA_MB", 0) << 20;
2191
+ c.ram_budget_bytes = env_size("SDQ_RAM_BUDGET_MB", 0) << 20;
2192
+ c.slot_bytes = env_size("SDQ_SLOT_KB", 0) << 10;
2193
+ // 預設關閉。docs/07 的實測是「開 5.68 / 關 6.20 tok/s」的負結果,而且
2194
+ // 預取中的槽位會讓 acquire() 靜默跳過 expert,曾造成輸出不一致(stage S7)。
2195
+ // 這裡原本是 true,等於讓那個負結果從來沒有真正生效。
2196
+ c.prefetch = env_flag("SDQ_PREFETCH", false);
2197
+ // ---- OPT-1:熱門 expert 剖面(Strata 式)----
2198
+ if (const char * pi = getenv("SDQ_PROFILE_IN")) { c.profile_in = pi; }
2199
+ if (const char * po = getenv("SDQ_PROFILE_OUT")) { c.profile_out = po; }
2200
+ c.hot_pin = (int) env_size("SDQ_HOT_PIN", 0);
2201
+ c.admit_min_hits = (int) env_size("SDQ_ADMIT_MIN_HITS", 1);
2202
+ c.fadvise_dontneed = env_flag("SDQ_FADV_DONTNEED", true);
2203
+ c.io_parallel = env_flag("SDQ_IO_PARALLEL", true);
2204
+ // I/O 讀取執行緒數。sdq-chat 會在載入模型「之前」自己 setenv 這個值
2205
+ // (3 × 計算執行緒,上限 48,docs/05 量到序列 3.79 → 24 條 6.68 tok/s)。
2206
+ // 但 cfg_.n_threads 從來沒有被賦值,所以任何沒有自行設定的前端
2207
+ // (llama-server 就��)會拿到 0,然後在 io_pool_start() 被夾到下限 4 條 ——
2208
+ // 等於一直跑在「序列讀」那一端。改成在 pager 自己算,量測與 sdq-chat 一致。
2209
+ if (!getenv("SDQ_IO_THREADS")) {
2210
+ const int hw = (int) std::max(1u, std::thread::hardware_concurrency());
2211
+ const int compute = hw < 8 ? hw : 8; // 與 sdq-chat 同一個上限 8
2212
+ c.io_threads = compute * 3;
2213
+ } else {
2214
+ c.io_threads = (int) env_size("SDQ_IO_THREADS", 0);
2215
+ }
2216
+ c.io_split = (int) env_size("SDQ_IO_SPLIT", 1);
2217
+ c.verbose = env_flag("SDQ_VERBOSE", false);
2218
+ const char * sp = getenv("SDQ_STATS_FILE");
2219
+ if (sp) {
2220
+ c.stats_path = sp;
2221
+ }
2222
+ sdq_set_config(c);
2223
+ try {
2224
+ pager::instance().init(c, nullptr);
2225
+ } catch (const std::exception & e) {
2226
+ fprintf(stderr, "[sdq] pager 初始化失敗:%s\n", e.what());
2227
+ return false;
2228
+ }
2229
+ // 長駐前端(llama-server)不會呼叫 sdq_shutdown();用 atexit 在收尾時補上摘要。
2230
+ // 註冊時機必須在 pager::instance() 建構之後:static 物件的析構與 atexit 依
2231
+ // 「後註冊者先執行」,反過來就會在 pager 已經析構後才跑這段(讀到已釋放的物件)。
2232
+ static std::once_flag once_exit;
2233
+ std::call_once(once_exit, [] { std::atexit([] { sdq_shutdown(); }); });
2234
+ return pager::instance().ready();
2235
+ }
2236
+
2237
+
2238
+ // ---------------------------------------------------------------- 除錯:dump 任意 F32 張量
2239
+ // 目的:讓「原生 mul_mat_id 路徑」與「SDQ 分頁路徑」輸出同一層的 MoE 結果,
2240
+ // 兩邊各自 dump 一次就能直接 A/B 對照,判定到底是哪一邊算錯。
2241
+ namespace {
2242
+ struct dump_arg { std::string path; };
2243
+
2244
+ void dump_op(ggml_tensor * dst, int ith, int /*nth*/, void * userdata) {
2245
+ if (ith != 0) {
2246
+ return;
2247
+ }
2248
+ const ggml_tensor * src = dst->src[0];
2249
+ dump_arg * a = (dump_arg *) userdata;
2250
+ FILE * f = fopen(a->path.c_str(), "wb");
2251
+ if (!f) {
2252
+ fprintf(stderr, "[sdq-dump] 無法開啟 %s\n", a->path.c_str());
2253
+ return;
2254
+ }
2255
+ const int64_t meta[6] = { src->ne[0], src->ne[1], src->ne[2], src->ne[3],
2256
+ (int64_t) ggml_nbytes(src), 0 };
2257
+ fwrite(meta, sizeof(int64_t), 6, f);
2258
+ // 只支援 F32 contiguous;其他情況就明確報錯,不要假設
2259
+ if (src->type != GGML_TYPE_F32 || !ggml_is_contiguous(src)) {
2260
+ fprintf(stderr, "[sdq-dump] %s 不是 contiguous F32(type=%d)\n",
2261
+ a->path.c_str(), (int) src->type);
2262
+ } else {
2263
+ fwrite(src->data, sizeof(float), (size_t) (ggml_nelements(src)), f);
2264
+ }
2265
+ fclose(f);
2266
+ fprintf(stderr, "[sdq-dump] %s (%lld 個元素)\n", a->path.c_str(),
2267
+ (long long) ggml_nelements(src));
2268
+ }
2269
+ } // namespace
2270
+
2271
+ ggml_tensor * sdq_maybe_dump_tensor(ggml_context * ctx, ggml_tensor * t, int il) {
2272
+ const char * prefix = getenv("SDQ_MOE_OUT_DUMP");
2273
+ if (!prefix || !t) {
2274
+ return t;
2275
+ }
2276
+ int layer = getenv("SDQ_MOE_OUT_DUMP_LAYER") ? atoi(getenv("SDQ_MOE_OUT_DUMP_LAYER")) : -1;
2277
+ if (layer >= 0 && il != layer) {
2278
+ return t;
2279
+ }
2280
+ dump_arg * a = new dump_arg{ std::string(prefix) + "." + std::to_string(il) };
2281
+ return ggml_custom_4d(ctx, GGML_TYPE_F32, t->ne[0], t->ne[1], t->ne[2], t->ne[3],
2282
+ &t, 1, dump_op, 1, a);
2283
+ }
2284
+
2285
+ void sdq_shutdown() {
2286
+ pager & P = pager::instance();
2287
+ sdq_print_time_profile();
2288
+ P.dump_stats("shutdown");
2289
+ P.print_summary();
2290
+ P.shutdown();
2291
+ }
2292
+ } // namespace sdq
sdq-ling/sdq_pager.h ADDED
@@ -0,0 +1,471 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ // SDQ expert pager:SSD → RAM → CPU 的 MoE 專家分層快取
2
+ //
3
+ // 設計要點
4
+ // * 分頁單位 = 一個 expert(layer, expert id),大小 ~1.8 MB(三個矩陣:gate/up/down)
5
+ // * RAM 端是一個固定槽位的 arena(slab),每槽 2 MiB,O(1) 配置、不碎片化
6
+ // * 淘汰策略:LRU + 本 token 使用中(pinned)的保護
7
+ // * 預取:依「上一個 token 的專家選擇」用一階 Markov 轉移預測下一個 token 的專家,
8
+ // 交給背景執行緒 pread(),量測「預取命中率 / 預取浪費量」
9
+ // * 指標:hit/miss、SSD 讀取位元組、demand 停滯時間、prefetch 領先時間、arena 使用量
10
+ //
11
+ // 不使用 swap:所有權重資料只存在 arena(實體 RAM)與檔案(SSD)兩處。
12
+
13
+ #pragma once
14
+
15
+ #include <ggml.h>
16
+ #include <ggml-backend.h>
17
+ #include "llama.h"
18
+
19
+ #include <atomic>
20
+ #include <chrono>
21
+ #include <condition_variable>
22
+ #include <cstddef>
23
+ #include <cstdint>
24
+ #include <deque>
25
+ #include <memory>
26
+ #include <mutex>
27
+ #include <string>
28
+ #include <thread>
29
+ #include <unordered_map>
30
+ #include <vector>
31
+
32
+ namespace sdq {
33
+
34
+ // ---------------------------------------------------------------- GGUF 讀取
35
+ struct gguf_info {
36
+ int64_t n_layer = 0;
37
+ int64_t n_expert = 0;
38
+ int64_t n_expert_used = 0;
39
+ int64_t n_embd = 0;
40
+ int64_t n_ff_exp = 0;
41
+ int64_t n_ff_shexp = 0;
42
+ int64_t n_head = 0;
43
+ int64_t n_head_kv = 0;
44
+ int64_t n_embd_head_k = 0;
45
+ int64_t n_embd_head_v = 0;
46
+ int64_t context_length = 0;
47
+ int64_t full_attention_interval = 4;
48
+ // bailingmoe3 / Ling-3.0-flash
49
+ int64_t n_layer_dense_lead = 0;
50
+ int64_t n_expert_groups = 0;
51
+ int64_t n_expert_group_used = 0;
52
+ int64_t is_recr[512] = {0};
53
+ std::string arch;
54
+ std::string name;
55
+ };
56
+
57
+ // 一個 expert 槽位:gate/up/down 三段(有些模型把 gate+up 合成 gate_up)
58
+ struct seg_desc {
59
+ uint64_t off = 0; // 在「所屬 shard 檔案內」的位元組偏移
60
+ uint64_t size = 0; // 位元組數
61
+ // sharded GGUF:同一層的三段不一定在同一個檔案裡(unsplit 後 100+ GB 的模型一定會被切),
62
+ // 所以每一段都要記得自己屬於哪一個 shard,pread 才知道要開哪個 fd。
63
+ int shard = 0;
64
+ };
65
+
66
+ enum part : int { GATE = 0, UP = 1, DOWN = 2, COUNT = 3 };
67
+
68
+ struct expert_desc {
69
+ seg_desc seg[3];
70
+ int64_t slot_off[3] = {0, 0, 0}; // 三段在 arena 槽位內的相對偏移
71
+ int64_t ne0 = 0; // gate/up row 長度(= n_embd)
72
+ int64_t ne1 = 0; // gate/up row 數(= n_ff_exp)
73
+ int64_t down_ne0 = 0; // down row 長度(= n_ff_exp)
74
+ int64_t down_ne1 = 0; // down row 數(= n_embd)
75
+ ggml_type type[3] = {GGML_TYPE_COUNT, GGML_TYPE_COUNT, GGML_TYPE_COUNT};
76
+ int64_t bytes = 0; // 三段總大小
77
+ bool fused_gate_up = false;
78
+ };
79
+
80
+ // 一層的所有 expert 描述(40 層 × 256 個)
81
+ struct layer_desc {
82
+ std::vector<expert_desc> experts;
83
+ int64_t bytes_per_expert = 0;
84
+ // 這個層的 expert 比「大多數層」大(Q4_K/Q5_K vs Q6_K)→ 需要大槽位。
85
+ // 不然的話所有槽位都得按最大的那種配,白白浪費 6.8% 的 arena。
86
+ bool needs_big_slot = false;
87
+ };
88
+
89
+ // ---------------------------------------------------------------- 指標
90
+ struct stats {
91
+ std::atomic<uint64_t> expert_requests{0}; // 每個 (layer, token) 需要幾個 expert
92
+ std::atomic<uint64_t> expert_hits{0};
93
+ std::atomic<uint64_t> expert_misses{0};
94
+ std::atomic<uint64_t> ssb_read_bytes_demand{0};
95
+ std::atomic<uint64_t> ssb_read_bytes_prefetch{0};
96
+ std::atomic<uint64_t> prefetch_issued{0};
97
+ std::atomic<uint64_t> prefetch_completed{0};
98
+ std::atomic<uint64_t> prefetch_used{0}; // 預取後真的被用到
99
+ std::atomic<uint64_t> prefetch_duplicate{0};// 預取時已經在 RAM 裡
100
+ std::atomic<uint64_t> evictions{0};
101
+ std::atomic<uint64_t> demand_wait_us{0}; // 因為缺頁而停下來等 SSD 的時間
102
+ std::atomic<uint64_t> prefetch_wait_us{0};
103
+ std::atomic<uint64_t> prefetch_bytes_used{0}; // prefetched bytes a demand request actually hit
104
+ std::atomic<uint64_t> tokens{0};
105
+ std::atomic<uint64_t> moe_calls{0};
106
+ std::atomic<uint64_t> flops{0};
107
+ };
108
+
109
+ // ---- SDQ-0004: per-layer cache counters (Phase 0: is any layer starved?)
110
+ struct layer_stat {
111
+ std::atomic<uint64_t> req{0};
112
+ std::atomic<uint64_t> hit{0};
113
+ std::atomic<uint64_t> miss{0};
114
+ std::atomic<uint64_t> demand_bytes{0};
115
+ std::atomic<uint64_t> prefetch_bytes{0};
116
+ };
117
+
118
+
119
+ // ---------------------------------------------------------------- pager
120
+ struct config {
121
+ size_t arena_bytes = 0; // 0 = 自動(依 RAM 預算)
122
+ size_t slot_bytes = 0; // 0 = 自動(依模型推導)
123
+ int n_threads = 0;
124
+ bool prefetch = true;
125
+ int prefetch_ahead_layers = 0; // >0 時限制同時在飛的預取數量
126
+ bool fadvise_dontneed = true; // 讀完丟掉 kernel page cache,讓 RAM 帳目誠實
127
+ bool verbose = false;
128
+ uint64_t ram_budget_bytes = 0; // 0 = 自動(實體 RAM 的一小部分)
129
+ std::string model_path;
130
+ std::string stats_path;
131
+ int slot_bias = 0;
132
+
133
+ // ---- I/O 執行緒池 ----
134
+ // 為什麼需要:一個 expert 在檔案上是 gate / up / down 三段「不相鄰」的資料,
135
+ // 循序讀要 3 次 pread,而單一 pread 在本機量到約 7 ms(1.9 MB / 275 MB/s)。
136
+ // 也就是說「讀一個 expert」的延遲完全由這三段循序決定,執行緒再多也沒用。
137
+ // 把三段交給三條執行緒同時讀,單一 expert 的延遵掉到約 1/2~1/3。
138
+ // (實測:8 個 expert 全部 miss 的一層,序列讀 12.6 ms → 段平行 7.6 ms。)
139
+ bool io_parallel = true; // SDQ_IO_PARALLEL=0 可關(回到序列讀)
140
+ int io_threads = 0; // 0 = 自動(3 × n_threads,上限 48)
141
+ int io_split = 1; // SDQ_IO_SPLIT:每一段再切成幾塊並行讀(實測 1 最好)
142
+
143
+ // ---- Strata 式「熱門 expert 剖面」(OPT-1)----
144
+ // 離線用代表性文字跑一次,把每個 (layer, expert) 的請求次數記成剖面;
145
+ // 執行時把排行榜前 hot_pin 個釘成「熱頁」:熱頁可以被讀取,但**永遠不被淘汰**,
146
+ // 淘汰時一律先從冷頁挑。這解決的是「prefill 的一次性專家把熱區洗掉」。
147
+ std::string profile_in; // 讀取剖面(SDQ_PROFILE_IN)
148
+ std::string profile_out; // 結束時輸出剖面(SDQ_PROFILE_OUT)
149
+ int hot_pin = 0; // 釘選前幾名(SDQ_HOT_PIN)
150
+ int admit_min_hits = 1; // 請求次數低於此值就不收進快取(SDQ_ADMIT_MIN_HITS)
151
+ };
152
+
153
+ class pager {
154
+ public:
155
+ static pager & instance();
156
+
157
+ // 從 GGUF 檔頭建立 expert 頁面表,並配置 RAM arena
158
+ void init(const config & cfg, ggml_context * ctx);
159
+
160
+ bool ready() const { return ready_; }
161
+ size_t arena_bytes() const { return arena_bytes_; }
162
+ size_t slot_bytes() const { return slot_bytes_; }
163
+ size_t n_slots() const { return n_slots_; }
164
+ size_t small_bytes() const { return small_bytes_; }
165
+ size_t big_bytes() const { return big_bytes_; }
166
+ size_t n_small_slots() const { return n_small_; }
167
+ size_t n_big_slots() const { return n_big_; }
168
+ int n_moe_layers() const { return n_moe_layers_; }
169
+ uint64_t expert_bytes_total() const { return expert_bytes_total_; }
170
+ const gguf_info & info() const { return info_; }
171
+
172
+ // 取得 layer 描述(idx 越界或沒資料回 nullptr)
173
+ const layer_desc * layer(int il) const;
174
+
175
+ // 一次把「連續編號的一串 expert」讀進 RAM:同一層的三個矩陣(gate/up/down)本來
176
+ // 就各自連續存放,所以 k 個連續 expert 只需要 3 次大讀取(而不是 3k 次)。
177
+ // 對高延遲的儲存裝置(每個 request 都貴)這是最大的單項加速。
178
+ // experts 必須已依編號遞增排序;slots[i] 是 experts[i] 對應的槽位(由呼叫端配置)
179
+ // 回傳實際讀取的位元組數。
180
+ uint64_t read_expert_span(int il, const int32_t * experts, int n, const int32_t * slots);
181
+
182
+ // 只配置槽位(不讀資料):回傳槽位索引,-1 表示沒有空槽
183
+ int reserve_slot(int il, int ie);
184
+
185
+ // 查這個 expert 是否已經在 RAM:命中則 pin 並回傳槽位,未命中回傳 -1(不配置)
186
+ int probe(int il, int ie);
187
+
188
+ // 記錄一次合併讀取的位元組數(給指標用)
189
+ void note_span_read(uint64_t bytes);
190
+
191
+ // 確保 (layer, expert) 在 RAM,回傳槽位索引;-1 表示失敗
192
+ // pinned: 本 token 使用中,不可被淘汰;blocking: 缺頁時是否同步讀 SSD
193
+ int acquire(int il, int ie, bool blocking);
194
+
195
+ // 背景預取:把 page 讀進 arena(非同步),依歷史預測下一個 token 會用到的 expert
196
+ void prefetch_expert(int il, int ie);
197
+ // 「下一層」預取:在 layer L 的 op 開頭呼叫,替 layer L+1 排預取。
198
+ // 與 end_token() 的差別在時機 —— 後者在一個 token 的 40 層全算完才跑,
199
+ // 那時排的預取沒有任何計算可以重疊,實測使用率只有 0~8 %。
200
+ void prefetch_layer_ahead(int il);
201
+
202
+ // 每層計算開始時呼叫:解除上一層的 pin(本層用到的槽位才受保護)
203
+ void unpin_all();
204
+ // 用完一個槽位後立刻放掉 pin(成對呼叫 acquire())。
205
+ //
206
+ // 為什麼需要:原本 pin 是「整層釘住、下一層開頭才 unpin_all」。mode C 的
207
+ // expert_forward 回傳之後就再也不讀那個槽位(最後的歸約只讀各執行緒自己的
208
+ // partial 累加緩衝),所以繼續釘著只是白白把工作集撐大。
209
+ // 撐大的後果很具體:prefill 一個 ubatch 有 128 個 token,一層最多需要
210
+ // 256 個不同的 expert → 整層釘住 = 256 個槽位不能回收。大小兩種槽位之下,
211
+ // 大槽位只有約 n_big_layers �� n_expert_used 個,於是**必定**在 prefill 時
212
+ // 「找不到能放 il=34 ie=… 的槽位」而中止。
213
+ void release(int s);
214
+ void note_used(int il, int ie);
215
+ void end_token();
216
+ // ---- SDQ-0005: one call per llama_decode() from any front end.
217
+ void note_decode(int n_tokens);
218
+
219
+ // 依歷史預測某層下一個 token 的 top-k expert(回傳信心 0..1)
220
+ void predict(int il, int32_t * out, int k);
221
+
222
+ void shutdown();
223
+ stats & get_stats() { return stats_; }
224
+
225
+ // 把當前指標寫成一行 JSON(append 到 stats_path)
226
+ void dump_stats(const char * phase);
227
+ void print_summary() const;
228
+
229
+ // arena 內的槽位指標
230
+ // 槽位索引空間是統一的:[0, n_small_) 是小槽位(連續),
231
+ // [n_small_, n_slots_) 是大槽位(接在小槽位區之後)。
232
+ uint8_t * slot_ptr(int s) {
233
+ return (size_t) s < n_small_
234
+ ? arena_ + (size_t) s * small_bytes_
235
+ : arena_ + small_bytes_ * n_small_ + (size_t)(s - (int) n_small_) * big_bytes_;
236
+ }
237
+ size_t slot_size_of(int s) const {
238
+ return (size_t) s < n_small_ ? small_bytes_ : big_bytes_;
239
+ }
240
+ bool need_big_slot(int il, int ie) const;
241
+
242
+ // 這個槽位算不算「大槽位」。
243
+ //
244
+ // 判斷依據是**槽位的實際大小**,不是索引 >= n_small_:
245
+ // 統一尺寸(SDQ_FORCE_UNIFORM_SLOT=1,或模型只有一種 expert 大小)時,
246
+ // 所有槽位的索引都 < n_small_ 但大小都是大的 —— 用索引判斷會讓大 expert
247
+ // 永遠配不到槽位,整個推論卡在 admit 的重試迴圈裡。
248
+ bool is_big_slot(int s) const {
249
+ return big_bytes_ && slot_size_of(s) >= big_bytes_;
250
+ }
251
+
252
+ size_t used_bytes() const { return used_bytes_snapshot(); }
253
+ double hit_rate() const;
254
+
255
+ // arena 裡實際被占用的位元組數。大小兩種槽位之後不能用「槽位數 × slot_bytes_」
256
+ // 計算(slot_bytes_ 是大的那一種),所以真的把槽位走一遍。
257
+ // 呼叫端:dump_stats(每次解碼階段結束)與 print_summary,不是熱路徑。
258
+ size_t used_bytes_snapshot() const {
259
+ size_t b = 0;
260
+ for (size_t s = 0; s < slots_.size(); s++) {
261
+ if (slots_[s].layer >= 0) {
262
+ b += slot_size_of((int) s);
263
+ }
264
+ }
265
+ return b;
266
+ }
267
+
268
+ // 讀 SSD(同步,給 acquire 用)
269
+ bool read_expert(int il, int ie, uint8_t * dst);
270
+
271
+ // ---- I/O 執行緒池(見 config::io_parallel 的說明)----
272
+ // 一個「讀取批次」的完成狀態;放在呼叫端的 stack 上,worker 只在 remaining==0
273
+ // 之後才會再碰它,所以離開 read_expert_par() 之後不會有 dangling pointer。
274
+ struct io_ctx {
275
+ std::atomic<int> remaining{0};
276
+ bool ok = true;
277
+ };
278
+ struct io_job {
279
+ int shard;
280
+ uint64_t off;
281
+ uint8_t * dst;
282
+ size_t size;
283
+ io_ctx * ctx;
284
+ };
285
+
286
+ void io_pool_start();
287
+ void io_pool_stop();
288
+ void io_worker(size_t wid);
289
+ bool read_expert_par(int il, int ie, uint8_t * dst);
290
+ bool read_range_staged(int shard, uint64_t off, uint8_t * dst, size_t size,
291
+ uint8_t * stage, size_t stage_sz);
292
+
293
+ // I/O 執行緒池每條 worker 自帶的 O_DIRECT 對齊緩衝(啟動時配置,已計入 RSS)
294
+ struct io_stage { uint8_t * buf = nullptr; size_t sz = 0; };
295
+ std::vector<io_stage> io_stage_;
296
+
297
+ bool io_pool_on_ = false;
298
+ int io_threads_ = 0;
299
+ int io_split_ = 1;
300
+ std::vector<std::thread> io_pool_;
301
+ std::mutex io_mu_;
302
+ std::condition_variable io_cv_; // 有工作 / 要停
303
+ std::condition_variable io_done_cv_; // 某個批次讀完了
304
+ std::deque<io_job> io_q_;
305
+ bool io_stop_ = false;
306
+ std::atomic<uint64_t> io_jobs_{0};
307
+ std::atomic<uint64_t> io_wait_ns_{0};
308
+
309
+ private:
310
+ pager() = default;
311
+ ~pager();
312
+
313
+ struct slot_state {
314
+ int32_t layer = -1;
315
+ int32_t expert = -1;
316
+ uint64_t last_use = 0; // LRU 記時
317
+ uint32_t freq = 0; // 這個 expert 被用到幾次(衰減式;用來做 LFU 混合策略)
318
+ uint32_t pin = 0; // >0 時不可淘汰
319
+ bool busy = false; // 背景執行緒正在寫入
320
+ bool is_hot = false; // 屬於剖面排行榜:淘汰時最後才會被選中
321
+ bool from_prefetch = false; // page was filled by the prefetcher
322
+ };
323
+
324
+ struct pf_item {
325
+ int slot;
326
+ int il;
327
+ int ie;
328
+ uint64_t issue_time_us;
329
+ };
330
+
331
+ void touch_freq_locked(slot_state & st);
332
+ // ---- OPT-1:剖面 / 熱頁 ----
333
+ void count_request(int il, int ie); // 記一次請求(給剖面用)
334
+ bool load_profile(const std::string & path); // 讀剖面並標記前 hot_pin 名
335
+ void dump_profile(const std::string & path); // 輸出排行榜
336
+ bool hot_lookup(int il, int ie) const;
337
+ void load_gguf_header(const std::string & path);
338
+ void build_page_map(const std::string & path);
339
+ void alloc_arena();
340
+ int find_slot_locked(int il, int ie);
341
+ int admit_slot_locked(int il, int ie);
342
+ int victim_locked() { return admit_slot_locked(-1, -1); }
343
+ int victim_stale_locked(uint64_t stale_before);
344
+ int victim_stale_for(int il, int ie, uint64_t stale_before);
345
+ void pf_worker();
346
+ bool read_range(int shard, uint64_t off, uint8_t * dst, size_t size);
347
+ bool read_range_direct(int shard, uint64_t off, uint8_t * dst, size_t size);
348
+ void open_model_fd();
349
+
350
+ // ---- sharded GGUF:模型被切成多個 -0000N-of-000M.gguf,每個檔有自己的
351
+ // header、自己的 data_start 與自己的張量子集。pager 必須同時開著全部的 fd,
352
+ // 並把每個張量的「檔內偏移」記成 (shard, off)。
353
+ std::vector<std::string> shard_paths_;
354
+ std::vector<int> shard_fd_;
355
+ std::vector<int> shard_fd_direct_;
356
+ std::vector<uint64_t> shard_data_start_;
357
+ bool use_direct_ = false;
358
+ std::string shard_path(int i) const {
359
+ return (i >= 0 && i < (int) shard_paths_.size()) ? shard_paths_[(size_t) i] : std::string();
360
+ }
361
+ // 依 GGUF 的 split 命名規則列舉同目錄的所有 shard
362
+ static std::vector<std::string> list_splits(const std::string & first_path, int n_split);
363
+
364
+ bool ready_ = false;
365
+ config cfg_;
366
+ gguf_info info_;
367
+ std::vector<layer_desc> layers_;
368
+ int n_moe_layers_ = 0;
369
+ uint64_t expert_bytes_total_ = 0;
370
+
371
+ uint8_t * arena_ = nullptr;
372
+ size_t arena_bytes_ = 0;
373
+ size_t slot_bytes_ = 0; // = max(small_bytes_, big_bytes_),保留給舊的訊息輸出
374
+ size_t small_bytes_ = 0; // 小槽位大小(大多數層用)
375
+ size_t big_bytes_ = 0; // 大槽位大小;0 = 這個模型沒有大層
376
+ size_t n_small_ = 0;
377
+ size_t n_big_ = 0;
378
+ size_t n_slots_ = 0;
379
+ mutable std::mutex mu_;
380
+ std::vector<slot_state> slots_;
381
+ std::unordered_map<int64_t, int32_t> page_; // (layer<<16|expert) -> slot
382
+ std::vector<int32_t> free_slots_; // 空的小槽位
383
+ std::vector<int32_t> free_big_slots_;
384
+
385
+ // ---- OPT-1:剖面 / 熱頁(Strata 式:靜態排行榜 + 熱頁不淘汰)----
386
+ std::vector<uint32_t> prof_; // 每個 (layer, expert) 的累計請求次數
387
+ std::unique_ptr<layer_stat[]> lstats_; // SDQ-0004
388
+ void ensure_layer_stats(int n_layer); // SDQ-0004
389
+ std::vector<uint8_t> hot_; // 是否在排行榜前 hot_pin 名
390
+ int n_hot_ = 0;
391
+ std::string profile_out_;
392
+ uint32_t freq_tick_ = 0;
393
+ std::atomic<size_t> used_{0};
394
+ uint64_t clock_ = 0;
395
+
396
+ // 背景預取
397
+ std::thread pf_;
398
+ std::mutex pf_mu_;
399
+ std::condition_variable pf_cv_;
400
+ std::deque<pf_item> pf_q_;
401
+ bool pf_stop_ = false;
402
+ std::atomic<uint64_t> pf_inflight_{0};
403
+
404
+ // 路由歷史(一階 Markov:上個 token 的 expert -> 這次被選到的次數)
405
+ struct trans { int32_t to[64]; uint32_t cnt[64]; int32_t n = 0; };
406
+ std::vector<std::vector<trans>> hist_; // [layer][prev_expert]
407
+ std::vector<std::vector<int32_t>> prev_used_; // [layer] 上一個 token 用到的 expert(預測用)
408
+ std::vector<std::vector<int32_t>> cur_used_; // [layer] 當前 token 用到的 expert
409
+ std::mutex hist_mu_; // 保護 cur_used_ / prev_used_ / hist_(多執行緒會同時 note_used)
410
+ std::atomic<uint64_t> hist_lock_{0};
411
+
412
+ stats stats_;
413
+ };
414
+
415
+ // ---------------------------------------------------------------- 工具
416
+ uint64_t now_us();
417
+ size_t process_rss_bytes();
418
+ size_t physical_ram_bytes();
419
+
420
+ // ---------------------------------------------------------------- ggml 介接
421
+ // 自訂 buffer type:expert 權重佔 0 個 RAM 位元組,資料留在檔案裡
422
+ ggml_backend_buffer_type_t sdq_expert_buft();
423
+
424
+ // 註冊 model params 的 tensor buffer override(expert 權重 → 我們的 buffer)
425
+ void sdq_install_buft_overrides(llama_model_params * params);
426
+
427
+ // 建立被分頁的 MoE FFN(替代 build_moe_ffn 裡的三次 mul_mat_id)
428
+ ggml_tensor * sdq_build_moe(ggml_context * ctx, ggml_tensor * cur,
429
+ ggml_tensor * ids, ggml_tensor * weights, int il,
430
+ int64_t n_expert, int64_t n_expert_used,
431
+ float swiglu_limit);
432
+
433
+ // 設定 arena 大小 / 預取開關等(CLI 在載入模型前呼叫)
434
+ bool sdq_init(const std::string & model_path);
435
+ void sdq_set_config(const config & c);
436
+ const config & sdq_get_config();
437
+ void sdq_set_enabled(bool on);
438
+ void sdq_shutdown();
439
+
440
+ bool sdq_enabled();
441
+
442
+ // SDQ-0005: bridge for front ends that are not sdq-chat (llama-server).
443
+ // Defined in sdq_pager.cpp; safe to call when the pager was never initialised.
444
+ void sdq_on_decode(int n_tokens);
445
+
446
+ // 效能計時探針的結果(SDQ_PROFILE_TIME=1)
447
+ void sdq_print_time_profile();
448
+
449
+ // 自我測試:算一次指定 expert 的 MoE ��出(給 NumPy 對照用)
450
+ int sdq_moe_selftest(int il, int ie, const float * x, float * out_io,
451
+ int64_t n_embd, int64_t n_ff, int n_tok);
452
+
453
+ // 建立我們的 MoE 自訂節點。
454
+ // **必須把 gf 傳進來並呼叫 ggml_build_forward_expand(gf, out)**:
455
+ // ggml 的排程器只會為「進了圖的節點」配置 compute buffer,沒註冊的節點其輸出
456
+ // 緩衝區會被 liveness 分析判成可重用,導致 layer>=1 的結果被其他張量覆蓋
457
+ // (實測症狀:layer 0 正確、layer 1 起整個錯掉)。
458
+ ggml_tensor * sdq_build_moe(ggml_cgraph * gf, ggml_context * ctx, ggml_tensor * cur,
459
+ ggml_tensor * ids, ggml_tensor * weights, int il,
460
+ int64_t n_expert, int64_t n_expert_used, float swiglu_limit);
461
+
462
+ // 這一層的分頁是否可用(graph build 時呼叫)
463
+ bool sdq_pager_active_for_layer(int il);
464
+
465
+ // 除錯/驗證用:把某個 F32 張量 dump 到檔案。
466
+ // 必須在「原生 mul_mat_id 路徑」與「SDQ 分頁路徑」兩邊都呼叫,才能做真正的 A/B 對照
467
+ // (env: SDQ_MOE_OUT_DUMP=prefix、SDQ_MOE_OUT_DUMP_LAYER=層編號,-1 = 全部)
468
+ ggml_tensor * sdq_maybe_dump_tensor(ggml_context * ctx, ggml_tensor * t, int il);
469
+
470
+
471
+ } // namespace sdq
sdq-ling/serve.sh ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ set -euo pipefail
3
+
4
+ ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
5
+ MODEL="${MODEL:-$ROOT/models/Ling-3.0-tiny-Q4_K_M.gguf}"
6
+ BIN="${BIN:-$ROOT/llama.cpp/build/bin/sdq-chat}"
7
+ RAM_BUDGET_MB="${RAM_BUDGET_MB:-4096}"
8
+ RESERVE_MB="${RESERVE_MB:-700}"
9
+ CTX="${CTX:-256}"
10
+ THREADS="${THREADS:-4}"
11
+ UBATCH="${UBATCH:-32}"
12
+
13
+ [[ -x "$BIN" ]] || { printf 'Missing %s; build the engine as documented in README.md\n' "$BIN" >&2; exit 1; }
14
+ [[ -s "$MODEL" ]] || { printf 'Missing model: %s\n' "$MODEL" >&2; exit 1; }
15
+
16
+ export SDQ_MODEL_PATH="$MODEL"
17
+ export SDQ_RAM_BUDGET_MB="$RAM_BUDGET_MB"
18
+ export SDQ_RESERVE_MB="$RESERVE_MB"
19
+ export SDQ_PREFETCH=0
20
+ export SDQ_FADV_DONTNEED=1
21
+ export SDQ_IO_THREADS="$((THREADS * 2))"
22
+ export LLAMA_ARG_N_GPU_LAYERS=0
23
+
24
+ exec "$BIN" -m "$MODEL" -c "$CTX" -t "$THREADS" -b "$UBATCH" \
25
+ -p "${PROMPT:-What is the capital of France?}" -n "${N_PREDICT:-48}" \
26
+ --temp 0 --no-interactive --ram-budget-mb "$RAM_BUDGET_MB"
sdq-ling/strict_bench.py ADDED
@@ -0,0 +1,336 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """Strict-limit RAM sweep for EPQ / sdq-ling on Ling-3.0-flash Q4_K_M.
3
+
4
+ Unlike the original memory_levels_ling.py (which only sets the pager's own
5
+ budget), every run here is launched under a *hard* OS limit:
6
+
7
+ prlimit --data=<level> --nofile=65536
8
+
9
+ On Linux >= 4.7 RLIMIT_DATA covers brk() AND mmap'd anonymous memory, so it is
10
+ a real ceiling on the arena + heap + every later anonymous allocation. If the
11
+ process exceeds it, malloc fails and the run aborts -- we record that as a hard
12
+ failure instead of a silent overshoot.
13
+
14
+ Two caveats this harness records explicitly:
15
+
16
+ * RLIMIT_DATA polices *anonymous* memory only. The ~1.3-2.1 GiB of
17
+ non-expert weights are mmap'd file-backed pages and are therefore NOT
18
+ policed here, so peak RSS can exceed the level. Both numbers are reported.
19
+ * `-march=native` on this host defines __AMX_INT8__, which makes llama.cpp
20
+ allocate a ~1.88 GiB AMX repack buffer before the arena-sized context. That
21
+ alone pushes the floor past 4 GiB, so the sweep runs against the
22
+ GGML_NATIVE=OFF (no-AMX) build unless --amx is passed.
23
+
24
+ Swap is proven absent at run time by sampling /proc/<pid>/status VmSwap; a
25
+ nonzero value fails the run.
26
+ """
27
+ from __future__ import annotations
28
+
29
+ import argparse
30
+ import json
31
+ import os
32
+ import platform
33
+ import re
34
+ import statistics
35
+ import subprocess
36
+ import sys
37
+ import threading
38
+ import time
39
+ from pathlib import Path
40
+
41
+ MIB = 1024 * 1024
42
+ GIB = 1024 ** 3
43
+
44
+ PROMPT = ("Write a detailed paragraph about how a mixture-of-experts model routes "
45
+ "tokens to expert weights, and why paging those weights from SSD matters.")
46
+
47
+
48
+ def args_config(args):
49
+ """JSON-safe view of the argparse namespace (drops the resolve helper)."""
50
+ return {k: v for k, v in vars(args).items() if not callable(v)}
51
+
52
+
53
+ def host_facts() -> dict:
54
+ cpu = ""
55
+ for line in Path('/proc/cpuinfo').read_text().splitlines():
56
+ if line.startswith('model name'):
57
+ cpu = line.split(':', 1)[1].strip()
58
+ break
59
+ mem_total = 0
60
+ for line in Path('/proc/meminfo').read_text().splitlines():
61
+ if line.startswith('MemTotal'):
62
+ mem_total = int(line.split()[1]) * 1024
63
+ break
64
+ cg = {}
65
+ for k, f in (('memory.max', 'memory.max'),
66
+ ('memory.swap.max', 'memory.swap.max'),
67
+ ('memory.high', 'memory.high')):
68
+ try:
69
+ cg[k] = Path('/sys/fs/cgroup/' + f).read_text().strip()
70
+ except Exception:
71
+ cg[k] = None
72
+ return {'cpu_model': cpu, 'cpu_threads': os.cpu_count(),
73
+ 'mem_total_bytes': mem_total, 'cgroup': cg}
74
+
75
+
76
+ class ProcSampler(threading.Thread):
77
+ """Sample RSS / AnonHWM / VmSwap until told to stop.
78
+
79
+ AnonHWM is smaps_rollup's own anonymous high-water mark and is the quantity
80
+ RLIMIT_DATA actually polices, so it is what we compare against the level.
81
+ """
82
+
83
+ def __init__(self, pid: int, interval: float = 0.03):
84
+ super().__init__(daemon=True)
85
+ self.pid = pid
86
+ self.interval = interval
87
+ self.stop_flag = threading.Event()
88
+ self.peak_rss = 0
89
+ self.peak_anon = 0
90
+ self.peak_swap = 0
91
+ self.n = 0
92
+
93
+ def _read(self) -> tuple[int, int, int]:
94
+ rss = anon = swp = 0
95
+ try:
96
+ for line in Path(f'/proc/{self.pid}/status').read_text().splitlines():
97
+ if line.startswith('VmRSS:'):
98
+ rss = int(line.split()[1]) * 1024
99
+ elif line.startswith('VmSwap:'):
100
+ swp = int(line.split()[1]) * 1024
101
+ except OSError:
102
+ pass
103
+ try:
104
+ for line in Path(f'/proc/{self.pid}/smaps_rollup').read_text().splitlines():
105
+ if line.startswith('Rss:'):
106
+ rss = max(rss, int(line.split()[1]) * 1024)
107
+ elif line.startswith('Anonymous:'):
108
+ anon = int(line.split()[1]) * 1024
109
+ elif line.startswith('AnonHWM:'):
110
+ anon = max(anon, int(line.split()[1]) * 1024)
111
+ except OSError:
112
+ pass
113
+ return rss, anon, swp
114
+
115
+ def run(self):
116
+ while not self.stop_flag.is_set():
117
+ r, a, s = self._read()
118
+ self.n += 1
119
+ self.peak_rss = max(self.peak_rss, r)
120
+ self.peak_anon = max(self.peak_anon, a)
121
+ self.peak_swap = max(self.peak_swap, s)
122
+ time.sleep(self.interval)
123
+ r, a, s = self._read()
124
+ self.peak_rss = max(self.peak_rss, r)
125
+ self.peak_anon = max(self.peak_anon, a)
126
+ self.peak_swap = max(self.peak_swap, s)
127
+
128
+
129
+ def evict(shards: list[str], helper: str) -> None:
130
+ subprocess.run(['sync'], check=False)
131
+ if helper:
132
+ subprocess.run([helper, '--evict'] + shards, check=False)
133
+
134
+
135
+ def run_once(args, level_mb: int, repeat: int) -> dict:
136
+ stats_file = Path(args.output_dir) / f'level_{level_mb}_r{repeat}.jsonl'
137
+ stats_file.unlink(missing_ok=True)
138
+ os.makedirs(args.output_dir, exist_ok=True)
139
+
140
+ inner = [
141
+ str(Path(args.bin).resolve()),
142
+ '-m', str(Path(args.model).resolve()),
143
+ '-p', PROMPT, '-n', str(args.n_predict), '-c', str(args.ctx),
144
+ '-t', str(args.threads), '-b', str(args.ubatch), '--temp', '0',
145
+ '--no-interactive', '--stats-file', str(stats_file),
146
+ '--ram-budget-mb', str(level_mb),
147
+ ]
148
+ # The strict ceiling: RLIMIT_DATA = arena + heap + all anon mmap.
149
+ cmd = ['prlimit', f'--data={level_mb * MIB}', '--nofile=65536'] + inner
150
+
151
+ env = dict(os.environ)
152
+ env.update({
153
+ 'SDQ_RAM_BUDGET_MB': str(level_mb),
154
+ 'SDQ_RESERVE_MB': str(args.reserve_mb_for(level_mb)),
155
+ 'SDQ_PREFETCH': '0',
156
+ 'SDQ_FADV_DONTNEED': '1',
157
+ 'SDQ_BENCH_NO_EOG': '1',
158
+ })
159
+
160
+ evict(args.shards, args.evict_helper)
161
+
162
+ t0 = time.perf_counter()
163
+ proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
164
+ text=True, env=env)
165
+ sampler = ProcSampler(proc.pid)
166
+ sampler.start()
167
+ try:
168
+ out, err = proc.communicate(timeout=args.timeout)
169
+ except subprocess.TimeoutExpired:
170
+ proc.kill()
171
+ out, err = proc.communicate()
172
+ sampler.stop_flag.set()
173
+ return {'level_mb': level_mb, 'repeat': repeat + 1,
174
+ 'error': f'timeout after {args.timeout}s'}
175
+ sampler.stop_flag.set()
176
+ sampler.join(timeout=2)
177
+ wall = time.perf_counter() - t0
178
+ both = out + '\n' + err
179
+
180
+ res = {
181
+ 'level_mb': level_mb,
182
+ 'level_gib': level_mb / 1024,
183
+ 'repeat': repeat + 1,
184
+ 'wall_s': round(wall, 2),
185
+ 'exit_code': proc.returncode,
186
+ 'peak_rss_mb': round(sampler.peak_rss / MIB, 1),
187
+ 'peak_anon_mb': round(sampler.peak_anon / MIB, 1),
188
+ 'peak_swap_mb': round(sampler.peak_swap / MIB, 1),
189
+ 'swap_free': sampler.peak_swap == 0,
190
+ 'anon_within_limit': sampler.peak_anon <= level_mb * MIB,
191
+ 'limit_enforced_by': 'prlimit --data (RLIMIT_DATA)',
192
+ }
193
+
194
+ if proc.returncode != 0:
195
+ sig = -proc.returncode if proc.returncode < 0 else None
196
+ res['signal'] = sig
197
+ res['status'] = 'OOM_OR_ABORT'
198
+ res['failure'] = ' | '.join(
199
+ [l.strip() for l in err.splitlines()
200
+ if re.search(r'bad_alloc|cannot allocate|MemoryError|std::terminate|'
201
+ r'abort|Killed|Expert arena|too small|error', l, re.I)][-5:])
202
+ res['wall_s'] = round(wall, 2)
203
+ return res
204
+
205
+ res['status'] = 'OK'
206
+
207
+ def grab(pat, text=both):
208
+ m = re.search(pat, text, re.M)
209
+ return m
210
+
211
+ d = grab(r'decode:\s*(\d+)\s+tok\s*/\s*([\d.]+)s\s*=\s*([\d.]+)\s+tok/s')
212
+ p = grab(r'prefill:\s*(\d+)\s+tok\s*/\s*([\d.]+)s\s*=\s*([\d.]+)\s+tok/s')
213
+ mem = grab(r'總 RSS\s+(\d+)\s+MB\s*/\s+峰值\s+(\d+)\s+MB\s*/\s+swap\s+(\d+)\s+MB')
214
+ split = grab(r'匿名 RSS\s+(\d+)\s+MB\s*/\s+檔案對映 RSS\s+(\d+)\s+MB\s*/\s+總 RSS\s+(\d+)\s+MB\s*/\s+峰值\s+(\d+)\s+MB')
215
+ arena = grab(r'arena\s+([\d.]+)\s+GiB')
216
+ slots = grab(r'slots=(\d+)')
217
+ ssd_tot = grab(r'全程 SSD 實際讀取:\s*([\d.]+)\s*GB')
218
+ mib = grab(r'SSD 讀\s*([\d.]+)\s*MB\s*\(([\d.]+)\s*MB/tok\)')
219
+
220
+ if d:
221
+ res['decode_tok_s'] = float(d.group(3))
222
+ res['decode_tok'] = int(d.group(1))
223
+ if p:
224
+ res['prefill_tok_s'] = float(p.group(3))
225
+ res['prefill_tok'] = int(p.group(1))
226
+ if mem:
227
+ res['final_rss_mb'] = int(mem.group(1))
228
+ res['selfpeak_rss_mb'] = int(mem.group(2))
229
+ res['self_swap_mb'] = int(mem.group(3))
230
+ if split:
231
+ res['rss_anon_mb'] = int(split.group(1))
232
+ res['rss_file_mb'] = int(split.group(2))
233
+ res['rss_total_mb'] = int(split.group(3))
234
+ res['selfpeak_rss_mb'] = int(split.group(4))
235
+ if arena:
236
+ res['arena_gib'] = float(arena.group(1))
237
+ if slots:
238
+ res['n_slots'] = int(slots.group(1))
239
+ if ssd_tot:
240
+ res['total_ssd_gb'] = float(ssd_tot.group(1))
241
+ if mib:
242
+ res['decode_ssd_mb'] = float(mib.group(1))
243
+ res['ssd_mb_per_token'] = float(mib.group(2))
244
+
245
+ rec = {}
246
+ if stats_file.exists():
247
+ for line in stats_file.read_text().splitlines():
248
+ r = json.loads(line)
249
+ if r.get('phase') == 'decode':
250
+ rec = r
251
+ stats_file.unlink(missing_ok=True)
252
+
253
+ pl = rec.get('per_layer') or []
254
+ rates = [x['hit_rate'] for x in pl if x.get('req', 0) > 0]
255
+ hr = rec.get('hit_rate')
256
+ res.update({
257
+ 'expert_requests': rec.get('expert_requests'),
258
+ 'expert_hits': rec.get('expert_hits'),
259
+ 'hot_rate_pct': hr,
260
+ 'cold_rate_pct': (100.0 - hr) if hr is not None else None,
261
+ 'evictions': rec.get('evictions'),
262
+ 'per_layer_hit_min': min(rates) if rates else None,
263
+ 'per_layer_hit_max': max(rates) if rates else None,
264
+ })
265
+
266
+ asst = ''
267
+ started = False
268
+ for line in out.splitlines():
269
+ if line.startswith('assistant:'):
270
+ started = True
271
+ rest = line[len('assistant:'):].strip()
272
+ if rest:
273
+ asst = rest
274
+ continue
275
+ if started and line.strip() and not line.startswith('['):
276
+ asst += ' ' + line.strip()
277
+ res['assistant'] = asst.strip()[:200]
278
+ return res
279
+
280
+
281
+ def main() -> None:
282
+ ap = argparse.ArgumentParser()
283
+ ap.add_argument('--model', required=True)
284
+ ap.add_argument('--shards', nargs='+', required=True)
285
+ ap.add_argument('--bin', required=True)
286
+ ap.add_argument('--levels-mb', type=int, nargs='+', default=[4096, 8192, 16384, 32768])
287
+ ap.add_argument('--repeat', type=int, default=2)
288
+ ap.add_argument('--ctx', type=int, default=2048)
289
+ ap.add_argument('--ubatch', type=int, default=128)
290
+ ap.add_argument('--threads', type=int, default=16)
291
+ ap.add_argument('--n-predict', type=int, default=48)
292
+ ap.add_argument('--reserve-mb', type=int, default=1024)
293
+ ap.add_argument('--timeout', type=int, default=1800)
294
+ ap.add_argument('--evict-helper', default='/tmp/pagecache_resident')
295
+ ap.add_argument(
296
+ '--reserve-map',
297
+ default='4096:2600,8192:3600,16384:3600,32768:3600',
298
+ help='per-level SDQ_RESERVE_MB, "level_mb:reserve_mb,...". Calibrated '
299
+ 'by measuring anon_hwm - arena, i.e. every anonymous allocation '
300
+ 'that happens after the arena is sized.')
301
+ ap.add_argument('--output-dir', default='/root/app/ling/results_strict')
302
+ ap.add_argument('--out-json', default='/root/app/ling/results_strict/strict_memory_levels.json')
303
+ args = ap.parse_args()
304
+
305
+ args._reserve = {}
306
+ for item in args.reserve_map.split(','):
307
+ k, v = item.split(':')
308
+ args._reserve[int(k)] = int(v)
309
+ args.reserve_mb_for = lambda lvl: args._reserve.get(lvl, 3600)
310
+
311
+ Path(args.output_dir).mkdir(parents=True, exist_ok=True)
312
+
313
+ print(f'host: {host_facts()}', file=sys.stderr)
314
+ runs = []
315
+ for lvl in args.levels_mb:
316
+ for r in range(args.repeat):
317
+ print(f'--- level {lvl} MB repeat {r + 1}/{args.repeat} ---',
318
+ file=sys.stderr, flush=True)
319
+ try:
320
+ res = run_once(args, lvl, r)
321
+ except Exception as e:
322
+ res = {'level_mb': lvl, 'repeat': r + 1, 'error': repr(e)}
323
+ runs.append(res)
324
+ print(json.dumps(res, ensure_ascii=False)[:400], file=sys.stderr, flush=True)
325
+ Path(args.out_json).write_text(json.dumps(
326
+ {'host': host_facts(), 'config': args_config(args), 'runs': runs},
327
+ indent=2, ensure_ascii=False))
328
+
329
+ print('DONE', file=sys.stderr)
330
+ Path(args.out_json).write_text(json.dumps(
331
+ {'host': host_facts(), 'config': args_config(args), 'runs': runs},
332
+ indent=2, ensure_ascii=False))
333
+
334
+
335
+ if __name__ == '__main__':
336
+ main()
sdq-ling/verify_dequant.py ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """Byte-level cross-check of the NumPy dequantizers against ggml's own.
3
+
4
+ `ggml_deq <type> <gguf> <offset> <n>` (built from sdq-ling/ggml_deq.cpp) dumps one
5
+ dequantized row using ggml's dequantize_row_*; this script compares it with the
6
+ NumPy implementation used by verify_ling.py / verify_moe_layer.py.
7
+ """
8
+ import importlib.util
9
+ import subprocess
10
+ import sys
11
+
12
+ import numpy as np
13
+
14
+ spec = importlib.util.spec_from_file_location('vl', str(__import__('pathlib').Path(__file__).with_name('verify_ling.py')))
15
+ vl = importlib.util.module_from_spec(spec)
16
+ spec.loader.exec_module(vl)
17
+
18
+ GGML_DEQ = str(__import__('pathlib').Path(__file__).resolve().parents[1] / 'llama.cpp/build/bin/ggml_deq')
19
+
20
+
21
+ def main():
22
+ kv, ts = vl.parse_headers()
23
+ cases = [(2, 'ffn_gate_exps.weight', 12), (2, 'ffn_up_exps.weight', 12),
24
+ (2, 'ffn_down_exps.weight', 14), (5, 'ffn_gate_exps.weight', 12),
25
+ (5, 'ffn_down_exps.weight', 14), (20, 'ffn_gate_exps.weight', 12),
26
+ (23, 'ffn_down_exps.weight', 14)]
27
+ all_ok = True
28
+ for il, name, tt in cases:
29
+ si, off, dims, t = ts[f'blk.{il}.{name}']
30
+ assert t == tt, (name, t, tt)
31
+ nrow = vl.row_bytes(dims[0], tt)
32
+ ne0 = dims[0]
33
+ for ie in (0, 17, 127):
34
+ eo = off + ie * nrow * dims[1]
35
+ out = subprocess.run([GGML_DEQ, str(tt), vl.SHARDS[si], str(eo), str(ne0)],
36
+ capture_output=True, check=True)
37
+ ref = np.frombuffer(out.stdout, dtype=np.float32)
38
+ with open(vl.SHARDS[si], 'rb') as f:
39
+ f.seek(eo)
40
+ mine = vl.dequant(f.read(nrow), ne0, tt)
41
+ d = float(np.abs(ref - mine).max())
42
+ ok = d == 0.0
43
+ all_ok &= ok
44
+ print(f'blk.{il}.{name} E{ie} type={vl.NAME.get(tt)} n={ne0}: '
45
+ f'max|ggml - numpy| = {d:g} {"PASS" if ok else "FAIL"}')
46
+ print('\nALL PASS' if all_ok else '\nSOME FAILED')
47
+ return 0 if all_ok else 1
48
+
49
+
50
+ if __name__ == '__main__':
51
+ sys.exit(main())
sdq-ling/verify_ling.py ADDED
@@ -0,0 +1,219 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """Validate the SDQ paged MoE kernel for Ling-3.0-flash (bailingmoe3) with pure NumPy.
3
+
4
+ The weights are read straight out of the sharded GGUF, dequantized with ggml's
5
+ definitions, and the SiLU-gated expert forward pass is recomputed in fp32. The
6
+ result is compared against `sdq-chat --selftest IL:IE:SEED:OUT.bin`.
7
+
8
+ Ling-3.0-flash expert tensors (Q4_K_M):
9
+ ffn_gate_exps [n_embd=2560, n_ff_exp=768, n_expert=512] q4_K
10
+ ffn_up_exps [2560, 768, 512] q4_K
11
+ ffn_down_exps [768, 2560, 512] q4_K or q6_K
12
+ """
13
+ import os
14
+ import struct
15
+ import sys
16
+
17
+ import numpy as np
18
+
19
+ QK_K = 256
20
+ MODEL = os.environ.get('MODEL', '/root/app/ling/tiny/models/Ling-3.0-tiny-Q4_K_M.gguf')
21
+ SHARDS = [MODEL.replace('00001-of', f'{i:05d}-of') for i in (1, 2)]
22
+
23
+ TS = {0: 1, 1: 1, 2: 2, 3: 2, 4: 4, 5: 4, 6: 4, 7: 1, 10: 8, 11: 8, 12: 8}
24
+ BLK = {12: (256, 144), 13: (256, 176), 14: (256, 210), 7: (32, 24), 8: (32, 34),
25
+ 2: (32, 18), 3: (32, 20), 6: (32, 22), 0: (1, 4), 1: (1, 2)}
26
+ FMT = {0: '<B', 1: '<b', 2: '<H', 3: '<h', 4: '<I', 5: '<i', 6: '<f', 7: '<B',
27
+ 10: '<Q', 11: '<q', 12: '<d'}
28
+ NAME = {12: 'q4_K', 13: 'q5_K', 14: 'q6_K', 7: 'q5_1', 8: 'q8_0', 6: 'q5_0', 2: 'q4_0'}
29
+
30
+
31
+ def rstr(f):
32
+ n = struct.unpack('<Q', f.read(8))[0]
33
+ return f.read(n)
34
+
35
+
36
+ def read_val(f, t):
37
+ if t == 8:
38
+ return rstr(f).decode('utf-8', 'replace')
39
+ if t == 9:
40
+ et = struct.unpack('<I', f.read(4))[0]
41
+ n = struct.unpack('<Q', f.read(8))[0]
42
+ if et == 8:
43
+ return [rstr(f).decode('utf-8', 'replace') for _ in range(n)]
44
+ buf = f.read(n * TS[et])
45
+ return np.frombuffer(buf, dtype=np.dtype(FMT[et])).tolist()
46
+ buf = f.read(TS[t])
47
+ return np.frombuffer(buf, dtype=np.dtype(FMT[t]))[0].item()
48
+
49
+
50
+ def parse_headers():
51
+ kv = {}
52
+ tensors = {}
53
+ for si, path in enumerate(SHARDS):
54
+ with open(path, 'rb') as f:
55
+ assert f.read(4) == b'GGUF'
56
+ f.read(4)
57
+ n_tensors = struct.unpack('<Q', f.read(8))[0]
58
+ n_kv = struct.unpack('<Q', f.read(8))[0]
59
+ for _ in range(n_kv):
60
+ k = rstr(f).decode()
61
+ kv[k] = read_val(f, struct.unpack('<I', f.read(4))[0])
62
+ align = kv.get('general.alignment', 32)
63
+ pending = []
64
+ for _ in range(n_tensors):
65
+ name = rstr(f).decode()
66
+ nd = struct.unpack('<I', f.read(4))[0]
67
+ dims = struct.unpack(f'<{nd}Q', f.read(8 * nd))
68
+ tt = struct.unpack('<I', f.read(4))[0]
69
+ off = struct.unpack('<Q', f.read(8))[0]
70
+ pending.append((name, dims, tt, off))
71
+ # data_start is only known after the whole tensor table has been read
72
+ data_start = (f.tell() + align - 1) // align * align
73
+ for name, dims, tt, off in pending:
74
+ tensors[name] = (si, data_start + off, dims, tt)
75
+ return kv, tensors
76
+
77
+
78
+ def fp16(x):
79
+ return np.frombuffer(x, dtype=np.float16)[0].astype(np.float32)
80
+
81
+
82
+ def dequant_q4_K(raw, n):
83
+ nb = n // QK_K
84
+ out = np.empty(n, dtype=np.float32)
85
+ for i in range(nb):
86
+ b = raw[i * 144:(i + 1) * 144]
87
+ d = fp16(b[0:2])
88
+ dmin = fp16(b[2:4])
89
+ sc = np.frombuffer(b, dtype=np.uint8, count=12, offset=4)
90
+ qs = np.frombuffer(b, dtype=np.uint8, count=128, offset=16)
91
+ base = i * QK_K
92
+ is_ = 0
93
+ for j in range(0, QK_K, 64):
94
+ if is_ < 4:
95
+ sc1, m1 = int(sc[is_] & 63), int(sc[is_ + 4] & 63)
96
+ sc2, m2 = int(sc[is_ + 1] & 63), int(sc[is_ + 5] & 63)
97
+ else:
98
+ sc1 = int((sc[is_ + 4] & 0xF) | ((sc[is_ - 4] >> 6) << 4))
99
+ m1 = int((sc[is_ + 4] >> 4) | ((sc[is_] >> 6) << 4))
100
+ sc2 = int((sc[is_ + 5] & 0xF) | ((sc[is_ - 3] >> 6) << 4))
101
+ m2 = int((sc[is_ + 5] >> 4) | ((sc[is_ + 1] >> 6) << 4))
102
+ q = qs[j // 2: j // 2 + 32]
103
+ out[base + j: base + j + 32] = d * sc1 * (q & 0xF) - dmin * m1
104
+ out[base + j + 32: base + j + 64] = d * sc2 * (q >> 4) - dmin * m2
105
+ is_ += 2
106
+ return out
107
+
108
+
109
+ def dequant_q6_K(raw, n):
110
+ nb = n // QK_K
111
+ out = np.empty(n, dtype=np.float32)
112
+ for i in range(nb):
113
+ b = raw[i * 210:(i + 1) * 210]
114
+ d = fp16(b[208:210])
115
+ ql = np.frombuffer(b, dtype=np.uint8, count=128, offset=0)
116
+ qh = np.frombuffer(b, dtype=np.uint8, count=64, offset=128)
117
+ sc = np.frombuffer(b, dtype=np.int8, count=16, offset=192)
118
+ base = i * QK_K
119
+ for g in range(2): # two 128-element groups per block
120
+ for l in range(32):
121
+ is_ = l // 16
122
+ q1 = ((ql[g * 64 + l] & 0xF) | (((qh[g * 32 + l] >> 0) & 3) << 4)).astype(np.int16) - 32
123
+ q2 = ((ql[g * 64 + l + 32] & 0xF) | (((qh[g * 32 + l] >> 2) & 3) << 4)).astype(np.int16) - 32
124
+ q3 = ((ql[g * 64 + l] >> 4) | (((qh[g * 32 + l] >> 4) & 3) << 4)).astype(np.int16) - 32
125
+ q4 = ((ql[g * 64 + l + 32] >> 4) | (((qh[g * 32 + l] >> 6) & 3) << 4)).astype(np.int16) - 32
126
+ off0 = base + g * 128
127
+ out[off0 + l] = d * sc[g * 8 + is_] * q1
128
+ out[off0 + l + 32] = d * sc[g * 8 + is_ + 2] * q2
129
+ out[off0 + l + 64] = d * sc[g * 8 + is_ + 4] * q3
130
+ out[off0 + l + 96] = d * sc[g * 8 + is_ + 6] * q4
131
+ return out
132
+
133
+
134
+ def dequant(raw, n, tt):
135
+ if tt == 12:
136
+ return dequant_q4_K(raw, n)
137
+ if tt == 14:
138
+ return dequant_q6_K(raw, n)
139
+ raise ValueError(f'unsupported type {tt} ({NAME.get(tt, tt)})')
140
+
141
+
142
+ def row_bytes(n, tt):
143
+ bs, ts = BLK[tt]
144
+ return n // bs * ts
145
+
146
+
147
+ def read_expert(tensors, il, ie):
148
+ def seg(name):
149
+ si, off, dims, tt = tensors[f'blk.{il}.{name}']
150
+ nrow = row_bytes(dims[0], tt)
151
+ n = dims[1] * nrow
152
+ with open(SHARDS[si], 'rb') as f:
153
+ f.seek(off + ie * n)
154
+ return f.read(n), nrow, tt, dims[0]
155
+ g = seg('ffn_gate_exps.weight')
156
+ u = seg('ffn_up_exps.weight')
157
+ d = seg('ffn_down_exps.weight')
158
+ return g, u, d
159
+
160
+
161
+ def load_matrix(seg, n_rows):
162
+ raw, nrow, tt, ne0 = seg
163
+ return np.stack([dequant(raw[r * nrow:(r + 1) * nrow], ne0, tt) for r in range(n_rows)])
164
+
165
+
166
+ def main():
167
+ cases = []
168
+ for path in sys.argv[1:]:
169
+ # st_l<IL>e<IE>.bin
170
+ base = os.path.basename(path)
171
+ body = base[len('st_l'):-len('.bin')]
172
+ il, ie = body.split('e')
173
+ cases.append((int(il), int(ie), path))
174
+ if not cases:
175
+ print('usage: verify_ling.py st_l<IL>e<IE>.bin ...')
176
+ return 1
177
+
178
+ kv, tensors = parse_headers()
179
+ n_embd = int(kv['bailingmoe3.embedding_length'])
180
+ n_ff = int(kv['bailingmoe3.expert_feed_forward_length'])
181
+ print(f"arch=bailingmoe3 n_expert={int(kv['bailingmoe3.expert_count'])} "
182
+ f"n_expert_used={int(kv['bailingmoe3.expert_used_count'])} "
183
+ f"n_embd={n_embd} n_ff_exp={n_ff}")
184
+
185
+ all_ok = True
186
+ for il, ie, path in cases:
187
+ raw = open(path, 'rb').read()
188
+ n_tok = len(raw) // 4 // (2 * n_embd)
189
+ got = np.frombuffer(raw[:n_tok * n_embd * 4], dtype=np.float32).reshape(n_tok, n_embd)
190
+ x = np.frombuffer(raw[n_tok * n_embd * 4:], dtype=np.float32).reshape(n_tok, n_embd)
191
+
192
+ g_seg, u_seg, d_seg = read_expert(tensors, il, ie)
193
+ Wg = load_matrix(g_seg, n_ff)
194
+ Wu = load_matrix(u_seg, n_ff)
195
+ Wd = load_matrix(d_seg, n_embd)
196
+
197
+ rels, corrs = [], []
198
+ for t in range(n_tok):
199
+ xt = x[t]
200
+ g = Wg @ xt
201
+ u = Wu @ xt
202
+ h = (g / (1.0 + np.exp(-g))) * u
203
+ ref = Wd @ h
204
+ denom = max(float(np.abs(ref).max()), 1e-9)
205
+ rels.append(float(np.abs(got[t] - ref).max()) / denom)
206
+ corrs.append(float(np.corrcoef(got[t], ref)[0, 1]))
207
+ rel = max(rels)
208
+ corr = min(corrs)
209
+ ok = rel < 0.05 and corr > 0.99
210
+ all_ok &= ok
211
+ print(f'L{il} E{ie}: gate={NAME[g_seg[2]]} up={NAME[u_seg[2]]} down={NAME[d_seg[2]]}'
212
+ f' | max_rel_err={rel:.4f} corr={corr:.6f} | {"PASS" if ok else "FAIL"}')
213
+
214
+ print('\nALL PASS' if all_ok else '\nSOME FAILED')
215
+ return 0 if all_ok else 1
216
+
217
+
218
+ if __name__ == '__main__':
219
+ sys.exit(main())
sdq-ling/verify_moe_layer.py ADDED
@@ -0,0 +1,154 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """Recompute a whole paged MoE layer in NumPy and compare with sdq-chat's output.
3
+
4
+ `sdq-chat` with SDQ_MOE_DUMP=<prefix> SDQ_MOE_DUMP_LAYER=<il> writes, from inside
5
+ the custom MoE op:
6
+ int32 meta[6] = { n_embd, n_tok, n_expert_used, n_expert, 1, 2 }
7
+ float cur [n_tok][n_embd] (FFN input)
8
+ int32 ids [n_tok][n_expert_used] (routed experts)
9
+ float wts [n_tok][n_expert_used] (routed weights)
10
+ float out [n_tok][n_embd] (SDQ MoE output)
11
+
12
+ This script reads the same expert weights straight out of the GGUF, runs the
13
+ SiLU-gated expert forward in fp32 and diffs the result, so the whole
14
+ pager + custom kernel + weight-scaling path is checked at once.
15
+ """
16
+ import importlib.util
17
+ import struct
18
+ import sys
19
+
20
+ import numpy as np
21
+
22
+ spec = importlib.util.spec_from_file_location('vl', str(__import__('pathlib').Path(__file__).with_name('verify_ling.py')))
23
+ vl = importlib.util.module_from_spec(spec)
24
+ spec.loader.exec_module(vl)
25
+
26
+ QK_K = 256
27
+
28
+
29
+ def dequant_q4_K_vec(raw, n_elem):
30
+ """raw: uint8 [..., 144] -> float32 [..., 256]"""
31
+ raw = raw.reshape(-1, 144)
32
+ nb = raw.shape[0]
33
+ out = np.empty((nb, QK_K), dtype=np.float32)
34
+ d = raw[:, 0:2].copy().view(np.float16).astype(np.float32)[:, 0]
35
+ dmin = raw[:, 2:4].copy().view(np.float16).astype(np.float32)[:, 0]
36
+ b = np.ascontiguousarray(raw[:, 4:16]) # 12 packed 6-bit scale bytes
37
+ sc = np.empty((nb, 8), dtype=np.int32)
38
+ mn = np.empty((nb, 8), dtype=np.int32)
39
+ for j in range(4):
40
+ sc[:, j] = b[:, j] & 63
41
+ mn[:, j] = b[:, j + 4] & 63
42
+ for j in range(4, 8):
43
+ sc[:, j] = (b[:, j + 4] & 0xF) | ((b[:, j - 4] >> 6) << 4)
44
+ mn[:, j] = (b[:, j + 4] >> 4) | ((b[:, j] >> 6) << 4)
45
+ qs = raw[:, 16:144].reshape(nb, 4, 32).astype(np.int32)
46
+ q = np.empty((nb, QK_K), dtype=np.int32)
47
+ for j in range(4):
48
+ q[:, j * 64:j * 64 + 32] = qs[:, j, :] & 0xF
49
+ q[:, j * 64 + 32:j * 64 + 64] = qs[:, j, :] >> 4
50
+ gidx = np.repeat(np.arange(8), 32) # 32-value group -> (sc, mn) index
51
+ out[:] = d[:, None] * sc[:, gidx] * q - dmin[:, None] * mn[:, gidx]
52
+ return out
53
+
54
+
55
+ def dequant_q6_K_vec(raw, n_elem):
56
+ raw = raw.reshape(-1, 210)
57
+ nb = raw.shape[0]
58
+ out = np.empty((nb, QK_K), dtype=np.float32)
59
+ d = raw[:, 208:210].copy().view(np.float16).astype(np.float32)[:, 0]
60
+ ql = raw[:, 0:128].reshape(nb, 4, 32).astype(np.int32) # 4 x 32 low/high nibbles
61
+ qh = raw[:, 128:192].reshape(nb, 2, 32).astype(np.int32)
62
+ sc = raw[:, 192:208].copy().view(np.int8).astype(np.int32)
63
+ q = np.empty((nb, QK_K), dtype=np.int32)
64
+ sidx = np.empty(QK_K, dtype=np.int32)
65
+ for g in range(2):
66
+ for l in range(32):
67
+ is_ = l // 16
68
+ o = g * 128
69
+ a = ql[:, 2 * g, l] # ql[g*64 + l]
70
+ b_ = ql[:, 2 * g + 1, l] # ql[g*64 + l + 32]
71
+ h = qh[:, g, l]
72
+ q[:, o + l] = ((a & 0xF) | ((h & 3) << 4)) - 32
73
+ q[:, o + l + 32] = ((b_ & 0xF) | (((h >> 2) & 3) << 4)) - 32
74
+ q[:, o + l + 64] = ((a >> 4) | (((h >> 4) & 3) << 4)) - 32
75
+ q[:, o + l + 96] = ((b_ >> 4) | (((h >> 6) & 3) << 4)) - 32
76
+ sidx[o + l] = g * 8 + is_
77
+ sidx[o + l + 32] = g * 8 + is_ + 2
78
+ sidx[o + l + 64] = g * 8 + is_ + 4
79
+ sidx[o + l + 96] = g * 8 + is_ + 6
80
+ out[:] = d[:, None] * sc[:, sidx] * q
81
+ return out
82
+
83
+
84
+ def dequant_mat(raw, n_rows, ne0, tt):
85
+ nrow = vl.row_bytes(ne0, tt)
86
+ nblk = ne0 // QK_K
87
+ arr = np.frombuffer(raw[:n_rows * nrow], dtype=np.uint8).reshape(n_rows, nblk, nrow // nblk)
88
+ if tt == 12:
89
+ return dequant_q4_K_vec(arr, ne0).reshape(n_rows, ne0)
90
+ if tt == 14:
91
+ return dequant_q6_K_vec(arr, ne0).reshape(n_rows, ne0)
92
+ raise ValueError(tt)
93
+
94
+
95
+ def main():
96
+ dump = sys.argv[1]
97
+ il = int(sys.argv[2])
98
+ kv, ts = vl.parse_headers()
99
+ n_embd = int(kv['bailingmoe3.embedding_length'])
100
+ n_ff = int(kv['bailingmoe3.expert_feed_forward_length'])
101
+ n_expert = int(kv['bailingmoe3.expert_count'])
102
+
103
+ raw = open(dump, 'rb').read()
104
+ meta = struct.unpack('<6i', raw[:24])
105
+ n_tok, n_used = meta[1], meta[2]
106
+ off = 24
107
+ cur = np.frombuffer(raw, dtype=np.float32, count=n_tok * n_embd, offset=off).reshape(n_tok, n_embd)
108
+ off += 4 * n_tok * n_embd
109
+ ids = np.frombuffer(raw, dtype=np.int32, count=n_tok * n_used, offset=off).reshape(n_tok, n_used)
110
+ off += 4 * n_tok * n_used
111
+ wts = np.frombuffer(raw, dtype=np.float32, count=n_tok * n_used, offset=off).reshape(n_tok, n_used)
112
+ off += 4 * n_tok * n_used
113
+ got = np.frombuffer(raw, dtype=np.float32, count=n_tok * n_embd, offset=off).reshape(n_tok, n_embd)
114
+ print(f'layer {il}: n_tok={n_tok} n_expert_used={n_used} n_expert={n_expert} '
115
+ f'n_embd={n_embd} n_ff={n_ff}')
116
+
117
+ def seg(nm, ie):
118
+ si, off_t, dims, tt = ts[f'blk.{il}.{nm}']
119
+ nrow = vl.row_bytes(dims[0], tt)
120
+ nb = nrow * dims[1]
121
+ with open(vl.SHARDS[si], 'rb') as f:
122
+ f.seek(off_t + ie * nb)
123
+ return f.read(nb), nrow, tt, dims[0]
124
+
125
+ ref = np.zeros((n_tok, n_embd), dtype=np.float32)
126
+ experts = sorted(set(int(e) for e in ids.reshape(-1)))
127
+ print(f'distinct experts: {len(experts)}')
128
+ for ie in experts:
129
+ g_raw, g_nrow, g_t, g_ne0 = seg('ffn_gate_exps.weight', ie)
130
+ u_raw, _, u_t, u_ne0 = seg('ffn_up_exps.weight', ie)
131
+ d_raw, _, d_t, d_ne0 = seg('ffn_down_exps.weight', ie)
132
+ Wg = dequant_mat(g_raw, n_ff, g_ne0, g_t)
133
+ Wu = dequant_mat(u_raw, n_ff, u_ne0, u_t)
134
+ Wd = dequant_mat(d_raw, n_embd, d_ne0, d_t)
135
+ rows = [t for t in range(n_tok) if ie in ids[t]]
136
+ x = cur[rows].T # [n_embd, T]
137
+ w = np.array([wts[t][list(ids[t]).index(ie)] for t in rows], dtype=np.float32)
138
+ h = ((Wg @ x) / (1.0 + np.exp(-(Wg @ x)))) * (Wu @ x)
139
+ y = Wd @ h # [n_embd, T]
140
+ ref[rows] += (y * w).T
141
+ del Wg, Wu, Wd
142
+
143
+ denom = max(float(np.abs(ref).max()), 1e-9)
144
+ diff = np.abs(got - ref)
145
+ corr = float(np.corrcoef(got.reshape(-1), ref.reshape(-1))[0, 1])
146
+ print(f'max|got-ref| = {diff.max():.6g} max|ref| = {denom:.6g} '
147
+ f'max_rel = {diff.max() / denom:.6g} corr = {corr:.8f}')
148
+ ok = corr > 0.99 and diff.max() / denom < 0.05
149
+ print('PASS' if ok else 'FAIL')
150
+ return 0 if ok else 1
151
+
152
+
153
+ if __name__ == '__main__':
154
+ sys.exit(main())
sdq-ling/watch_io.py ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """Run a command while sampling how the model file is actually read.
3
+
4
+ Reports, per sample and at the end:
5
+ read_bytes /proc/<pid>/io bytes actually fetched from the block device
6
+ file_resident mincore(model file) pages of the GGUF sitting in the page cache
7
+ RssAnon/RssFile process RSS split (arena+weights vs mapped file)
8
+ Cached /proc/meminfo system-wide page cache
9
+
10
+ The point: a buffered read would show up as file_resident growing; O_DIRECT reads
11
+ never touch the cache, so read_bytes grows while file_resident stays flat.
12
+
13
+ usage: watch_io.py <model.gguf>[,<model2.gguf>,...] <cmd> [args...]
14
+ """
15
+ import os
16
+ import re
17
+ import subprocess
18
+ import sys
19
+ import time
20
+
21
+ PAGE = os.sysconf('SC_PAGE_SIZE')
22
+ CHR = 1024 ** 3
23
+
24
+
25
+ def meminfo(key):
26
+ for line in open('/proc/meminfo'):
27
+ if line.startswith(key):
28
+ return int(line.split()[1]) * 1024
29
+ return 0
30
+
31
+
32
+ def proc_io(pid):
33
+ out = {}
34
+ try:
35
+ for line in open(f'/proc/{pid}/io'):
36
+ k, _, v = line.partition(':')
37
+ out[k] = int(v)
38
+ except OSError:
39
+ return None
40
+ return out
41
+
42
+
43
+ def proc_status(pid):
44
+ out = {}
45
+ try:
46
+ for line in open(f'/proc/{pid}/status'):
47
+ for k in ('RssAnon', 'RssFile', 'VmHWM'):
48
+ if line.startswith(k):
49
+ out[k] = int(line.split()[1]) * 1024
50
+ except OSError:
51
+ pass
52
+ return out
53
+
54
+
55
+ def file_resident(paths):
56
+ """resident bytes of the given files via mincore (helper binary)"""
57
+ r = subprocess.run(['/tmp/pagecache_resident'] + paths, capture_output=True, text=True)
58
+ m = re.search(r'TOTAL\s+resident\s+([\d.]+) GiB', r.stdout)
59
+ if m:
60
+ return float(m.group(1)) * CHR
61
+ m = re.search(r'resident\s+([\d.]+) GiB', r.stdout)
62
+ return float(m.group(1)) * CHR if m else -1
63
+
64
+
65
+ def main():
66
+ models = sys.argv[1].split(',')
67
+ cmd = sys.argv[2:]
68
+ subprocess.run(['sync'])
69
+ print(f'[watch] evicting {len(models)} shard(s) from the page cache', flush=True)
70
+ subprocess.run(['/tmp/pagecache_resident', '--evict'] + models)
71
+
72
+ base_cached = meminfo('Cached:')
73
+ base_read = None
74
+ t0 = time.time()
75
+ # NOTE: never use pipes here - sdq-chat prints several hundred KB of
76
+ # tokenizer warnings and would block on a full 64 KB pipe.
77
+ out_f = open('/tmp/watch_stdout.log', 'w')
78
+ err_f = open('/tmp/watch_stderr.log', 'w')
79
+ proc = subprocess.Popen(cmd, stdout=out_f, stderr=err_f)
80
+ pid = proc.pid
81
+
82
+ peak_res = 0.0
83
+ peak_rssanon = peak_rssfile = 0
84
+ peak_cached = base_cached
85
+ samples = 0
86
+ while proc.poll() is None:
87
+ io_ = proc_io(pid)
88
+ st = proc_status(pid)
89
+ if io_ is None:
90
+ break
91
+ if base_read is None:
92
+ base_read = io_['read_bytes']
93
+ res = file_resident(models)
94
+ cached = meminfo('Cached:')
95
+ peak_res = max(peak_res, res)
96
+ peak_cached = max(peak_cached, cached)
97
+ peak_rssanon = max(peak_rssanon, st.get('RssAnon', 0))
98
+ peak_rssfile = max(peak_rssfile, st.get('RssFile', 0))
99
+ samples += 1
100
+ print(f'[watch] t={time.time()-t0:6.1f}s read_bytes={(io_["read_bytes"]-base_read)/CHR:7.3f} GiB '
101
+ f' rchar={(io_["rchar"])/CHR:8.3f} GiB model_file_resident={res/CHR:6.3f} GiB '
102
+ f'RssAnon={st.get("RssAnon",0)/CHR:6.3f} RssFile={st.get("RssFile",0)/CHR:6.3f} '
103
+ f'Cached_delta={(cached-base_cached)/CHR:+7.3f} GiB', flush=True)
104
+ time.sleep(float(os.environ.get("WATCH_INTERVAL", "2.0")))
105
+
106
+ proc.wait()
107
+ out_f.close()
108
+ err_f.close()
109
+ print(f'\n[watch] === summary over {samples} samples, {time.time()-t0:.1f}s ===')
110
+ print(f'[watch] peak model_file_resident in page cache : {peak_res/CHR:.3f} GiB')
111
+ print(f'[watch] peak process RssAnon : {peak_rssanon/CHR:.3f} GiB')
112
+ print(f'[watch] peak process RssFile : {peak_rssfile/CHR:.3f} GiB')
113
+ print(f'[watch] system Cached delta (peak) : {(peak_cached-base_cached)/CHR:+.3f} GiB')
114
+ print(f'[watch] final model_file_resident : {file_resident(models)/CHR:.3f} GiB')
115
+ return 0
116
+
117
+
118
+ if __name__ == '__main__':
119
+ sys.exit(main())