GB10 TP1 (Qwen3.8 Flash Next): alpha package at `--max-concurrency 16` gives +47% KV and faster C1/C4; one-box DFlash2+A16 export does not beat MTP

Đang mở
  • Có gì không ổn?
#263 1 bình luận 0 reaction 0 người được giao Xem trên GitHub

Maintainer thường phản hồi trong vòng 1 ngày

Chưa có ai nhận issue này.

Đánh giá

Độ khó
3/5
Thời gian dự kiến
1-2 ngày
Mức phù hợp với người mới
48/100
Loại issue
Tài liệu
Độ rõ ràng
Khá rõ ràng
Mức độ hoạt động
Sôi nổi
Công nghệ
github, rust
Lĩnh vực
infrastructure, performance

Hướng nghiên cứu

Start by reading the GB10 TP1 measurements and the alpha package README mentioned in the issue. Check how the README documents --max-concurrency and --mtp-draft-tokens, then decide whether to add the c16 recommendation and its benchmark context. Done when the guidance accurately reflects the reported results and distinguishes the one-box DFlash2 comparison.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Mô tả

Measurements from one DGX Spark (GB10, driver 615.71.09, kernel 7.0.0-1019-nvidia), in case they help with the GB10 TP1 defaults. Each package was benched on the same box, back to back, with no other traffic. Memory settings were identical: --memory-limit=-14GiB --embedding-table-storage disk --embedding-table-cache-size 2GiB, with page cache dropped before every launch.

Bench: llm_decode_bench.py, 8k context, 30 s per concurrency, temperature 1, 512 max tokens. Coding peak is the median of 10 runs. Single trials; run-to-run noise on this box is about ±5%.

Package Launch KV tokens C1 C4 C16 Coding peak (median / max)
local-inference-lab/Qwen3.8-Flash-Next-FLARE@gb10 (alpha, c32, MTP) default 466,944 41.7 90.9 178.0 66.9 / 82.5
same --max-concurrency 16 688,128 44.6 103.3 170.9 64.2 / 78.2
own TP1 export, DFlash2 + A16, c16 --mtp-draft-tokens 7 --mtp-adaptive-verification --mtp-adaptive-verification-alpha 0.5 618,496 41.3 85.5 169.2 71.9 / 82.0

1. Lowering concurrency at launch on the alpha package. --max-concurrency 16 on the c32 alpha package raised the KV pool 47%. It was also faster at C1 (+7%) and C4 (+14%), and within noise at C16 and coding peak. Quality on it was unchanged: Hotel Lights x8 at xhigh 8/8 (temperature 0.7, top-p 0.95, seed 1000), tool calls 5/5, vision 8/8. For a single 128 GB GB10, c16 looks like the better default, or at least worth a line in the package README next to "--max-concurrency and --mtp-draft-tokens can be reduced at launch".

2. DFlash2 + A16 on one box. We exported the same recipe we use on TP2 (two Sparks, one RoCE cable):

  • base checkpoint nvidia/Qwen3.8-Flash-Next-FP4-Local;
  • export flags --speculative-method dflash2 --mtp-draft-tokens 7 --online-mxfp8 --draft-head-quantization nvfp4 --force-moe-decode-a16 --max-concurrency 16;
  • main 972e28e4 plus host-system-reserve and dflash2-tp-embedding-contiguous.

On TP2, switching to this recipe from MTP5 took the coding-peak median from 81 to 142 tok/s. On TP1 it gained 12% coding peak over the alpha at c16, lost 17% at C4, and had 10% less KV. Fixed allocation was 76.78 GiB versus the alpha's 75.72 GiB, with 109 graphs. Our reading is that most of DFlash2's TP2 win comes from fewer cross-host verification rounds, so on one box MTP remains the better choice.

3. One-box export timing on GB10. The export took 93 minutes in total. Kernel tuning took 42 minutes from a cold cache. "Compile model forwards" took 36 minutes for 79 forwards, against 476 s for 56 forwards on the TP2 export: with about 89 GB resident in the exporting process, the automatic --compile-workers budget allowed only 2 compiler processes. Warmup, capture and packing were uneventful.

Happy to rerun anything with different flags.

Ngôn ngữ chính
Rust
Star
800
Fork
106
Merge trung bình
1 ngày 3 giờ
Pull request đã merge (30 ngày)
100

Chuẩn bị môi trường

Bắt đầu từ đâu

  1. Đọc hết issue, rồi đọc hướng dẫn đóng góp của dự án.
  2. Bình luận trên issue rằng bạn sẽ nhận — tránh hai người làm cùng một việc.
  3. Fork repository và làm thay đổi trên một nhánh.
  4. Mở pull request có tham chiếu số hiệu của issue.

Issue khác của NVIDIA/hyperflare-inference

Tất cả issue của NVIDIA/hyperflare-inference

Issue tương tự

Thêm issue về Rust

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.