b2797d9b82
This relands commit aa8499863ad23350da0912d99d189f306d0ea139. That commit was originally reverted because it caused failures in shared lib builds due to missing link dependencies. This patch relands the commit with the missing libs added. Signed-off-by: Fabian Mora <fabian.mora-cordero@amd.com>
87 lines
3.0 KiB
C++
87 lines
3.0 KiB
C++
//===- TestStridedMetadataRangeAnalysis.cpp - Test strided md analysis ----===//
|
|
//
|
|
// 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 "mlir/Analysis/DataFlow/ConstantPropagationAnalysis.h"
|
|
#include "mlir/Analysis/DataFlow/DeadCodeAnalysis.h"
|
|
#include "mlir/Analysis/DataFlow/IntegerRangeAnalysis.h"
|
|
#include "mlir/Analysis/DataFlow/StridedMetadataRangeAnalysis.h"
|
|
#include "mlir/Analysis/DataFlowFramework.h"
|
|
#include "mlir/IR/BuiltinAttributes.h"
|
|
#include "mlir/IR/Operation.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Pass/PassRegistry.h"
|
|
#include "llvm/ADT/STLExtras.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::dataflow;
|
|
|
|
static void printAnalysisResults(DataFlowSolver &solver, Operation *op,
|
|
raw_ostream &os) {
|
|
// Collect the strided metadata of the op results.
|
|
SmallVector<std::pair<unsigned, const StridedMetadataRangeLattice *>> results;
|
|
for (OpResult result : op->getResults()) {
|
|
const auto *state = solver.lookupState<StridedMetadataRangeLattice>(result);
|
|
// Skip the result if it's uninitialized.
|
|
if (!state || state->getValue().isUninitialized())
|
|
continue;
|
|
|
|
// Skip the result if the range is empty.
|
|
const mlir::StridedMetadataRange &md = state->getValue();
|
|
if (md.getOffsets().empty() && md.getSizes().empty() &&
|
|
md.getStrides().empty())
|
|
continue;
|
|
results.push_back({result.getResultNumber(), state});
|
|
}
|
|
|
|
// Early exit if there's no metadata to print.
|
|
if (results.empty())
|
|
return;
|
|
|
|
// Print the metadata.
|
|
os << "Op: " << OpWithFlags(op, OpPrintingFlags().skipRegions()) << "\n";
|
|
for (auto [idx, state] : results)
|
|
os << " result[" << idx << "]: " << state->getValue() << "\n";
|
|
os << "\n";
|
|
}
|
|
|
|
namespace {
|
|
struct TestStridedMetadataRangeAnalysisPass
|
|
: public PassWrapper<TestStridedMetadataRangeAnalysisPass,
|
|
OperationPass<>> {
|
|
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(
|
|
TestStridedMetadataRangeAnalysisPass)
|
|
|
|
StringRef getArgument() const override {
|
|
return "test-strided-metadata-range-analysis";
|
|
}
|
|
void runOnOperation() override {
|
|
Operation *op = getOperation();
|
|
|
|
DataFlowSolver solver;
|
|
solver.load<DeadCodeAnalysis>();
|
|
solver.load<SparseConstantPropagation>();
|
|
solver.load<IntegerRangeAnalysis>();
|
|
solver.load<StridedMetadataRangeAnalysis>();
|
|
if (failed(solver.initializeAndRun(op)))
|
|
return signalPassFailure();
|
|
|
|
op->walk(
|
|
[&](Operation *op) { printAnalysisResults(solver, op, llvm::errs()); });
|
|
}
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
namespace mlir {
|
|
namespace test {
|
|
void registerTestStridedMetadataRangeAnalysisPass() {
|
|
PassRegistration<TestStridedMetadataRangeAnalysisPass>();
|
|
}
|
|
} // end namespace test
|
|
} // end namespace mlir
|