blob: f17f5db2fa22fe9cfa101db893c88c1e03952abd (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
//===- TestMemRefStrideCalculation.cpp - Pass to test strides computation--===//
//
// 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/Dialect/MemRef/IR/MemRef.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/Pass/Pass.h"
using namespace mlir;
namespace {
struct TestMemRefStrideCalculation
: public PassWrapper<TestMemRefStrideCalculation,
InterfacePass<SymbolOpInterface>> {
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestMemRefStrideCalculation)
StringRef getArgument() const final {
return "test-memref-stride-calculation";
}
StringRef getDescription() const final {
return "Test operation constant folding";
}
void runOnOperation() override;
};
} // namespace
/// Traverse AllocOp and compute strides of each MemRefType independently.
void TestMemRefStrideCalculation::runOnOperation() {
llvm::outs() << "Testing: " << getOperation().getName() << "\n";
getOperation().walk([&](memref::AllocOp allocOp) {
auto memrefType = cast<MemRefType>(allocOp.getResult().getType());
int64_t offset;
SmallVector<int64_t, 4> strides;
if (failed(memrefType.getStridesAndOffset(strides, offset))) {
llvm::outs() << "MemRefType " << memrefType << " cannot be converted to "
<< "strided form\n";
return;
}
llvm::outs() << "MemRefType offset: ";
if (ShapedType::isDynamic(offset))
llvm::outs() << "?";
else
llvm::outs() << offset;
llvm::outs() << " strides: ";
llvm::interleaveComma(strides, llvm::outs(), [&](int64_t v) {
if (ShapedType::isDynamic(v))
llvm::outs() << "?";
else
llvm::outs() << v;
});
llvm::outs() << "\n";
});
llvm::outs().flush();
}
namespace mlir {
namespace test {
void registerTestMemRefStrideCalculation() {
PassRegistration<TestMemRefStrideCalculation>();
}
} // namespace test
} // namespace mlir
|
