105 lines · cpp
1//===- IndexOpsFoldersTest.cpp - unit tests for index op folders ----------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "mlir/Dialect/Index/IR/IndexDialect.h"10#include "mlir/Dialect/Index/IR/IndexOps.h"11#include "mlir/IR/OwningOpRef.h"12#include "gtest/gtest.h"13 14using namespace mlir;15 16namespace {17/// Test fixture for testing operation folders.18class IndexFolderTest : public testing::Test {19public:20 IndexFolderTest() { ctx.getOrLoadDialect<index::IndexDialect>(); }21 22 /// Instantiate an operation, invoke its folder, and return the attribute23 /// result.24 template <typename OpT>25 void foldOp(IntegerAttr &value, Type type, ArrayRef<Attribute> operands);26 27protected:28 /// The MLIR context to use.29 MLIRContext ctx;30 /// A builder to use.31 OpBuilder b{&ctx};32};33} // namespace34 35template <typename OpT>36void IndexFolderTest::foldOp(IntegerAttr &value, Type type,37 ArrayRef<Attribute> operands) {38 // This function returns null so that `ASSERT_*` works within it.39 OperationState state(UnknownLoc::get(&ctx), OpT::getOperationName());40 state.addTypes(type);41 OwningOpRef<OpT> op = cast<OpT>(b.create(state));42 SmallVector<OpFoldResult> results;43 LogicalResult result = op->getOperation()->fold(operands, results);44 // Propagate the failure to the test.45 if (failed(result)) {46 value = nullptr;47 return;48 }49 ASSERT_EQ(results.size(), 1u);50 value = dyn_cast_or_null<IntegerAttr>(dyn_cast<Attribute>(results.front()));51 ASSERT_TRUE(value);52}53 54TEST_F(IndexFolderTest, TestCastUOpFolder) {55 IntegerAttr value;56 auto fold = [&](Type type, Attribute input) {57 foldOp<index::CastUOp>(value, type, input);58 };59 60 // Target width less than or equal to 32 bits.61 fold(b.getIntegerType(16), b.getIndexAttr(8000000000));62 ASSERT_TRUE(value);63 EXPECT_EQ(value.getInt(), 20480u);64 65 // Target width greater than or equal to 64 bits.66 fold(b.getIntegerType(64), b.getIndexAttr(2000));67 ASSERT_TRUE(value);68 EXPECT_EQ(value.getInt(), 2000u);69 70 // Fails to fold, because truncating to 32 bits and then extending creates a71 // different value.72 fold(b.getIntegerType(64), b.getIndexAttr(8000000000));73 EXPECT_FALSE(value);74 75 // Target width between 32 and 64 bits.76 fold(b.getIntegerType(40), b.getIndexAttr(0x10000000010000));77 // Fold succeeds because the upper bits are truncated in the cast.78 ASSERT_TRUE(value);79 EXPECT_EQ(value.getInt(), 65536);80 81 // Fails to fold because the upper bits are not truncated.82 fold(b.getIntegerType(60), b.getIndexAttr(0x10000000010000));83 EXPECT_FALSE(value);84}85 86TEST_F(IndexFolderTest, TestCastSOpFolder) {87 IntegerAttr value;88 auto fold = [&](Type type, Attribute input) {89 foldOp<index::CastSOp>(value, type, input);90 };91 92 // Just test the extension cases to ensure signs are being respected.93 94 // Target width greater than or equal to 64 bits.95 fold(b.getIntegerType(64), b.getIndexAttr(-2000));96 ASSERT_TRUE(value);97 EXPECT_EQ(value.getInt(), -2000);98 99 // Target width between 32 and 64 bits.100 fold(b.getIntegerType(40), b.getIndexAttr(-0x10000000010000));101 // Fold succeeds because the upper bits are truncated in the cast.102 ASSERT_TRUE(value);103 EXPECT_EQ(value.getInt(), -65536);104}105