We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b7243a7 commit ba0c1faCopy full SHA for ba0c1fa
1 file changed
enzyme/Enzyme/MLIR/Passes/EnzymeMLIRPass.cpp
@@ -17,6 +17,7 @@
17
18
#include "mlir/Dialect/Func/IR/FuncOps.h"
19
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
20
+#include "mlir/Dialect/Linalg/IR/Linalg.h"
21
#include "mlir/IR/Builders.h"
22
#include "mlir/Interfaces/FunctionInterfaces.h"
23
#include "mlir/Pass/PassManager.h"
@@ -52,7 +53,7 @@ struct DifferentiatePass
52
53
54
registry.insert<mlir::arith::ArithDialect, mlir::complex::ComplexDialect,
55
mlir::cf::ControlFlowDialect, mlir::tensor::TensorDialect,
- mlir::enzyme::EnzymeDialect>();
56
+ mlir::linalg::LinalgDialect, mlir::enzyme::EnzymeDialect>();
57
}
58
59
static std::vector<DIFFE_TYPE> mode_from_fn(FunctionOpInterface fn,
0 commit comments