30 lines
740 B
C++
30 lines
740 B
C++
|
//===- UtilityTest.cpp - Tests for
|
||
|
|
// Utility----------------------------------===//
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#include "triton/Analysis/Utility.h"
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
|
||
|
|
namespace mlir {
|
||
|
|
|
||
|
|
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
|