Skip to content

[1/5] refactor(attention): define execution contracts and device workspace - #1717

Draft
sufubao wants to merge 1 commit into
mainfrom
refactor/attention-01-contracts
Draft

sufubao wants to merge 1 commit into
mainfrom
refactor/attention-01-contracts

Conversation

@sufubao

@sufubao sufubao commented Nov 12, 2026 •

Copy link
Copy Markdown
Collaborator

新增 LayerSpec、LayerAttentionBinding、AttentionBatch、ContextCapacity、GraphBinding 和设备/工作区接口。保留旧模型运行入口;此步只引入后续 backend 所需的契约。

验证:本机 H200,包导入及 context/batch/out/device/workspace 契约16项通过。

由已关闭的 #1700 拆分,本PR为第1/5步。依赖顺序:

  1. [1/5] refactor(attention): define execution contracts and device workspace #1717 — API与设备/工作区
  2. [2/5] refactor(attention): relocate kernels without changing execution #1718 — kernel目录迁移
  3. [3/5] refactor(attention): add provider backends and explicit selection #1719 — provider backend与选择
  4. [4/5] refactor(attention): use varlen operators for vision and audio #1720 — 视觉/音频Varlen
  5. [5/5] refactor(attention): integrate model bindings and state lifecycle #1721 — 模型/State/Graph运行时

当前base为main,Files changed只展示本步增量。请按序合入;前序合入main后,将后续PR的base依次改为main并更新分支。所有步骤保持Draft。

每层已运行仓库固定版本Black/Flake8。仓库CI仅监听base=main,因此第2–5步当前以本地lint和实验记录为准,retarget后再检查CI。实验记录位于/data6/exp_logs/runs。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant