mirror of
https://github.com/intel/llvm.git
synced 2026-02-09 01:52:26 +08:00
72c65b698e
TestDialect has many operations and they all live in ::mlir namespace. Sometimes it is not clear whether the ops used in the code for the test passes belong to Standard or to Test dialects. Also, with this change it is easier to understand what test passes registered in mlir-opt are actually passes in mlir/test. Differential Revision: https://reviews.llvm.org/D90794
46 lines
1.5 KiB
C++
46 lines
1.5 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 {
|
|
applyPatternsAndFoldGreedily(getFunction(), OwningRewritePatternList());
|
|
}
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
namespace mlir {
|
|
void registerTestTraitsPass() {
|
|
PassRegistration<TestTraitFolder>("test-trait-folder", "Run trait folding");
|
|
}
|
|
} // namespace mlir
|