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
56 lines
1.7 KiB
C++
56 lines
1.7 KiB
C++
//===- TestMemRefBoundCheck.cpp - Test out of bound access checks ---------===//
|
|
//
|
|
// 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 a pass to check memref accesses for out of bound
|
|
// accesses.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Analysis/AffineAnalysis.h"
|
|
#include "mlir/Analysis/AffineStructures.h"
|
|
#include "mlir/Analysis/Utils.h"
|
|
#include "mlir/Dialect/Affine/IR/AffineOps.h"
|
|
#include "mlir/Dialect/StandardOps/IR/Ops.h"
|
|
#include "mlir/IR/Builders.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "llvm/ADT/TypeSwitch.h"
|
|
#include "llvm/Support/Debug.h"
|
|
|
|
#define DEBUG_TYPE "memref-bound-check"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
|
|
/// Checks for out of bound memref access subscripts..
|
|
struct TestMemRefBoundCheck
|
|
: public PassWrapper<TestMemRefBoundCheck, FunctionPass> {
|
|
void runOnFunction() override;
|
|
};
|
|
|
|
} // end anonymous namespace
|
|
|
|
void TestMemRefBoundCheck::runOnFunction() {
|
|
getFunction().walk([](Operation *opInst) {
|
|
TypeSwitch<Operation *>(opInst)
|
|
.Case<AffineReadOpInterface, AffineWriteOpInterface>(
|
|
[](auto op) { boundCheckLoadOrStoreOp(op); });
|
|
|
|
// TODO: do this for DMA ops as well.
|
|
});
|
|
}
|
|
|
|
namespace mlir {
|
|
namespace test {
|
|
void registerMemRefBoundCheck() {
|
|
PassRegistration<TestMemRefBoundCheck>(
|
|
"test-memref-bound-check", "Check memref access bounds in a Function");
|
|
}
|
|
} // namespace test
|
|
} // namespace mlir
|