20 #include "mlir/Transforms/DialectConversion.h"
22 using namespace circt;
33 DenseMap<hw::InnerRefAttr, std::pair<hw::StructType, Location>>
argTypes;
37 DenseMap<std::pair<ClassOp, StringAttr>, ClassOp>
instanceMap;
40 ClassOp
lookupNext(ClassOp scope, StringAttr instSym)
const {
41 auto entry = instanceMap.find(std::make_pair(scope, instSym));
42 if (entry == instanceMap.end())
48 static hw::InnerRefAttr
getSymbol(MethodOp method) {
49 auto design = method->getParentOfType<DesignOp>();
51 method.getInnerSym().getSymName());
56 auto *
ctxt = design.getContext();
57 StringAttr modName = design.getNameAttr();
60 for (
auto cls : design.getOps<ClassOp>())
62 modName, cls.getInnerSymAttr().getSymName())] = cls;
64 for (
auto cls : design.getOps<ClassOp>()) {
66 for (
auto method : cls.getOps<MethodOp>()) {
69 SmallVector<hw::StructType::FieldInfo> argFields;
70 for (
auto [argName, argType] :
71 llvm::zip(method.getArgNamesAttr().getAsRange<StringAttr>(),
72 cast<MethodLikeOpInterface>(method.getOperation())
74 argFields.push_back({argName, argType});
80 if (method->getNumRegions() > 0) {
81 SmallVector<Location> argLocs;
82 Block *body = &method.getBody().front();
83 for (
auto arg : body->getArguments())
84 argLocs.push_back(arg.getLoc());
90 std::make_pair(getSymbol(method), std::make_pair(argStruct, argLoc)));
94 for (
auto inst : cls.getOps<InstanceOp>()) {
95 auto clsEntry = classSymbols.find(inst.getTargetNameAttr());
96 assert(clsEntry != classSymbols.end() &&
97 "class being instantiated doesn't exist");
98 instanceMap[std::make_pair(cls, inst.getInnerSym().getSymName())] =
111 void rewrite(CallOp, OpAdaptor adaptor,
112 ConversionPatternRewriter &rewriter)
const final;
113 LogicalResult match(CallOp)
const override {
return success(); }
120 void MergeCallArgs::rewrite(CallOp call, OpAdaptor adaptor,
121 ConversionPatternRewriter &rewriter)
const {
122 auto loc = call.getLoc();
123 rewriter.setInsertionPoint(call);
124 auto method = call->getParentOfType<ibis::MethodLikeOpInterface>();
127 auto argStructEntry = info.
argTypes.find(call.getCalleeAttr());
128 assert(argStructEntry != info.argTypes.end() &&
"Method symref not found!");
129 auto [argStruct, argLoc] = argStructEntry->second;
133 adaptor.getOperands());
134 newArg->setAttr(
"sv.namehint",
135 rewriter.getStringAttr(call.getCallee().getName().getValue() +
136 "_args_called_from_" +
137 method.getMethodName().getValue()));
140 rewriter.modifyOpInPlace(call, [&]() {
141 call.getOperandsMutable().clear();
142 call.getOperandsMutable().append(newArg.getResult());
153 void rewrite(MethodOp, OpAdaptor adaptor,
154 ConversionPatternRewriter &rewriter)
const final;
155 LogicalResult match(MethodOp)
const override {
return success(); }
162 void MergeMethodArgs::rewrite(MethodOp func, OpAdaptor adaptor,
163 ConversionPatternRewriter &rewriter)
const {
164 auto loc = func.getLoc();
165 auto *
ctxt = getContext();
168 auto argStructEntry =
170 assert(argStructEntry != info.argTypes.end() &&
"Cannot find symref!");
171 auto [argStruct, argLoc] = argStructEntry->second;
174 FunctionType funcType = func.getFunctionType();
175 FunctionType newFuncType =
179 rewriter.create<MethodOp>(loc, func.getInnerSym(), newFuncType,
180 newArgNames, ArrayAttr(), ArrayAttr());
182 if (func->getNumRegions() > 0) {
186 rewriter.createBlock(&newMethod.getRegion(), {}, {argStruct}, {argLoc});
187 rewriter.setInsertionPointToStart(b);
188 auto replacementArgs =
189 rewriter.create<hw::StructExplodeOp>(loc, b->getArgument(0));
192 Block *funcBody = &func.getBody().front();
193 rewriter.mergeBlocks(funcBody, b, replacementArgs.getResults());
196 rewriter.eraseOp(func);
201 struct CallPrepPass :
public IbisCallPrepBase<CallPrepPass> {
202 void runOnOperation()
override;
210 void CallPrepPass::runOnOperation() {
213 if (failed(merge(info))) {
221 ConversionTarget target(getContext());
222 target.markUnknownOpDynamicallyLegal([](Operation *) {
return true; });
223 target.addDynamicallyLegalOp<CallOp>([](CallOp call) {
224 auto argValues = call.getArgOperands();
225 return argValues.size() == 1 &&
226 hw::type_isa<hw::StructType>(argValues.front().getType());
228 target.addDynamicallyLegalOp<MethodOp>([](MethodOp func) {
229 ArrayRef<Type> argTypes = func.getFunctionType().getInputs();
230 return argTypes.size() == 1 &&
231 hw::type_isa<hw::StructType>(argTypes.front());
235 RewritePatternSet
patterns(&getContext());
236 patterns.insert<MergeCallArgs>(&getContext(), info);
237 patterns.insert<MergeMethodArgs>(&getContext(), info);
239 return applyPartialConversion(getOperation(), target, std::move(
patterns));
243 return std::make_unique<CallPrepPass>();
assert(baseType &&"element must be base type")
Direction get(bool isOutput)
Returns an output direction if isOutput is true, otherwise returns an input direction.
std::unique_ptr< mlir::Pass > createCallPrepPass()
The InstanceGraph op interface, see InstanceGraphInterface.td for more details.
Build indexes to make lookups faster. Create the new argument types as well.
ClassOp lookupNext(ClassOp scope, StringAttr instSym) const
DenseMap< std::pair< ClassOp, StringAttr >, ClassOp > instanceMap
DenseMap< hw::InnerRefAttr, std::pair< hw::StructType, Location > > argTypes
CallPrepPrecomputed(DesignOp)
DenseMap< hw::InnerRefAttr, ClassOp > classSymbols
static hw::InnerRefAttr getSymbol(MethodOp method)