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

[Triton-MLIR][BACKEND] Refine dot conversion (#710)

Browse Source 
This PR does

1. Refine the dot conversion
2. some other tiny code refinement
This commit is contained in:
Yan Chunwei
2022-09-27 14:38:34 +08:00
committed by GitHub
parent 61b61755e5
commit 3a84278530
11 changed files with 439 additions and 291 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
unittest/Analysis/UtilityTest.cpp
View File

@@ -4,11 +4,26 @@
//===----------------------------------------------------------------------===//
#include "triton/Analysis/Utility.h"
#include <gmock/gmock.h>
#include <gtest/gtest.h>
namespace mlir {
TEST(UtilityTest, DummyTest) { EXPECT_EQ(true, true); }
TEST(Analysis, reorder) {
SmallVector<int> shape({10, 20, 30});
{
SmallVector<unsigned> order({2, 1, 0});
auto reordered = reorder<int>(shape, order);
EXPECT_EQ(reordered[0], 30);
EXPECT_EQ(reordered[1], 20);
EXPECT_EQ(reordered[2], 10);
}
{
SmallVector<unsigned> order({1, 0, 2});
auto reordered = reorder<int>(shape, order);
EXPECT_EQ(reordered[0], 20);
EXPECT_EQ(reordered[1], 10);
EXPECT_EQ(reordered[2], 30);
}
}
} // namespace mlir