Skip to content

feat: enable Ascend950 causal convolution QKV. - #30

Open
wuyifan2017 wants to merge 1 commit into
xLLM-AI:mainfrom
wuyifan2017:codex/a5-causal-conv1d-qkv
Open

feat: enable Ascend950 causal convolution QKV.#30
wuyifan2017 wants to merge 1 commit into
xLLM-AI:mainfrom
wuyifan2017:codex/a5-causal-conv1d-qkv

Conversation

@wuyifan2017

Copy link
Copy Markdown

变更说明

  • causal_conv1d_qkv 加入 Ascend950 ACLNN 构建清单。
  • CausalConv1dQkv 注册 ascend950 Host 配置。
  • 增加 Qwen3.5-2B prefill 模型形状测试,覆盖 FP16 和 BF16。
  • 保留现有 Ascend 910B、910_93 及 A2/A3 路径不变。

验证结果

  • Ascend950 完整算子包构建:PASS。
  • 原有 CausalConv1dQkv 测试矩阵:10 passed。
  • Qwen3.5-2B 模型形状测试:2 passed,覆盖 FP16/BF16。
  • 精度测试:8 个用例通过,覆盖 21、128、129 和 2595 token。
  • git diff --check、Shell 语法检查和 Python AST 检查:PASS。

兼容性

  • 新增逻辑仅进入 Ascend950 构建分支和注册配置。
  • Ascend 910B、910_93 的既有注册保持不变。

Register and package causal_conv1d_qkv for Ascend950. Cover the Qwen3.5-2B prefill shape in FP16 and BF16 tests.
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