已关闭
22个optim/loss/conv/quant/activation算子补齐ascend350(DVLite)平台适配 #6623
柴海伦创建于  8 天前关闭于  6 天前
柴海伦
柴海伦成员
8 天前 创建

需求描述

为 optim、loss、conv、quant、activation 下的 22 个算子补齐 DVLite(ascend350)平台配置,以已有 ascend950 实现为基线,复用 arch35 构建路径。

涉及算子

  • optim:AdamApplyOne、AdamApplyOneWithDecay、AdamApplyOneWithDecayAssign、ApplyCamePart1–4、ApplyFtrl、ApplyMomentum。
  • loss:KlDivLossGrad、MseLossGrad、SmoothL1LossGradV2、CrossEntropyLoss、CrossEntropyLossGrad、CTCLossV2、CTCLossV2Grad。
  • conv:DeformableOffsets、DeformableOffsetsGrad。
  • quant:DequantSwigluQuant、SwigluMxQuant、SwigluMxQuantWithDualAxis。
  • activation:ClippedSwiglu。

实现范围

补齐平台注册、host/tiling/kernel 构建配置、350 binary JSON 和公共 ascendc_config.json 的对应条目。保留已有算法、tiling 策略及非350平台功能行为。

当前进展与验收

对应 PR: https://gitcode.com/cann/ops-nn/merge_requests/11894

当前缺少配套 CANN 工具链、测试依赖和 DVLite 设备,编译、UT、真机验证未执行;本 PR 尚未完成 DVLite 运行验收。具体待验证项和范围外问题见 PR 描述。

likedislike
yuning_chenyuning_chen成员
8 天前 将 chaihailun 设为负责人
CANN-robotCANN-robot成员
6 天前 关闭了 issue
CANN-robotCANN-robot成员
6 天前 添加了label:resolved