Page MenuHomePhabricator
Differential D115491 Diff 393462 mlir/test/lib/Dialect/Linalg/TestComprehensiveBufferize.cpp

Changeset View

Standalone View

View Options

mlir/test/lib/Dialect/Linalg/TestComprehensiveBufferize.cpp

Show First 20 Lines • Show All 95 Lines • ▼ Show 20 Linesvoid TestComprehensiveFunctionBufferize::runOnFunction() {
BufferizationOptions options; BufferizationOptions options;
// Enable InitTensorOp elimination. // Enable InitTensorOp elimination.
options.addPostAnalysisStep< options.addPostAnalysisStep<
linalg_ext::InsertSliceAnchoredInitTensorEliminationStep>(); linalg_ext::InsertSliceAnchoredInitTensorEliminationStep>();
// TODO: Find a way to enable this step automatically when bufferizing // TODO: Find a way to enable this step automatically when bufferizing
// tensor dialect ops. // tensor dialect ops.
options.addPostAnalysisStep<tensor_ext::InplaceInsertSliceOpAnalysis>(); options.addPostAnalysisStep<tensor_ext::InplaceInsertSliceOpAnalysis>();
if (!allowReturnMemref)
options.addPostAnalysisStep<scf_ext::AssertDestinationPassingStyle>();
options.allowReturnMemref = allowReturnMemref; options.allowReturnMemref = allowReturnMemref;
options.allowUnknownOps = allowUnknownOps; options.allowUnknownOps = allowUnknownOps;
options.testAnalysisOnly = testAnalysisOnly; options.testAnalysisOnly = testAnalysisOnly;
options.analysisFuzzerSeed = analysisFuzzerSeed; options.analysisFuzzerSeed = analysisFuzzerSeed;
if (dialectFilter.hasValue()) { if (dialectFilter.hasValue()) {
options.dialectFilter.emplace(); options.dialectFilter.emplace();
Show All 22 Lines