vmla

功能说明

vmla指令抽象。

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

目前支持：

type = f16，f16 = f16 * f16 + f16。

type = f32，f32 = f32 * f32 + f32。

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

接口原型

1

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

参数说明

约束说明

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

使用示例

1
2
3
4
5
6

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)()