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
62 lines
1.9 KiB
C++
62 lines
1.9 KiB
C++
//===- TestLinalgHoisting.cpp - Test Linalg hoisting functions ------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements logic for testing Linalg hoisting functions.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/Affine/IR/AffineOps.h"
|
|
#include "mlir/Dialect/Linalg/IR/LinalgOps.h"
|
|
#include "mlir/Dialect/Linalg/Transforms/Hoisting.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::linalg;
|
|
|
|
namespace {
|
|
struct TestLinalgHoisting
|
|
: public PassWrapper<TestLinalgHoisting, FunctionPass> {
|
|
TestLinalgHoisting() = default;
|
|
TestLinalgHoisting(const TestLinalgHoisting &pass) {}
|
|
void getDependentDialects(DialectRegistry ®istry) const override {
|
|
registry.insert<AffineDialect>();
|
|
}
|
|
|
|
void runOnFunction() override;
|
|
|
|
Option<bool> testHoistViewAllocs{
|
|
*this, "test-hoist-view-allocs",
|
|
llvm::cl::desc("Test hoisting alloc used by view"),
|
|
llvm::cl::init(false)};
|
|
Option<bool> testHoistRedundantTransfers{
|
|
*this, "test-hoist-redundant-transfers",
|
|
llvm::cl::desc("Test hoisting transfer_read/transfer_write pairs"),
|
|
llvm::cl::init(false)};
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
void TestLinalgHoisting::runOnFunction() {
|
|
if (testHoistViewAllocs) {
|
|
hoistViewAllocOps(getFunction());
|
|
return;
|
|
}
|
|
if (testHoistRedundantTransfers) {
|
|
hoistRedundantVectorTransfers(getFunction());
|
|
return;
|
|
}
|
|
}
|
|
|
|
namespace mlir {
|
|
namespace test {
|
|
void registerTestLinalgHoisting() {
|
|
PassRegistration<TestLinalgHoisting> testTestLinalgHoistingPass(
|
|
"test-linalg-hoisting", "Test Linalg hoisting functions.");
|
|
}
|
|
} // namespace test
|
|
} // namespace mlir
|