vconv

功能说明

vconv指令抽象。

y = vconv(x, dtype)， vconv表示对输入数据进行类型转换的向量计算。

目前支持转换类型包括：BF16->FP32, FP16->FP32, FP16->INT16, FP16->INT32, FP16->INT4, FP16->INT8, FP16->UINT8, FP32->BF16, FP32->FP16, FP32->INT32, FP32->INT64，INT4->FP16, INT64->FP32, INT8->FP16, UINT8->FP16。

接口原型

class vconv(x, y, dtype)

参数说明

使用示例

1
2
3
4
5

from mskpp import vconv, Tensor
ub_x, ub_y = Tensor("UB", "FP16"), Tensor("UB")
gm_x = Tensor("GM")
ub_x.load(gm_x)
out = vconv(ub_x, ub_y, "FP32")()