gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Triton-MLIR] uncomment the UT in test_gemm that has already been fixed (#920)

Browse Source
This commit is contained in:
goostavz
2022-11-28 11:23:20 +08:00
committed by GitHub
parent 35c9ec1103
commit 630dc315ee
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/tests/test_gemm.py
View File

@@ -187,7 +187,8 @@ def get_proper_err(a, b, golden):
[128, 256, 128, 4, 128, 256, 32, False, False], [128, 256, 128, 4, 128, 256, 32, False, False],
[256, 128, 64, 4, 256, 128, 16, False, False], [256, 128, 64, 4, 256, 128, 16, False, False],
[128, 64, 128, 4, 128, 64, 32, False, False], [128, 64, 128, 4, 128, 64, 32, False, False],
# [16, 16, 64, 4, 16, 16, 16, False, False], # TODO failed due to pipeline pass [16, 16, 64, 4, 16, 16, 16, False, False],
[32, 32, 64, 4, 32, 32, 32, False, False],
# trans # trans
[128, 64, 128, 4, 128, 64, 32, True, False], [128, 64, 128, 4, 128, 64, 32, True, False],
[128, 64, 128, 4, 128, 64, 32, False, True], [128, 64, 128, 4, 128, 64, 32, False, True],