Skip to content

Commit ba0c1fa

Browse files
authored
load linalg dialect (#2838)
1 parent b7243a7 commit ba0c1fa

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

enzyme/Enzyme/MLIR/Passes/EnzymeMLIRPass.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717

1818
#include "mlir/Dialect/Func/IR/FuncOps.h"
1919
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
20+
#include "mlir/Dialect/Linalg/IR/Linalg.h"
2021
#include "mlir/IR/Builders.h"
2122
#include "mlir/Interfaces/FunctionInterfaces.h"
2223
#include "mlir/Pass/PassManager.h"
@@ -52,7 +53,7 @@ struct DifferentiatePass
5253

5354
registry.insert<mlir::arith::ArithDialect, mlir::complex::ComplexDialect,
5455
mlir::cf::ControlFlowDialect, mlir::tensor::TensorDialect,
55-
mlir::enzyme::EnzymeDialect>();
56+
mlir::linalg::LinalgDialect, mlir::enzyme::EnzymeDialect>();
5657
}
5758

5859
static std::vector<DIFFE_TYPE> mode_from_fn(FunctionOpInterface fn,

0 commit comments

Comments
 (0)