已关闭
NPUAllocFloatStatus算子支持Ascend950 #2677
wkkk0528创建于  20 天前关闭于  16 天前
wkkk0528
20 天前 创建

描述

NPUAllocFloatStatus算子支持Ascend950

算子功能

NPUAllocFloatStatus 用于分配并初始化浮点异常状态张量。算子接收一个可选的输入张量(仅用于形状推导),输出一个固定长度为8的float32张量,所有元素初始化为零值。该算子是 NPUFloatStatus 类算子的基础组件,通常与 NPUClearFloatStatus 配合使用,用于管理 NPU 浮点运算的异常状态。

实现方式

  • 架构: Ascend950 (arch35) SIMT kernel
  • Kernel 实现: 单核 Scalar 模式,使用 SetValue 逐元素将零值写入 GM(绕过 UB,规避 DYN/BIN 首字节污染),完成后执行 DataCacheCleanAndInvalid 保证 GM 数据一致性
  • Tiling: arch35 专用 Tiling,固定输出 8 个 float32 元素,单核无切分
  • Shape/Dtype 推导: 输出 shape 固定为 [8],dtype 为 float32
  • aclnn 接口: 提供 aclnnNPUAllocFloatStatus 两段式 API
  • REG_OP: NPUAllocFloatStatus 原型注册,支持图模式调用
likedislike
Ssongkai111成员
20 天前 将 wkkk0528 设为负责人
CANN-robotCANN-robot成员
16 天前 关闭了 issue
CANN-robotCANN-robot成员
16 天前 添加了label:resolved