asc_set_ndim_pad_value

产 品 支 持 情 况

产 品	是 否 支 持
Ascend 950PR/Ascend 950DT	√

功 能 说 明

当asc_ndim_copy_gm2ub接 口 的padding_mode参 数 为true时，该 值 将 设 置 为Padding的 填 充 固 定 值。fp8_e8m0_t，fp8_e5m2_t，fp8_e4m3fn_t类 型 的 数 据 需 转 换 成int8_t类 型 后 再 调 用 本 接 口。

函 数 原 型

C++

__aicore__ inline void asc_set_ndim_pad_value(int8_t pad_value)
__aicore__ inline void asc_set_ndim_pad_value(uint8_t pad_value)
__aicore__ inline void asc_set_ndim_pad_value(int16_t pad_value)
__aicore__ inline void asc_set_ndim_pad_value(uint16_t pad_value)
__aicore__ inline void asc_set_ndim_pad_value(half pad_value)
__aicore__ inline void asc_set_ndim_pad_value(bfloat16_t pad_value)
__aicore__ inline void asc_set_ndim_pad_value(int32_t pad_value)
__aicore__ inline void asc_set_ndim_pad_value(uint32_t pad_value)
__aicore__ inline void asc_set_ndim_pad_value(float pad_value)

参 数 说 明

参 数 名	输 入/输 出	描 述
pad_value	输 入	用 于 设 置asc_ndim_copy_gm2ub接 口Padding的 填 充 固 定 值。数 据 类 型 需 要 与asc_ndim_copy_gm2ub接 口 中dst和src的 数 据 类 型 保 持 一 致。

返 回 值 说 明

无

流 水 类 型

PIPE_S

约 束 说 明

需 配 合asc_ndim_copy_gm2ub使 用。

调 用 示 例

C++

half pad_value = 2.0;
asc_set_ndim_pad_value(pad_value);