AstrAI

Files

T

ViperEkura a3c8296135 fix: page cache 分配失败越界崩溃 + 长度超限终止

- astrai/inference/scheduler.py: add_task 增加 max_seq_len 检查，超限时直接发 STOP 信号终止
- astrai/inference/scheduler.py: _maybe_alloc_page 返回 bool，alloc 失败时标记 ABORTED + 发 STOP
- astrai/inference/scheduler.py: _execute_decode 过滤分配失败任务，避免 page_table 越界
- astrai/inference/scheduler.py: _remove_finished_tasks 清理 ABORTED 任务并释放 pages
- astrai/inference/scheduler.py: _execute_prefill input_mask 改为覆盖全部 prompt_len
- astrai/model/transformer.py: seq_mask is None 分支补全 start_pos + seq_len 列

2026-05-10 20:14:38 +08:00

__init__.py

fix: 修复 CLI 参数缺失/重复、device_ids 越界、generate 参数名不一致、scheduler 时序、非流式截断等 bug

2026-05-09 14:36:42 +08:00

cache.py

perf: PagedCache 持久前缀缓存 + LRU 逐出

2026-05-10 18:05:11 +08:00

engine.py

feat: _generate_streaming 支持 batch 模式

2026-05-10 17:42:20 +08:00

sampling.py

refactor: 分页 KV cache 替换固定 slot，删除 PrefixCache 及相关死代码