16#include "mlir/CAPI/IR.h"
17#include "mlir/CAPI/Registration.h"
18#include "mlir/CAPI/Support.h"
19#include "llvm/ADT/PostOrderIterator.h"
45 return wrap(ArrayType::get(
unwrap(element), size));
49 return wrap(cast<ArrayType>(
unwrap(type)).getElementType());
53 return cast<ArrayType>(
unwrap(type)).getNumElements();
59 return wrap(IntType::get(cast<TypedAttr>(
unwrap(parameter))));
63 return wrap(cast<IntType>(
unwrap(type)).getWidth());
71 return wrap(cast<InOutType>(
unwrap(type)).getElementType());
77 return isa<ModuleType>(
unwrap(type));
82 SmallVector<ModulePort> modulePorts;
83 for (intptr_t i = 0; i < numPorts; ++i) {
88 case HWModulePortDirection::Input:
89 dir = ModulePort::Direction::Input;
91 case HWModulePortDirection::Output:
92 dir = ModulePort::Direction::Output;
94 case HWModulePortDirection::InOut:
95 dir = ModulePort::Direction::InOut;
99 StringAttr name = cast<StringAttr>(
unwrap(port.
name));
102 modulePorts.push_back(
ModulePort{name, type, dir});
105 return wrap(ModuleType::get(
unwrap(ctx), modulePorts));
109 return cast<ModuleType>(
unwrap(type)).getNumInputs();
113 return wrap(cast<ModuleType>(
unwrap(type)).getInputType(index));
117 return wrap(cast<ModuleType>(
unwrap(type)).getInputName(index));
121 return cast<ModuleType>(
unwrap(type)).getNumOutputs();
125 return wrap(cast<ModuleType>(
unwrap(type)).getOutputType(index));
129 return wrap(cast<ModuleType>(
unwrap(type)).getOutputName(index));
133 auto port = cast<ModuleType>(
unwrap(type)).getPorts()[index];
137 case ModulePort::Direction::Input:
138 dir = HWModulePortDirection::Input;
140 case ModulePort::Direction::Output:
141 dir = HWModulePortDirection::Output;
143 case ModulePort::Direction::InOut:
144 dir = HWModulePortDirection::InOut;
147 llvm_unreachable(
"unknown ModulePort::Direction");
150 ret->
name =
wrap(
static_cast<Attribute
>(port.name));
156 return isa<StructType>(
unwrap(type));
161 SmallVector<StructType::FieldInfo> fieldInfos;
164 fieldInfos.push_back(StructType::FieldInfo{
165 cast<StringAttr>(
unwrap(elements[i].name)),
unwrap(elements[i].type)});
167 return wrap(StructType::get(
unwrap(ctx), fieldInfos));
171 StructType st = cast<StructType>(
unwrap(structType));
172 return wrap(st.getFieldType(
unwrap(fieldName)));
176 MlirStringRef fieldName) {
177 StructType st = cast<StructType>(
unwrap(structType));
178 if (
auto idx = st.getFieldIndex(
unwrap(fieldName)))
179 return wrap(IntegerAttr::get(IntegerType::get(st.getContext(), 32), *idx));
180 return wrap(UnitAttr::get(st.getContext()));
184 StructType st = cast<StructType>(
unwrap(structType));
185 return st.getElements().size();
189 StructType st = cast<StructType>(
unwrap(structType));
190 auto cppField = st.getElements()[idx];
198 return isa<TypeAliasType>(
unwrap(type));
202 MlirType cInnerType) {
203 StringRef scope =
unwrap(cScope);
204 StringRef name =
unwrap(cName);
205 Type innerType =
unwrap(cInnerType);
206 FlatSymbolRefAttr nameRef =
207 FlatSymbolRefAttr::get(innerType.getContext(), name);
209 SymbolRefAttr::get(innerType.getContext(), scope, {nameRef});
210 return wrap(TypeAliasType::get(ref, innerType));
214 TypeAliasType type = cast<TypeAliasType>(
unwrap(typeAlias));
215 return wrap(type.getCanonicalType());
219 TypeAliasType type = cast<TypeAliasType>(
unwrap(typeAlias));
220 return wrap(type.getInnerType());
224 TypeAliasType type = cast<TypeAliasType>(
unwrap(typeAlias));
225 return wrap(type.getRef().getLeafReference().getValue());
229 TypeAliasType type = cast<TypeAliasType>(
unwrap(typeAlias));
230 return wrap(type.getRef().getRootReference().getValue());
238 return isa<InnerSymAttr>(
unwrap(attr));
242 return wrap(InnerSymAttr::get(cast<StringAttr>(
unwrap(symName))));
250 return wrap((Attribute)cast<InnerSymAttr>(
unwrap(innerSymAttr)).getSymName());
254 return isa<InnerRefAttr>(
unwrap(attr));
258 MlirAttribute innerSym) {
259 auto moduleNameAttr = cast<StringAttr>(
unwrap(moduleName));
260 auto innerSymAttr = cast<StringAttr>(
unwrap(innerSym));
261 return wrap(InnerRefAttr::get(moduleNameAttr, innerSymAttr));
265 return wrap((Attribute)cast<InnerRefAttr>(
unwrap(innerRefAttr)).getName());
269 return wrap((Attribute)cast<InnerRefAttr>(
unwrap(innerRefAttr)).getModule());
273 return isa<ParamDeclAttr>(
unwrap(attr));
277 MlirAttribute cValue) {
278 auto type =
unwrap(cType);
279 auto name = StringAttr::get(type.getContext(),
unwrap(cName));
281 ParamDeclAttr::get(type.getContext(), name, type,
unwrap(cValue)));
284 return wrap(cast<ParamDeclAttr>(
unwrap(decl)).getName().getValue());
287 return wrap(cast<ParamDeclAttr>(
unwrap(decl)).getType());
290 return wrap(cast<ParamDeclAttr>(
unwrap(decl)).getValue());
294 return isa<ParamDeclRefAttr>(
unwrap(attr));
298 MlirStringRef cName) {
299 auto name = StringAttr::get(
unwrap(ctx),
unwrap(cName));
300 return wrap(ParamDeclRefAttr::get(
unwrap(ctx), name,
301 IntegerType::get(
unwrap(ctx), 32)));
305 return wrap(cast<ParamDeclRefAttr>(
unwrap(decl)).getName().getValue());
308 return wrap(cast<ParamDeclRefAttr>(
unwrap(decl)).getType());
312 return isa<ParamVerbatimAttr>(
unwrap(attr));
315 auto textAttr = cast<StringAttr>(
unwrap(text));
316 MLIRContext *ctx = textAttr.getContext();
317 auto type = NoneType::get(ctx);
318 return wrap(ParamVerbatimAttr::get(ctx, textAttr, type));
322 return isa<OutputFileAttr>(
unwrap(attr));
324MLIR_CAPI_EXPORTED MlirAttribute
326 bool includeReplicatedOp) {
327 auto fileNameStrAttr = cast<StringAttr>(
unwrap(fileName));
328 return wrap(OutputFileAttr::getFromFilename(
329 fileNameStrAttr.getContext(), fileNameStrAttr.getValue(),
330 excludeFromFileList, includeReplicatedOp));
334 auto outputFileAttr = cast<OutputFileAttr>(
unwrap(outputFile));
335 return wrap(outputFileAttr.getFilename().getValue());
343 delete unwrap(instanceGraph);
346MLIR_CAPI_EXPORTED HWInstanceGraphNode
348 return wrap(
unwrap(instanceGraph)->getTopLevelNode());
351MLIR_CAPI_EXPORTED
void
356 for (
const auto &inst : llvm::post_order(graph)) {
357 callback(
wrap(inst), userData);
362 HWInstanceGraphNode rhs) {
366MLIR_CAPI_EXPORTED MlirModule
368 return wrap(dyn_cast<ModuleOp>(
unwrap(node)->getModule().getOperation()));
371MLIR_CAPI_EXPORTED MlirOperation
return wrap(CMemoryType::get(unwrap(ctx), baseType, numElements))
MlirType uint64_t numElements
MLIR_DEFINE_CAPI_DIALECT_REGISTRATION(CHIRRTL, chirrtl, circt::chirrtl::CHIRRTLDialect) MlirType chirrtlTypeGetCMemory(MlirContext ctx
static void registerPasses()
MLIR_CAPI_EXPORTED MlirAttribute hwParamDeclAttrGet(MlirStringRef cName, MlirType cType, MlirAttribute cValue)
MlirAttribute hwInnerRefAttrGetModule(MlirAttribute innerRefAttr)
MLIR_CAPI_EXPORTED MlirType hwParamDeclAttrGetType(MlirAttribute decl)
MlirType hwInOutTypeGet(MlirType element)
Creates an HW inout type in the context associated with element.
MlirType hwArrayTypeGet(MlirType element, size_t size)
Creates a fixed-size HW array type in the context associated with element.
intptr_t hwArrayTypeGetSize(MlirType type)
returns the size of an array type
MlirType hwStructTypeGet(MlirContext ctx, intptr_t numElements, HWStructFieldInfo const *elements)
Creates an HW struct type in the context associated with the elements.
MlirType hwArrayTypeGetElementType(MlirType type)
returns the element type of an array type
MLIR_CAPI_EXPORTED HWInstanceGraphNode hwInstanceGraphGetTopLevelNode(HWInstanceGraph instanceGraph)
intptr_t hwModuleTypeGetNumInputs(MlirType type)
Get an HW module type's number of inputs.
MLIR_CAPI_EXPORTED MlirOperation hwInstanceGraphNodeGetModuleOp(HWInstanceGraphNode node)
intptr_t hwStructTypeGetNumFields(MlirType structType)
bool hwTypeIsAArrayType(MlirType type)
If the type is an HW array.
bool hwTypeIsATypeAliasType(MlirType type)
If the type is an HW type alias.
MlirType hwTypeAliasTypeGetInnerType(MlirType typeAlias)
MlirStringRef hwOutputFileGetFileName(MlirAttribute outputFile)
MlirType hwTypeAliasTypeGetCanonicalType(MlirType typeAlias)
MlirType hwStructTypeGetField(MlirType structType, MlirStringRef fieldName)
MlirAttribute hwInnerRefAttrGet(MlirAttribute moduleName, MlirAttribute innerSym)
MLIR_CAPI_EXPORTED bool hwInstanceGraphNodeEqual(HWInstanceGraphNode lhs, HWInstanceGraphNode rhs)
MLIR_CAPI_EXPORTED MlirModule hwInstanceGraphNodeGetModule(HWInstanceGraphNode node)
MLIR_CAPI_EXPORTED HWInstanceGraph hwInstanceGraphGet(MlirOperation operation)
bool hwTypeIsAIntType(MlirType type)
If the type is an HW int.
void hwModuleTypeGetPort(MlirType type, intptr_t index, HWModulePort *ret)
Get an HW module type's port info at a specific index.
MlirType hwModuleTypeGetInputType(MlirType type, intptr_t index)
Get an HW module type's input type at a specific index.
bool hwTypeIsAInOut(MlirType type)
If the type is an HW inout.
MlirStringRef hwModuleTypeGetInputName(MlirType type, intptr_t index)
Get an HW module type's input name at a specific index.
MlirType hwModuleTypeGet(MlirContext ctx, intptr_t numPorts, HWModulePort const *ports)
Creates an HW module type.
MlirAttribute hwStructTypeGetFieldIndex(MlirType structType, MlirStringRef fieldName)
MLIR_CAPI_EXPORTED MlirAttribute hwParamDeclAttrGetValue(MlirAttribute decl)
MlirAttribute hwInnerSymAttrGetSymName(MlirAttribute innerSymAttr)
MlirAttribute hwInnerSymAttrGet(MlirAttribute symName)
MLIR_CAPI_EXPORTED void hwInstanceGraphDestroy(HWInstanceGraph instanceGraph)
MLIR_CAPI_EXPORTED void hwInstanceGraphForEachNode(HWInstanceGraph instanceGraph, HWInstanceGraphNodeCallback callback, void *userData)
MlirStringRef hwTypeAliasTypeGetScope(MlirType typeAlias)
MlirType hwInOutTypeGetElementType(MlirType type)
Returns the element type of an inout type.
MlirType hwTypeAliasTypeGet(MlirStringRef cScope, MlirStringRef cName, MlirType cInnerType)
bool hwTypeIsAValueType(MlirType type)
Return true if the specified type can be used as an HW value type, that is the set of types that can ...
bool hwAttrIsAInnerSymAttr(MlirAttribute attr)
MLIR_CAPI_EXPORTED bool hwAttrIsAParamVerbatimAttr(MlirAttribute attr)
bool hwTypeIsAStructType(MlirType type)
If the type is an HW struct.
MLIR_CAPI_EXPORTED bool hwAttrIsAOutputFileAttr(MlirAttribute attr)
MlirType hwParamIntTypeGet(MlirAttribute parameter)
MlirStringRef hwModuleTypeGetOutputName(MlirType type, intptr_t index)
Get an HW module type's output name at a specific index.
MLIR_CAPI_EXPORTED MlirAttribute hwOutputFileGetFromFileName(MlirAttribute fileName, bool excludeFromFileList, bool includeReplicatedOp)
MlirAttribute hwInnerRefAttrGetName(MlirAttribute innerRefAttr)
MlirAttribute hwInnerSymAttrGetEmpty(MlirContext ctx)
HWStructFieldInfo hwStructTypeGetFieldNum(MlirType structType, unsigned idx)
bool hwTypeIsAModuleType(MlirType type)
If the type is an HW module type.
MLIR_CAPI_EXPORTED bool hwAttrIsAParamDeclRefAttr(MlirAttribute attr)
int64_t hwGetBitWidth(MlirType type)
Return the hardware bit width of a type.
MLIR_CAPI_EXPORTED MlirStringRef hwParamDeclRefAttrGetName(MlirAttribute decl)
MlirStringRef hwTypeAliasTypeGetName(MlirType typeAlias)
bool hwAttrIsAInnerRefAttr(MlirAttribute attr)
MLIR_CAPI_EXPORTED MlirAttribute hwParamDeclRefAttrGet(MlirContext ctx, MlirStringRef cName)
MLIR_CAPI_EXPORTED bool hwAttrIsAParamDeclAttr(MlirAttribute attr)
MlirType hwModuleTypeGetOutputType(MlirType type, intptr_t index)
Get an HW module type's output type at a specific index.
MlirAttribute hwParamIntTypeGetWidthAttr(MlirType type)
intptr_t hwModuleTypeGetNumOutputs(MlirType type)
Get an HW module type's number of outputs.
MLIR_CAPI_EXPORTED MlirStringRef hwParamDeclAttrGetName(MlirAttribute decl)
MLIR_CAPI_EXPORTED MlirAttribute hwParamVerbatimAttrGet(MlirAttribute text)
MLIR_CAPI_EXPORTED MlirType hwParamDeclRefAttrGetType(MlirAttribute decl)
void(* HWInstanceGraphNodeCallback)(HWInstanceGraphNode, void *)
static EvaluatorValuePtr unwrap(OMEvaluatorValue c)
HW-specific instance graph with a virtual entry node linking to all publicly visible modules.
This is a Node in the InstanceGraph.
bool isHWValueType(mlir::Type type)
Return true if the specified type can be used as an HW value type, that is the set of types that can ...
int64_t getBitWidth(mlir::Type type)
Return the hardware bit width of a type.
The InstanceGraph op interface, see InstanceGraphInterface.td for more details.
HWModulePortDirection dir