diff --git a/mlir/include/mlir/Dialect/Linalg/Transforms/CodegenStrategy.h b/mlir/include/mlir/Dialect/Linalg/Transforms/CodegenStrategy.h --- a/mlir/include/mlir/Dialect/Linalg/Transforms/CodegenStrategy.h +++ b/mlir/include/mlir/Dialect/Linalg/Transforms/CodegenStrategy.h @@ -152,12 +152,12 @@ /// Options to control the application of late transformations. struct LateCodegenStrategyOptions { - bool enableLICM = true; - bool enableHoistRedundantVectorTransfers = true; - bool enableHoistRedundantVectorTransfersOnTensor = true; - bool enableVectorTransferPartialRewrite = true; - bool enableVectorContractLowering = true; - bool enableVectorToSCFConversion = true; + bool enableLICM = false; + bool enableHoistRedundantVectorTransfers = false; + bool enableHoistRedundantVectorTransfersOnTensor = false; + bool enableVectorTransferPartialRewrite = false; + bool enableVectorContractLowering = false; + bool enableVectorToSCFConversion = false; }; /// Codegen strategy controls how a Linalg op is progressively lowered. diff --git a/mlir/test/lib/Dialect/Linalg/TestLinalgCodegenStrategy.cpp b/mlir/test/lib/Dialect/Linalg/TestLinalgCodegenStrategy.cpp --- a/mlir/test/lib/Dialect/Linalg/TestLinalgCodegenStrategy.cpp +++ b/mlir/test/lib/Dialect/Linalg/TestLinalgCodegenStrategy.cpp @@ -148,6 +148,12 @@ .setAlignment(16) .setUseFullTileBuffersByDefault(registerPromoteFullTile)) .vectorizeIf(vectorize, anchorOpName) + .setEnableLICM(true) + .setEnableHoistRedundantVectorTransfers(true) + .setEnableHoistRedundantVectorTransfersOnTensor(true) + .setEnableVectorTransferPartialRewrite(true) + .setEnableVectorContractLowering(true) + .setEnableVectorToSCFConversion(true) .setVectorTransformsOptions( vector::VectorTransformsOptions() .setVectorTransformsOptions(vectorContractLowering) @@ -177,6 +183,12 @@ .setAlignment(16) .setUseFullTileBuffersByDefault(registerPromoteFullTile)) .template vectorizeIf(vectorize) + .setEnableLICM(true) + .setEnableHoistRedundantVectorTransfers(true) + .setEnableHoistRedundantVectorTransfersOnTensor(true) + .setEnableVectorTransferPartialRewrite(true) + .setEnableVectorContractLowering(true) + .setEnableVectorToSCFConversion(true) .setVectorTransformsOptions( vector::VectorTransformsOptions() .setVectorTransformsOptions(vectorContractLowering)