4157455425
The only benefit of FunctionPass is that it filters out function declarations. This isn't enough to justify carrying it around, as we can simplify filter out declarations when necessary within the pass. We can also explore with better scheduling primitives to filter out declarations at the pipeline level in the future. The definition of FunctionPass is left intact for now to allow time for downstream users to migrate. Differential Revision: https://reviews.llvm.org/D117182
56 lines
2.1 KiB
C++
56 lines
2.1 KiB
C++
//===- TestDataLayoutQuery.cpp - Test Data Layout Queries -----------------===//
|
|
//
|
|
// 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/Analysis/DataLayoutAnalysis.h"
|
|
#include "mlir/Dialect/DLTI/DLTI.h"
|
|
#include "mlir/IR/BuiltinAttributes.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace {
|
|
|
|
/// A pass that finds "test.data_layout_query" operations and attaches to them
|
|
/// attributes containing the results of data layout queries for operation
|
|
/// result types.
|
|
struct TestDataLayoutQuery
|
|
: public PassWrapper<TestDataLayoutQuery, OperationPass<FuncOp>> {
|
|
StringRef getArgument() const final { return "test-data-layout-query"; }
|
|
StringRef getDescription() const final { return "Test data layout queries"; }
|
|
void runOnOperation() override {
|
|
FuncOp func = getOperation();
|
|
Builder builder(func.getContext());
|
|
const DataLayoutAnalysis &layouts = getAnalysis<DataLayoutAnalysis>();
|
|
|
|
func.walk([&](test::DataLayoutQueryOp op) {
|
|
// Skip the ops with already processed in a deeper call.
|
|
if (op->getAttr("size"))
|
|
return;
|
|
|
|
const DataLayout &layout = layouts.getAbove(op);
|
|
unsigned size = layout.getTypeSize(op.getType());
|
|
unsigned bitsize = layout.getTypeSizeInBits(op.getType());
|
|
unsigned alignment = layout.getTypeABIAlignment(op.getType());
|
|
unsigned preferred = layout.getTypePreferredAlignment(op.getType());
|
|
op->setAttrs(
|
|
{builder.getNamedAttr("size", builder.getIndexAttr(size)),
|
|
builder.getNamedAttr("bitsize", builder.getIndexAttr(bitsize)),
|
|
builder.getNamedAttr("alignment", builder.getIndexAttr(alignment)),
|
|
builder.getNamedAttr("preferred", builder.getIndexAttr(preferred))});
|
|
});
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
namespace mlir {
|
|
namespace test {
|
|
void registerTestDataLayoutQuery() { PassRegistration<TestDataLayoutQuery>(); }
|
|
} // namespace test
|
|
} // namespace mlir
|