dvlite(Ascend350)SOC 适配:为 33 个算子启用 ascend350 计算单元支持。
Ascend350 与 Ascend950 同为 DAV_3510 架构(regbase),共用 arch35 tiling 与 kernel 源码, 本次按仓内既有 ascend350 适配模式批量扩展:
AICore().AddConfig("ascend350", config)
GetSocVersion()
GetCurNpuArch()
DAV_2201 || IsRegbase(curArch)
DAV_1001 || DAV_2002 || DAV_3002
!= DAV_2201 && !IsRegbase(curArch)
涉及算子(33): FastGeluV2、HardSigmoid、SoftShrinkGrad、NPUClearFloatStatus、IndexCheck、ScatterMin、 ForeachAbs/AddcdivList/Asin/Exp/MaximumScalar/Neg/Sign/SubScalarInplace/Tanh(9)、 SigmoidCrossEntropyWithLogitsGrad、SigmoidFocalLoss、AdaLayerNormGrad、 BN3DTrainingReduceGrad、BNTrainingUpdateGrad、INTrainingUpdateV2、L2NormalizeGrad、 AdamApplyOneAssign、ApplyAdamV2、ApplyAdamWQuant、ApplyProximalGradientDescent、 FusedMulApplyKerasMomentum、InplaceApplyFtrl、LambUpdateWithLrV2、SGD、 SparseApplyAdagradV2、Dequantize、FakeQuantWithMinMaxArgsGradient
/assign
dvlite(Ascend350)SOC 适配:为 33 个算子启用 ascend350 计算单元支持。
Ascend350 与 Ascend950 同为 DAV_3510 架构(regbase),共用 arch35 tiling 与 kernel 源码,
本次按仓内既有 ascend350 适配模式批量扩展:
AICore().AddConfig("ascend350", config),复用 ascend950 同配置对象(含 apply_adam_w_quant 回归框架默认 opFile 名,与 kernel
入口 apply_adam_w_quant 一致)。
COMPUTE_UNIT(S) 增加 "ascend350",TILING_DIR 复用 arch35。
逐字段一致)及 2 个 ascend350/_simplified_key.ini(adam_apply_one_assign、
fake_quant_with_min_max_args_gradient)。
GetSocVersion()硬编码 switch/if 统一改为GetCurNpuArch()判断,消除新增 SOC 的遗漏点:DAV_2201 || IsRegbase(curArch)DAV_1001 || DAV_2002 || DAV_3002→ 910 dtype 列表DAV_2201 || IsRegbase(curArch)选择 grad 计算路径!= DAV_2201 && !IsRegbase(curArch)决定是否跳过检查Tanh/Exp/Abs/Sign/Neg/MaximumScalar/LambUpdateWithLrV2/ScatterMin 11 个条目补充
ascend350 compute_units 及对应 compile_options。
涉及算子(33):
FastGeluV2、HardSigmoid、SoftShrinkGrad、NPUClearFloatStatus、IndexCheck、ScatterMin、
ForeachAbs/AddcdivList/Asin/Exp/MaximumScalar/Neg/Sign/SubScalarInplace/Tanh(9)、
SigmoidCrossEntropyWithLogitsGrad、SigmoidFocalLoss、AdaLayerNormGrad、
BN3DTrainingReduceGrad、BNTrainingUpdateGrad、INTrainingUpdateV2、L2NormalizeGrad、
AdamApplyOneAssign、ApplyAdamV2、ApplyAdamWQuant、ApplyProximalGradientDescent、
FusedMulApplyKerasMomentum、InplaceApplyFtrl、LambUpdateWithLrV2、SGD、
SparseApplyAdagradV2、Dequantize、FakeQuantWithMinMaxArgsGradient