#29184 [Core] NGram GPU Implementation compatible with Async Scheduler
原始 PR · 作者 PatchouliTIS · 合并时间 2026-03-08 05:51
实现GPU加速的ngram推测解码,并与异步调度兼容,提升推理性能。
该PR值得精读,重点关注GPU kernel的设计(如torch.compile优化和向量化操作)、async scheduling集成中的性能权衡(如内存与速度平衡),以及review中讨论的代码重构决策(如逻辑迁移以减少核心文件影响)。
参与讨论