Skip to content

Commit

Permalink
Add lit test for tkw.min
Browse files Browse the repository at this point in the history
Signed-off-by: nouman-10x <noumanamir453@gmail.com>
  • Loading branch information
NoumanAmir657 committed Dec 19, 2024
1 parent 502210e commit 1488483
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions lit_tests/kernel/wave/codegen.py
Original file line number Diff line number Diff line change
Expand Up @@ -1281,6 +1281,7 @@ def test(
res = a_reg - b_reg
res = res * a_reg
res = res / b_reg
res = tkw.minimum(a_reg, b_reg)
tkw.write(res, a, elements_per_thread=4)

a = torch.randn(16, 16, dtype=torch.float16)
Expand All @@ -1291,6 +1292,7 @@ def test(
# CHECK: %[[SUB:.+]] = arith.subf
# CHECK: %[[MUL:.+]] = arith.mulf %[[SUB]]
# CHECK: %[[DIV:.+]] = arith.divf %[[MUL]]
# CHECK: %[[MINIMUM:.+]] = arith.minimumf


# TODO: Something is broken in codegen and we are getting int in place of fx.Node
Expand Down

0 comments on commit 1488483

Please sign in to comment.