74 m.doc() =
"CIRCT Python Native Extension";
76 llvm::sys::PrintStackTraceOnErrorSignal(
"");
77 LLVMEnablePrettyStackTrace();
81 [](nb::object capsule) {
83 auto wrappedCapsule = capsule.attr(MLIR_PYTHON_CAPI_PTR_ATTR);
84 MlirContext context = mlirPythonCapsuleToContext(wrappedCapsule.ptr());
87 MlirDialectHandle
aig = mlirGetDialectHandle__aig__();
88 mlirDialectHandleRegisterDialect(
aig, context);
89 mlirDialectHandleLoadDialect(
aig, context);
91 MlirDialectHandle
comb = mlirGetDialectHandle__comb__();
92 mlirDialectHandleRegisterDialect(
comb, context);
93 mlirDialectHandleLoadDialect(
comb, context);
95 MlirDialectHandle
debug = mlirGetDialectHandle__debug__();
96 mlirDialectHandleRegisterDialect(
debug, context);
97 mlirDialectHandleLoadDialect(
debug, context);
99 MlirDialectHandle
emit = mlirGetDialectHandle__emit__();
100 mlirDialectHandleRegisterDialect(
emit, context);
101 mlirDialectHandleLoadDialect(
emit, context);
103 MlirDialectHandle
esi = mlirGetDialectHandle__esi__();
104 mlirDialectHandleRegisterDialect(
esi, context);
105 mlirDialectHandleLoadDialect(
esi, context);
107 MlirDialectHandle
msft = mlirGetDialectHandle__msft__();
108 mlirDialectHandleRegisterDialect(
msft, context);
109 mlirDialectHandleLoadDialect(
msft, context);
111 MlirDialectHandle
hw = mlirGetDialectHandle__hw__();
112 mlirDialectHandleRegisterDialect(
hw, context);
113 mlirDialectHandleLoadDialect(
hw, context);
115 MlirDialectHandle
hwarith = mlirGetDialectHandle__hwarith__();
116 mlirDialectHandleRegisterDialect(
hwarith, context);
117 mlirDialectHandleLoadDialect(
hwarith, context);
119 MlirDialectHandle index = mlirGetDialectHandle__index__();
120 mlirDialectHandleRegisterDialect(index, context);
121 mlirDialectHandleLoadDialect(index, context);
123 MlirDialectHandle scf = mlirGetDialectHandle__scf__();
124 mlirDialectHandleRegisterDialect(scf, context);
125 mlirDialectHandleLoadDialect(scf, context);
127 MlirDialectHandle
om = mlirGetDialectHandle__om__();
128 mlirDialectHandleRegisterDialect(
om, context);
129 mlirDialectHandleLoadDialect(
om, context);
131 MlirDialectHandle pipeline = mlirGetDialectHandle__pipeline__();
132 mlirDialectHandleRegisterDialect(pipeline, context);
133 mlirDialectHandleLoadDialect(pipeline, context);
135 MlirDialectHandle
rtg = mlirGetDialectHandle__rtg__();
136 mlirDialectHandleRegisterDialect(
rtg, context);
137 mlirDialectHandleLoadDialect(
rtg, context);
139#ifdef CIRCT_INCLUDE_TESTS
140 MlirDialectHandle
rtgtest = mlirGetDialectHandle__rtgtest__();
141 mlirDialectHandleRegisterDialect(
rtgtest, context);
142 mlirDialectHandleLoadDialect(
rtgtest, context);
145 MlirDialectHandle
seq = mlirGetDialectHandle__seq__();
146 mlirDialectHandleRegisterDialect(
seq, context);
147 mlirDialectHandleLoadDialect(
seq, context);
149 MlirDialectHandle
sv = mlirGetDialectHandle__sv__();
150 mlirDialectHandleRegisterDialect(
sv, context);
151 mlirDialectHandleLoadDialect(
sv, context);
153 MlirDialectHandle
fsm = mlirGetDialectHandle__fsm__();
154 mlirDialectHandleRegisterDialect(
fsm, context);
155 mlirDialectHandleLoadDialect(
fsm, context);
157 MlirDialectHandle
handshake = mlirGetDialectHandle__handshake__();
158 mlirDialectHandleRegisterDialect(
handshake, context);
159 mlirDialectHandleLoadDialect(
handshake, context);
161 MlirDialectHandle kanagawa = mlirGetDialectHandle__kanagawa__();
162 mlirDialectHandleRegisterDialect(kanagawa, context);
163 mlirDialectHandleLoadDialect(kanagawa, context);
165 MlirDialectHandle
ltl = mlirGetDialectHandle__ltl__();
166 mlirDialectHandleRegisterDialect(
ltl, context);
167 mlirDialectHandleLoadDialect(
ltl, context);
169 MlirDialectHandle
verif = mlirGetDialectHandle__verif__();
170 mlirDialectHandleRegisterDialect(
verif, context);
171 mlirDialectHandleLoadDialect(
verif, context);
173 MlirDialectHandle smt = mlirGetDialectHandle__smt__();
174 mlirDialectHandleRegisterDialect(smt, context);
175 mlirDialectHandleLoadDialect(smt, context);
177 "Register CIRCT dialects on a PyMlirContext.");
179 m.def(
"export_verilog", [](MlirModule mod, nb::object fileObject) {
181 nb::gil_scoped_release();
185 m.def(
"export_split_verilog", [](MlirModule mod, std::string directory) {
186 auto cDirectory = mlirStringRefCreateFromCString(directory.c_str());
190 nb::module_
aig = m.def_submodule(
"_aig",
"AIG API");
192 nb::module_
esi = m.def_submodule(
"_esi",
"ESI API");
194 nb::module_
msft = m.def_submodule(
"_msft",
"MSFT API");
196 nb::module_
hw = m.def_submodule(
"_hw",
"HW API");
198 nb::module_
seq = m.def_submodule(
"_seq",
"Seq API");
200 nb::module_
om = m.def_submodule(
"_om",
"OM API");
202 nb::module_
rtg = m.def_submodule(
"_rtg",
"RTG API");
204 nb::module_ rtgtool = m.def_submodule(
"_rtgtool",
"RTGTool API");
206#ifdef CIRCT_INCLUDE_TESTS
207 nb::module_
rtgtest = m.def_submodule(
"_rtgtest",
"RTGTest API");
208 circt::python::populateDialectRTGTestSubmodule(
rtgtest);
210 nb::module_
sv = m.def_submodule(
"_sv",
"SV API");
212 nb::module_ support = m.def_submodule(
"_support",
"CIRCT support");