通用定义
功能说明
三个源操作数时指令的通用格式,三个源操作数分别为src, dst和标量scalar。注:这不是真正的指令。
函数原型
instruction (mask, dst, src, scalar, repeat_times, dst_rep_stride, src_rep_stride, )
参数说明
参数名称 |
输入/输出 |
含义 |
---|---|---|
instruction |
输入 |
指令名称,TIK DSL中为全小写字符串。 |
mask |
输入 |
请参考单目/通用定义/表1中mask参数描述。 |
dst |
输出 |
矢量目的操作数/矢量源操作数1,tensor起始element,支持数据精度见具体指令。 Tensor的scope为Unified Buffer。 |
src |
输入 |
矢量源操作数0,tensor起始element,支持数据精度见具体指令。 Tensor的scope为Unified Buffer。 |
scalar |
输入 |
标量源操作数,支持Scalar/立即数 |
repeat_times |
输入 |
重复迭代次数。 |
dst_rep_stride |
输入 |
相邻迭代间,矢量目的操作数相同block地址步长。 |
src_rep_stride |
输入 |
相邻迭代间,矢量源操作数相同block地址步长。 |
注意事项
- repeat_times∈[0,255]。支持的数据类型为:Scalar(int16/int32/int64/uint16/uint32/uint64)、立即数(int)、Expr(int16/int32/int64/uint16/uint32/uint64),当repeat_times为立即数时,不支持0。
- dst_rep_stride/src_rep_stride,单位为32B。支持的数据类型:Scalar(int16/int32/int64/uint16/uint32/uint64)、立即数(int)、Expr(int16/int32/int64/uint16/uint32/uint64)。
- dst既是目的操作数,也是源操作数。
- 为了节省地址空间,开发者可以定义一个Tensor,供源操作数与目的操作数同时使用(即地址重叠),相关约束如下:
- 对于单次repeat(repeat_times=1),且源操作数与目的操作数之间要求100%完全重叠,不支持部分重叠。
- 对于多次repeat(repeat_times>1),若源操作数与目的操作数之间存在依赖,即第N次迭代的目的操作数是第N+1次的源操作数,这种情况是不支持地址重叠的。
- 操作数地址偏移对齐要求请见通用约束。
父主题: 标量三目