Skip to content
版 本

Mull(ISASI)

产 品 支 持 情 况

产 品

是 否 支 持

Ascend 950PR/Ascend 950DT

Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品

x

Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品

x

Atlas 200I/500 A2 推 理 产 品

x

Atlas 推 理 系 列 产 品AI Core

x

Atlas 推 理 系 列 产 品Vector Core

x

Atlas 训 练 系 列 产 品

x

功 能 说 明

对 前count个 输 入 数 据src0、src1按 元 素 相 乘 操 作,将 结 果 写 入dst0Local,溢 出 部 分 写 入dst1Local。计 算 过 程 如 下:

函 数 原 型

Text
template <typename T>
__aicore__ inline void Mull(const LocalTensor<T>& dst0, const LocalTensor<T>& dst1, const LocalTensor<T>& src0, const LocalTensor<T>& src1, const uint32_t count)

参 数 说 明

表 1 模 板 参 数 说 明

参 数 名

描 述

T

操 作 数 数 据 类 型。

Ascend 950PR/Ascend 950DT,支 持 的 数 据 类 型 为:int32_t、uint32_t。

表 2 参 数 说 明

参 数 名

输 入/输 出

描 述

dst0、dst1

输 出

目 的 操 作 数。

类 型 为LocalTensor,支 持 的TPosition为VECIN/VECCALC/VECOUT。

LocalTensor的 起 始 地 址 需 要32字 节 对 齐。

src0、src1

输 入

源 操 作 数。

类 型 为LocalTensor,支 持 的TPosition为VECIN/VECCALC/VECOUT。

LocalTensor的 起 始 地 址 需 要32字 节 对 齐。

两 个 源 操 作 数 的 数 据 类 型 需 要 与 目 的 操 作 数 保 持 一 致。

count

输 入

参 与 计 算 的 元 素 个 数。

返 回 值 说 明

约 束 说 明

调 用 示 例

更 多 样 例 可 参 考LINK

Text
AscendC::Mull(dst0Local, dst1Local, src0Local, src1Local, 512);

结 果 示 例 如 下:

Text
输 入 数 据src0Local:[-2039520159    95829766 -1522913829 ... ]
输 入 数 据src1Local:[-1791575569  -291976917 -1973597151 ... ]
输 出 数 据dst0Local:[-1560359025   212356098  1219226939 ... ]
输 出 数 据dst1Local:[850752575   -6514620  699800065 ... ]

免 责 声 明:本 站 内 容 由 asc-devkit 仓 master 分 支 自 动 编 译 生 成,属 于 持 续 开 发 版 本,可 能 存 在 缺 陷,仅 供 预 览 与 参 考。如 需 稳 定 及 商 用 资 料,请 查 阅 官 方 昇 腾 社 区