We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 68bf33c commit 96a2e78Copy full SHA for 96a2e78
python/src/passes.cc
@@ -17,7 +17,8 @@ namespace py = pybind11;
17
18
void init_triton_analysis(py::module &&m) {
19
py::class_<mlir::ModuleAllocation>(m, "allocation", py::module_local())
20
- .def(py::init<mlir::ModuleOp>());
+ .def(py::init(
21
+ &mlir::ModuleAllocation::get<mlir::triton::AllocationAnalysis>));
22
py::class_<mlir::ModuleMembarAnalysis>(m, "membar", py::module_local())
23
.def(py::init<mlir::ModuleAllocation *>())
24
.def("run", &mlir::ModuleMembarAnalysis::run);
0 commit comments