python-cpp/src/executable/mlir/Conversion/PythonToPythonBytecode/AttributeSubscriptPatterns.cpp at f2d126b3f32e06a6140710185506c40edbdade1c · gf712/python-cpp · GitHub

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
#include "Conversion/PythonToPythonBytecode/LoweringHelpers.hpp"
#include "Conversion/PythonToPythonBytecode/PatternPopulators.hpp"

#include "Dialect/EmitPythonBytecode/IR/EmitPythonBytecode.hpp"
#include "Dialect/Python/IR/PythonOps.hpp"

#include "mlir/IR/PatternMatch.h"

namespace mlir::py {
namespace {

	using LoadAttributeOpLowering = detail::DirectReplaceLowering<mlir::py::LoadAttributeOp,
		mlir::emitpybytecode::LoadAttribute>;

	using DeleteAttributeOpLowering = detail::DirectReplaceLowering<mlir::py::DeleteAttributeOp,
		mlir::emitpybytecode::DeleteAttribute>;

	using LoadMethodOpLowering = detail::DirectReplaceRegisterName<mlir::py::LoadMethodOp,
		mlir::emitpybytecode::LoadMethod,
		&mlir::py::LoadMethodOp::getMethodName>;

	using BinarySubscriptOpLowering = detail::DirectReplaceLowering<mlir::py::BinarySubscriptOp,
		mlir::emitpybytecode::BinarySubscript>;

	using StoreSubscriptOpLowering = detail::DirectReplaceLowering<mlir::py::StoreSubscriptOp,
		mlir::emitpybytecode::StoreSubscript>;

	using DeleteSubscriptOpLowering = detail::DirectReplaceLowering<mlir::py::DeleteSubscriptOp,
		mlir::emitpybytecode::DeleteSubscript>;

	using StoreAttributeOpLowering = detail::DirectReplaceRegisterName<mlir::py::StoreAttributeOp,
		mlir::emitpybytecode::StoreAttribute,
		&mlir::py::StoreAttributeOp::getAttr>;

}// namespace

void populateAttributeSubscriptPatterns(mlir::RewritePatternSet &patterns)
{
	auto *ctx = patterns.getContext();
	patterns.add<LoadAttributeOpLowering,
		DeleteAttributeOpLowering,
		LoadMethodOpLowering,
		BinarySubscriptOpLowering,
		StoreSubscriptOpLowering,
		DeleteSubscriptOpLowering,
		StoreAttributeOpLowering>(ctx);
}

}// namespace mlir::py