Commit Graph

  • 2c3cef1c87 feat: wire up paged decode CUDA kernel to Python extension main ViperEkura 2026-07-11 18:38:05 +0800
  • 89ece26c25 feat: paged decode attention with split-KV (scalar + MMA) ViperEkura 2026-07-11 18:13:53 +0800
  • 2c0b5d0b5e perf: enable MMA decode path for G=1 full attention ViperEkura 2026-07-11 11:45:13 +0800
  • a4ae7d17fb perf: increase decode split-K parallelism for short sequences ViperEkura 2026-07-11 11:26:42 +0800
  • 8a8550184f refactor: template AttentionParams, rename .cuh to .h ViperEkura 2026-07-11 11:03:14 +0800
  • b8b439b713 perf: fuse decode combine kernel to single-pass online-rescale reduction ViperEkura 2026-07-11 00:24:09 +0800
  • 41cd40363a perf: post-multiply attention scale in float instead of pre-scaling Q in bf16 ViperEkura 2026-07-11 00:12:12 +0800
  • d923ebe38d refactor: rename gqa_* to attn_*, split-KV for all decode paths ViperEkura 2026-07-10 23:35:14 +0800
  • 29b0423c4e refactor: deduplicate kernel code with shared MMA and entry-point helpers ViperEkura 2026-07-10 20:52:27 +0800
  • 88f8dca2c2 perf: enlarge prefill KV tile to BC=32 for D<=128 ViperEkura 2026-07-10 17:41:31 +0800
  • 9027fdc546 test: bench production MMA attention path with FLOP/s and bandwidth ViperEkura 2026-07-10 16:55:24 +0800
  • cbd140340d perf: load prefill Q fragments directly from global (drop sQ staging) ViperEkura 2026-07-10 12:27:45 +0800
  • 988e01314d perf: pipeline prefill MMA kernel (double-buffered K/V + packed stores) ViperEkura 2026-07-10 12:18:11 +0800
  • 7ba43a7c6f perf: add split-K (FlashDecoding) to decode MMA kernel ViperEkura 2026-07-10 11:42:21 +0800
  • dea59f7e1d fix: restore resident Qa to fix sQ overwrite bug ViperEkura 2026-07-10 00:49:58 +0800
  • 85dc771460 perf: reduce MMA kernel registers, switch to static smem ViperEkura 2026-07-10 00:39:47 +0800
  • 2c5629b81d docs: fix documentation errors across README and assets/docs ViperEkura 2026-07-09 10:09:53 +0800
  • 841a582b28 refactor: split mask builder by single/multi output ViperEkura 2026-07-08 21:18:02 +0800
  • c8567a6f65 fix: exclude embedding, lm_head, bias, and norm params from Muon optimizer, use AdamW ViperEkura 2026-07-08 19:41:16 +0800
  • 8035be9b1f fix: make MuonMix inherit from torch.optim.Optimizer ViperEkura 2026-07-08 17:03:05 +0800
  • e9b03f4fca perf: apply cp.async, XOR swizzle, pre-scaled Q to decode MMA kernel ViperEkura 2026-07-08 16:03:40 +0800
  • fd65b9bc23 feat: support HEAD_DIM=32 and split extension into loader/ops ViperEkura 2026-07-08 13:59:54 +0800
  • 9ebaea840f perf: cp.async K/V loads, shared sQ staging, causal skip, XOR swizzle ViperEkura 2026-07-08 12:30:32 +0800
  • 6adc221c10 refactor: extract shared MMA utils into gqa_mma_utils.cuh ViperEkura 2026-07-07 23:01:15 +0800
  • 9e63cb9ed0 feat: MMA head-packing decode kernel with scalar fallback dispatch ViperEkura 2026-07-07 22:56:02 +0800
  • 4225518cf3 perf: add fast-math and vectorization nvcc/cxx build flags ViperEkura 2026-07-07 22:28:32 +0800
  • c50adbaac0 feat : replace AdamW with MuonMix (Muon + AdamW) optimizer ViperEkura 2026-07-07 14:10:36 +0800
  • 536dbc0c9a fix: set tqdm postfix before update so first step shows metrics ViperEkura 2026-07-07 00:14:13 +0800
  • 4af7acd449 fix: support single .h5 file loading in load_h5 ViperEkura 2026-07-07 00:11:16 +0800
  • 53ed52b4b8 refactor: extension dispatch layer with CUDA/torch fallback ViperEkura 2026-07-06 21:07:16 +0800
  • f1cc7cedce feat: ldmatrix + smem padding for mma prefill kernel ViperEkura 2026-07-06 20:54:50 +0800
  • ddc4bd1cf6 feat: tensor-core mma prefill with build-time dispatch ViperEkura 2026-07-06 19:51:55 +0800
  • cc36530c73 perf: group-split register-blocking gqa_prefill kernel ViperEkura 2026-07-06 18:32:09 +0800
  • 11fa807cfc fix: correct prefill mask index, unify GQA kernel interface ViperEkura 2026-07-06 16:52:19 +0800
  • bcdd93e0eb feat: split kernel defs from bindings, add prefill tiled kernel and pure C tests ViperEkura 2026-07-06 16:14:34 +0800
  • 579b8c3129 fix: correct gqa_decode_attn reduction + add gqa_prefill_attn ViperEkura 2026-07-06 13:44:23 +0800
  • d7da51569f docs: update install instructions in EN/CN README ViperEkura 2026-07-06 12:25:17 +0800
  • e8e228d035 feat: add optional CUDA kernel system (csrc/) + fused GQA decode attention ViperEkura 2026-07-06 12:08:12 +0800
  • 2579658e15 chore : shields release badge from /release to /tag ViperEkura 2026-07-05 20:33:51 +0800
  • f0cd0134c6 fix : update benchmark for v1.3.8 cache API, add argparse and cache type switch ViperEkura 2026-07-05 20:27:47 +0800
  • abb96996f8 docs : sync 6 doc files to actual code ViperEkura 2026-07-05 19:34:30 +0800
  • bbe6ff2d8f release : v1.3.8 v1.3.8 ViperEkura 2026-07-05 19:04:56 +0800
  • db9b39b084 fix: resolve IFD token-set asymmetry and support single-token answers ViperEkura 2026-07-05 17:48:26 +0800
  • 849e1e00a3 refactor: clean up inference design patterns ViperEkura 2026-07-05 11:41:54 +0800
  • 5416c2e8fb perf: replace paged KV cache with contiguous ContiguousCache, decode all groups ViperEkura 2026-07-05 11:34:36 +0800
  • 599a51f4f7 fix: reliable test timeout, separate generate/test phases, dynamic pass@k ViperEkura 2026-07-05 08:47:07 +0800
  • 17d6eaa2f2 refactor: rewrite humaneval evaluation with functional pipeline design ViperEkura 2026-07-05 07:56:55 +0800
  • 2d908639e9 feat : add ROUGE evaluation script (manual impl, no deps) ViperEkura 2026-07-05 01:15:01 +0800
  • c7158418dd perf: add BFD bin-packing and custom attention mask to IFD batch scoring ViperEkura 2026-07-04 18:58:13 +0800
  • 4d3c9341c1 refactor: rewrite IFD evaluation with clean three-layer architecture ViperEkura 2026-07-04 18:26:07 +0800
  • 4e508afa2d fix : SFT pipeline position_ids default & doc boundary preservation ViperEkura 2026-07-04 15:59:11 +0800
  • 8999ca89b8 feat: add JSONL dataset store with on-the-fly tokenization ViperEkura 2026-07-04 15:42:33 +0800
  • 1adca39cd8 fix: handle long sequences and optimize IFD computation ViperEkura 2026-07-04 08:35:45 +0800
  • 204873fa2f fix: handle long sequences and optimize IFD computation ViperEkura 2026-07-04 08:21:53 +0800
  • a5c1de6b1b feat: add model_path temperature top_p top_k max_tokens system_prompt args to stream_chat ViperEkura 2026-07-04 07:02:06 +0800
  • 27524ad085 fix: reset sampler iter at epoch end so progress bar shows total after first epoch ViperEkura 2026-07-04 06:35:29 +0800
  • 27d1921d9c fix: scheduler division-by-zero, loss_mask bool ViperEkura 2026-07-03 22:03:36 +0800
  • 70c0e5de90 refactor: merge validation into MetricCallback, simplify progress bar to optimizer steps ViperEkura 2026-07-03 21:43:08 +0800
  • dfb151537b fix: ForwardRef._evaluate Python 3.12 compatibility ViperEkura 2026-07-03 18:41:19 +0800
  • 500c605fad fix: unify scheduler min_rate default to 0.01, clamp WSD warmup ViperEkura 2026-07-03 17:52:23 +0800
  • dc9faca3b1 fix: align docs with actual code (40+ inconsistencies) ViperEkura 2026-06-30 20:47:23 +0800
  • aabb0d83e9 refactor : replace iteration with consumed_samples ViperEkura 2026-06-30 18:42:33 +0800
  • 44579ea6dc refactor : metric 日志改为以 optimizer step 为单位,默认每步记录 ViperEkura 2026-06-30 15:12:31 +0800
  • 0f1fcb079f refactor : grad_norm 指标简化,clip_grad_norm 移至 executor ViperEkura 2026-06-30 14:59:43 +0800
  • 84d4769163 feat: SVD 有效秩/权重统计分析脚本 ViperEkura 2026-06-29 21:39:22 +0800
  • bf09a35c95 feat: optimizer 参数分组,bias/norm 不做 weight decay ViperEkura 2026-06-27 16:30:34 +0800
  • 6715461a36 chore : 升级 torch 2.11.0+cu128,移除自定义 Muon,修复 gloo device_id ViperEkura 2026-06-27 16:10:14 +0800
  • b4587c5d08 refactor : metric_logger 改用事件类型 (type=step/validation/epoch) ViperEkura 2026-06-25 17:18:20 +0800
  • 88ec63121d feat : GPT-2 residual scaling weight init ViperEkura 2026-06-25 15:04:27 +0800
  • 01d2da2893 feat : 训练支持 --schedule_type 及对应调度器参数 ViperEkura 2026-06-22 10:35:14 +0800
  • 25d4ea3f91 refactor : 压缩测试代码,消除重复 ViperEkura 2026-06-19 14:53:35 +0800
  • 39985840c7 refactor : neftune_alpha 在 Embedding 构造时传入,由模型配置链路负责 ViperEkura 2026-06-19 14:23:27 +0800
  • b1adc40cfb refactor : 将 config 对象直接传给 DecoderBlock,替代 16 个独立参数 ViperEkura 2026-06-19 14:15:20 +0800
  • 7348bac6ab fix: 规范 generate.py 命令行接口 ViperEkura 2026-06-19 14:02:22 +0800
  • 8ab7564d02 docs: 重构 README 结构,全文档添加目录导航 ViperEkura 2026-06-19 13:52:32 +0800
  • d096b6e29e docs: 修复文档中过时的字段、签名和缺失的类 ViperEkura 2026-06-18 18:49:46 +0800
  • d88a41f8f1 fix: 修复预处理流水线 4 个致命问题 ViperEkura 2026-06-18 17:38:01 +0800
  • 376e9eba80 feat: IFEval 使用 chat template 格式化 prompt,添加 model.eval() ViperEkura 2026-06-18 16:45:16 +0800
  • a62c2e11a2 feat: IFD 默认使用 chat template,支持裸文本模式 ViperEkura 2026-06-18 16:34:27 +0800
  • a4e5a8c81c feat: 新增 WSD 学习率调度器 ViperEkura 2026-06-18 15:55:15 +0800
  • 3e234c46f6 fix: 使用 threading.Event 替代裸 bool,补全公共 API ViperEkura 2026-06-18 15:38:13 +0800
  • 7a04b1f8ce docs: replace shields.io endpoint badges with github/ direct badges ViperEkura 2026-06-18 15:09:51 +0800
  • a30e3d5114 fix: 修复 shields.io GitHub badge 因 token 耗尽而无法显示 ViperEkura 2026-06-16 22:21:58 +0800
  • 1818d06576 feat: 新增 IFD 数据质量评分工具, 移动 ppl 至 eval ViperEkura 2026-06-16 22:03:11 +0800
  • 4e8d1ee24e feat: 新增 IFEval 指令遵循评测 ViperEkura 2026-06-16 21:57:23 +0800
  • fec376b0dd fix : 修复策略相关文件的类型注解与抽象方法体 ViperEkura 2026-06-14 16:20:10 +0800
  • a2512f8a5a fix : resume_dir 无权重文件时不强制加载,支持仅配置训练 ViperEkura 2026-06-13 15:40:14 +0800
  • 457e16ea3c fix : val_loss 默认改为 None,日志跳过空值;val_dataloader 补 Optional 注解 ViperEkura 2026-06-13 14:24:13 +0800
  • daf627a6de fix : _save_log 前确保日志目录存在,防止跨进程反序列化后目录丢失 ViperEkura 2026-06-12 15:39:54 +0800
  • 445378667f feat : NEFTune 噪声注入 + label_smoothing 默认值修正 ViperEkura 2026-06-11 15:31:08 +0800
  • 6ae1828449 refactor : 清理工厂和配置系统中的死代码与冗余抽象 ViperEkura 2026-06-07 11:39:50 +0800
  • e7b18b7c03 refactor : BaseFactory 基类类型自动推导 + 移除冗余代码 ViperEkura 2026-06-06 21:23:41 +0800
  • 9e31d4ef2b feat : BaseToolParser.feed 增加可选 token_ids 参数 ViperEkura 2026-06-06 11:19:02 +0800
  • 52aa4d01d5 feat : 推理层增加 vLLM 风格工具调用解析 ViperEkura 2026-06-06 08:52:30 +0800
  • 986be957ec refactor : on_batch_begin 移入 accumulate 上下文 ViperEkura 2026-06-06 01:19:21 +0800
  • cf9c60841b docs : 按代码反向修正所有文档错误 ViperEkura 2026-06-06 01:06:15 +0800
  • 31bc7f5c2a refactor : pipeline 策略化拆分,消除 _flush if/else ViperEkura 2026-06-06 00:45:33 +0800
  • 3057741de9 refactor : 合并 data config docstring 并实现 BFD 打包策略 ViperEkura 2026-06-05 17:41:51 +0800
  • acd1103bd0 fix : 使用 bool 注意力掩码并支持打包 SFT 文档边界阻断 ViperEkura 2026-06-05 17:02:00 +0800
  • dc7d2cfbca refactor : FastAPI 懒加载单例,消除模块级副作用 ViperEkura 2026-06-04 15:52:01 +0800