fix(compiler): predicate OpenMP loop parallelization of Linalg on the --parallelize flag.

This commit is contained in:
Antoniu Pop
2022-01-31 09:07:03 +00:00
committed by Antoniu Pop
parent f52965e22f
commit ec633d57f6
3 changed files with 16 additions and 8 deletions

View File

@@ -48,7 +48,8 @@ lowerConcreteToStd(mlir::MLIRContext &context, mlir::ModuleOp &module,
mlir::LogicalResult
lowerStdToLLVMDialect(mlir::MLIRContext &context, mlir::ModuleOp &module,
std::function<bool(mlir::Pass *)> enablePass);
std::function<bool(mlir::Pass *)> enablePass,
bool parallelizeLoops);
mlir::LogicalResult optimizeLLVMModule(llvm::LLVMContext &llvmContext,
llvm::Module &module);