project6 2102146c01 refactor(moe): translate thrust mode.cu pipeline — unique_consecutive replaces manual boundary detect ...

thrust/examples/mode.cu entire design (80 lines):
  Complete GPU pipeline: sort → unique_count → reduce_by_key → max_element
  Key operations:
    1. thrust::sort — bring equal keys together (we already do: argsort)
    2. thrust::unique_count — precompute number of unique keys for allocation
    3. thrust::reduce_by_key(data, constant_iterator<1>) — count per key
    4. thrust::max_element — find the mode (highest count)
  Design principle: every step is a GPU primitive, no CPU round-trips.
  constant_iterator<1> trick: turns reduce_by_key into count_by_key.

Translation to MoE segment detection:
  Previous (4 GPU ops + CPU tensors):
    changes = cat([True, sorted[1:] != sorted[:-1]])
    seg_starts = changes.nonzero()
    seg_ends = cat([seg_starts[1:], tensor([len])])
    seg_eids = sorted[seg_starts]

  Now (1 fused GPU op):
    seg_eids, _, seg_counts = torch.unique_consecutive(sorted, return_counts=True)
    seg_ends = seg_counts.cumsum(0)
    seg_starts = cat([0, seg_ends[:-1]])

  unique_consecutive IS mode.cu's sort+reduce_by_key fused: it returns
  (unique_keys, inverse, counts) — exactly the data mode.cu builds from
  reduce_by_key(data, constant_iterator<1>, keys_out, counts_out).
  3 fewer GPU kernel launches per MoE forward.

CCCL source: thrust/examples/mode.cu
Maps to: qwen3_6_scripts/qwen3_5.py (_pure_pytorch_experts)

2026-08-07 09:17:23 +00:00

..

qwen3_5

[DEPLOY] Complete submission: baseline + all optimizations

2026-07-30 16:06:20 +00:00

qwen3_5_moe

[DEPLOY] Complete submission: baseline + all optimizations

2026-07-30 16:06:20 +00:00

reasoning

fix(critical): 3 fixes from sub508 diagnosis — n>1 crash guard + thinking format + content fallback

2026-08-07 07:55:04 +00:00

_custom_ops.py

fix(critical): CCCL policy_selector degradation for MoE — PyTorch fallback for topk_softmax

2026-08-07 08:56:50 +00:00

api_server.py

[DEPLOY] Complete submission: baseline + all optimizations

2026-07-30 16:06:20 +00:00

arg_utils.py

[ARCH] Eliminate AST patch scripts — full file replacements only

2026-08-05 08:24:43 +00:00

chat_utils.py

fix(critical): 3 bugs causing 0.0 score — OOM crash + thinking param + multimodal

2026-08-07 07:05:40 +00:00

cli_args.py

[DEPLOY] Complete submission: baseline + all optimizations

2026-07-30 16:06:20 +00:00

logits_processor.py

[ARCH] Eliminate AST patch scripts — full file replacements only

2026-08-05 08:24:43 +00:00

mamba_cache.py

[fix/correctness] mamba_cache: safe swap via clone, not in-place fancy indexing

2026-08-06 06:33:26 +00:00

model_runner.py

fix(engine): CCCL overflow_cast + checked_allocator patterns for NaN/OOM

2026-08-07 08:56:36 +00:00

paged_attention_v2_pytorch.py

[ENGINE] CCCL SmemResource pattern: pre-allocate staging buffers in V2 paged attention

2026-08-07 04:37:44 +00:00

paged_attn.py

perf: increase prefix attention tile budget 96MB→256MB

2026-08-07 04:44:18 +00:00

patch_ops.sh

fix(critical): patch_ops.sh cwd bug — all cp/deploy ./paths resolved against Dockerfile WORKDIR=/workspace/ instead of script dir

2026-08-07 06:20:02 +00:00

patch_transformers_qwen3_5.py

[DEPLOY] Complete submission: baseline + all optimizations

2026-07-30 16:06:20 +00:00

prefix_prefill.py

fix(critical): v2 kernel softmax normalization was commented out — outputs were wrong

2026-08-07 02:46:46 +00:00

protocol.py

fix: max_num_seqs=2 for n=2 support + remove protocol n clamp

2026-08-07 08:18:41 +00:00

qwen3_5.py

refactor(moe): translate thrust mode.cu pipeline — unique_consecutive replaces manual boundary detect

2026-08-07 09:17:23 +00:00

qwen3coder_tool_parser.py

fix(critical): disable thinking for tool_call requests — fixes d03_tool_call FAIL

2026-08-07 07:45:28 +00:00

registry.py

[ARCH] Eliminate 2 more patch scripts — registry.py + tool_parsers __init__.py

2026-08-05 08:36:52 +00:00

sampler.py

[BASE] qwen3_6_scripts/sampler.py: CCCL topk unsorted output optimization

2026-08-06 02:55:51 +00:00

scheduler.py

[DEPLOY] Complete submission: baseline + all optimizations

2026-07-30 16:06:20 +00:00

sequence.py

[DEPLOY] Complete submission: baseline + all optimizations

2026-07-30 16:06:20 +00:00

serving_chat.py

fix(thinking): default enable_thinking=True for t1a/t1c PASS

2026-08-07 09:01:57 +00:00

tool_parsers_init.py

[ARCH] Eliminate 2 more patch scripts — registry.py + tool_parsers __init__.py

2026-08-05 08:36:52 +00:00

verify_functional.py

fix(verify): remove duplicate TC-22~30 test definitions — CCCL test_then.cu audit

2026-08-07 06:36:12 +00:00

xformers.py

feat(xformers): replace Q-only tiling with Q+KV tiling + online softmax

2026-08-07 01:54:52 +00:00