Skip to content

Commit

Permalink
Fix format issues
Browse files Browse the repository at this point in the history
  • Loading branch information
Jiawei-Shao committed Jun 28, 2024
1 parent 813db99 commit d59ede3
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 0 deletions.
1 change: 1 addition & 0 deletions python/tvm/tir/op.py
Original file line number Diff line number Diff line change
Expand Up @@ -1812,6 +1812,7 @@ def vectorcombine(dtype, vec1, vec2):
"""
return call_intrin(dtype, "tir.vectorcombine", vec1, vec2)


def dp4a(vec1, vec2, acc = 0):
"""Dot product of two int8x4 vectors and add an optional accumulation
Expand Down
2 changes: 2 additions & 0 deletions tests/python/tir-base/test_tir_op_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -294,13 +294,15 @@ def test_tir_op_vectorhigh():
expr = tir.vectorhigh("int8x8", vec)
assert expr.op.name == "tir.vectorhigh"


def test_tir_op_dp4a():
vec1 = tir.Var("vec1", dtype="int8x4")
vec2 = tir.Var("vec2", dtype="int8x4")
acc = tir.Var("acc", dtype="int32")
expr = tir.dp4a(vec1, vec2, acc)
assert expr.op.name == "tir.dp4a"


def test_tir_op_vectorcombine():
buffer = tir.decl_buffer((4, 4), "int8", offset_factor=1)
vec = buffer.vload([0, 0], dtype="int8x16")
Expand Down

0 comments on commit d59ede3

Please sign in to comment.