Please use GitHub pull requests for new patches. Avoid migrating existing patches. Phabricator shutdown timeline
Changeset View
Changeset View
Standalone View
Standalone View
mlir/lib/Dialect/SparseTensor/Pipelines/SparseTensorPipelines.cpp
| Show First 20 Lines • Show All 50 Lines • ▼ Show 20 Lines | |||||
| //===----------------------------------------------------------------------===// | //===----------------------------------------------------------------------===// | ||||
| void mlir::sparse_tensor::buildSparseCompiler( | void mlir::sparse_tensor::buildSparseCompiler( | ||||
| OpPassManager &pm, const SparseCompilerOptions &options) { | OpPassManager &pm, const SparseCompilerOptions &options) { | ||||
| pm.addNestedPass<func::FuncOp>(createLinalgGeneralizationPass()); | pm.addNestedPass<func::FuncOp>(createLinalgGeneralizationPass()); | ||||
| pm.addPass(createSparsificationAndBufferizationPass( | pm.addPass(createSparsificationAndBufferizationPass( | ||||
| getBufferizationOptions(options.testBufferizationAnalysisOnly), | getBufferizationOptions(options.testBufferizationAnalysisOnly), | ||||
| options.sparsificationOptions(), options.sparseTensorConversionOptions(), | options.sparsificationOptions(), options.sparseTensorConversionOptions(), | ||||
| options.enableRuntimeLibrary, options.enableBufferInitialization, | options.createSparseDeallocs, options.enableRuntimeLibrary, | ||||
| options.vectorLength, | options.enableBufferInitialization, options.vectorLength, | ||||
| /*enableVLAVectorization=*/options.armSVE, | /*enableVLAVectorization=*/options.armSVE, | ||||
| /*enableSIMDIndex32=*/options.force32BitVectorIndices)); | /*enableSIMDIndex32=*/options.force32BitVectorIndices)); | ||||
| if (options.testBufferizationAnalysisOnly) | if (options.testBufferizationAnalysisOnly) | ||||
| return; | return; | ||||
| pm.addNestedPass<func::FuncOp>(createCanonicalizerPass()); | pm.addNestedPass<func::FuncOp>(createCanonicalizerPass()); | ||||
| pm.addNestedPass<func::FuncOp>( | pm.addNestedPass<func::FuncOp>( | ||||
| mlir::bufferization::createFinalizingBufferizePass()); | mlir::bufferization::createFinalizingBufferizePass()); | ||||
| // TODO(springerm): Add sparse support to the BufferDeallocation pass and add | // TODO(springerm): Add sparse support to the BufferDeallocation pass and add | ||||
| Show All 33 Lines | |||||