e21adfa32d
This makes ignoring a result explicit by the user, and helps to prevent accidental errors with dropped results. Marking LogicalResult as no discard was always the intention from the beginning, but got lost along the way. Differential Revision: https://reviews.llvm.org/D95841
47 lines
1.6 KiB
C++
47 lines
1.6 KiB
C++
//===- TestTraits.cpp - Test trait folding --------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "TestDialect.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::test;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Trait Folder.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
OpFoldResult TestInvolutionTraitFailingOperationFolderOp::fold(
|
|
ArrayRef<Attribute> operands) {
|
|
// This failure should cause the trait fold to run instead.
|
|
return {};
|
|
}
|
|
|
|
OpFoldResult TestInvolutionTraitSuccesfulOperationFolderOp::fold(
|
|
ArrayRef<Attribute> operands) {
|
|
auto argumentOp = getOperand();
|
|
// The success case should cause the trait fold to be supressed.
|
|
return argumentOp.getDefiningOp() ? argumentOp : OpFoldResult{};
|
|
}
|
|
|
|
namespace {
|
|
struct TestTraitFolder : public PassWrapper<TestTraitFolder, FunctionPass> {
|
|
void runOnFunction() override {
|
|
(void)applyPatternsAndFoldGreedily(getFunction(),
|
|
OwningRewritePatternList());
|
|
}
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
namespace mlir {
|
|
void registerTestTraitsPass() {
|
|
PassRegistration<TestTraitFolder>("test-trait-folder", "Run trait folding");
|
|
}
|
|
} // namespace mlir
|