vmla

功能说明

vmla指令抽象。

z = x * y + z， x、y 按元素相乘，相乘的结果与z按元素相加，可以通过if_mix将输出的数据类型格式指定为FP32。

目前支持：

type = f16, f16 = f16 * f16 + f16。

type = f32, f32 = f32 * f32 + f32。

type = fmix, f32 = f16 * f16 + f32, 其中Xn,Xm向量使用64个元素的f16数据用于计算，源向量仅使用低4个block，4个高block被忽略。Xd是64个元素的包含8个block的f32数据，同时作为目标向量和第三个源向量。

接口原型

class vmla(x, y, z, if_mix=False)

参数说明

参数名	输入/输出	数据类型	说明
x	输入	tensor变量	输入x向量tensor，支持FP16、FP32。
y	输入	tensor变量	输入y向量tensor，支持FP16、FP32。
z	输出	tensor变量	输出向量tensor，支持FP16、FP32。
if_mix	输入	tensor变量	默认为False。若设置为True的，指定输出数据类型为FP32。

约束说明

vector指令输入输出数据的tensor均在“UB”空间中。

使用示例

from mskpp import vmla, Tensor
ub_x, ub_y, ub_z = Tensor("UB"), Tensor("UB"), Tensor("UB")
gm_x, gm_y = Tensor("GM"), Tensor("GM")
ub_x.load(gm_x)
ub_y.load(gm_y)
out = vmla(ub_x, ub_y, ub_z)()

父主题： msKPP对外接口使用说明