Add torch fallbacks for KDA and GDN layers.

未关闭
#406 0 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看

还没有人认领这个 Issue。

评估

难度
4/5
预计耗时
3-5 天
新手友好度
30/100
Issue 类型
功能
描述清晰度
基本清楚
活跃度
停滞
技术栈
python, pytorch

调研方向

The payload names only tests/test_attention.py as the place the KDA and GDN unit tests need to match the existing attention structure, and points to #395 for context, which is not included. Start by reading that test file and the existing attention test pattern, then locate the KDA and GDN layer implementations. Done means each layer has a torch fallback path and its tests follow the same structure as the other attention tests.

由索引模型根据 Issue 内容生成。

描述

See comments in #395.

  • add torch fallbacks
  • adapt kda and gdn unit tests to adhere to attention structure (Fast-LLM/tests/test_attention.py)

Postponed because this is not on the critical path.

主要语言
Python
星标
337
派生
52
PR 合并指标
30 天内没有已合并 PR

环境准备

从这里开始

  1. 先读完整个 Issue,再读项目的贡献指南。
  2. 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
  3. Fork 仓库,在一个分支上完成修改。
  4. 提交 Pull Request,并在描述里引用这个 Issue 编号。

ServiceNow/Fast-LLM 的其他 Issue

查看 ServiceNow/Fast-LLM 的全部 Issue

相似的 Issue

更多 Python Issue

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。