Skip to content

Commit 96a2e78

Browse files
committed
Fix build error
1 parent 68bf33c commit 96a2e78

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/src/passes.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,8 @@ namespace py = pybind11;
1717

1818
void init_triton_analysis(py::module &&m) {
1919
py::class_<mlir::ModuleAllocation>(m, "allocation", py::module_local())
20-
.def(py::init<mlir::ModuleOp>());
20+
.def(py::init(
21+
&mlir::ModuleAllocation::get<mlir::triton::AllocationAnalysis>));
2122
py::class_<mlir::ModuleMembarAnalysis>(m, "membar", py::module_local())
2223
.def(py::init<mlir::ModuleAllocation *>())
2324
.def("run", &mlir::ModuleMembarAnalysis::run);

0 commit comments

Comments
 (0)