tvm_tirx.tirx.script.gpu_dp4a

Contents

tvm_tirx.tirx.script.gpu_dp4a#

tvm_tirx.tirx.script.gpu_dp4a(vec1, vec2, acc=0, **kwargs)#

Dot product of two int8x4 vectors and add an optional accumulator

Parameters:

  • vec1 (int8x4) – The input vector.

  • vec2 (int8x4) – The input vector.

  • acc (int32) – The accumulator.

Returns:

call (Expr) – The call expression.

Contents