Please use GitHub pull requests for new patches. Avoid migrating existing patches. Phabricator shutdown timeline
Changeset View
Changeset View
Standalone View
Standalone View
mlir/include/mlir/Dialect/SparseTensor/Pipelines/Passes.h
| Show First 20 Lines • Show All 67 Lines • ▼ Show 20 Lines | struct SparseCompilerOptions | ||||
| PassOptions::Option<bool> testBufferizationAnalysisOnly{ | PassOptions::Option<bool> testBufferizationAnalysisOnly{ | ||||
| *this, "test-bufferization-analysis-only", | *this, "test-bufferization-analysis-only", | ||||
| desc("Run only the inplacability analysis"), init(false)}; | desc("Run only the inplacability analysis"), init(false)}; | ||||
| PassOptions::Option<bool> enableBufferInitialization{ | PassOptions::Option<bool> enableBufferInitialization{ | ||||
| *this, "enable-buffer-initialization", | *this, "enable-buffer-initialization", | ||||
| desc("Enable zero-initialization of memory buffers"), init(false)}; | desc("Enable zero-initialization of memory buffers"), init(false)}; | ||||
| PassOptions::Option<bool> createSparseDeallocs{ | |||||
| *this, "create-sparse-deallocs", | |||||
| desc("Specify if the temporary sparse buffer created by the sparse " | |||||
aartbik: by the sparse compiler | |||||
| "compiler should be deallocated. For compatibility with core " | |||||
| "bufferization passes. " | |||||
| "It only takes effect when enable-runtime-library=false, otherwise " | |||||
Should document how this option interacts with enableRuntimeLibrary. That is, I think this is only used/needed for the codegen path rather than the runtime path? wrengr: Should document how this option interacts with `enableRuntimeLibrary`. That is, I think this is… | |||||
Yeah, you are right. Peiming: Yeah, you are right. | |||||
| "the memory storage for sparse tensors are managed by the runtime " | |||||
| "library. See also create-deallocs for BufferizationOption."), | |||||
Not Done ReplyInline ActionsI would rephrase this to "Specifies whether to deallocate the temporary sparse buffers created by the sparse compiler when enable-runtime-library=false. For compatibility with core bufferization passes. See also..." wrengr: I would rephrase this to "Specifies whether to deallocate the temporary sparse buffers created… | |||||
| init(true)}; | |||||
| PassOptions::Option<int32_t> vectorLength{ | PassOptions::Option<int32_t> vectorLength{ | ||||
| *this, "vl", desc("Set the vector length (0 disables vectorization)"), | *this, "vl", desc("Set the vector length (0 disables vectorization)"), | ||||
| init(0)}; | init(0)}; | ||||
| // These options must be kept in sync with `SparseTensorConversionBase`. | // These options must be kept in sync with `SparseTensorConversionBase`. | ||||
| PassOptions::Option<int32_t> sparseToSparse{ | PassOptions::Option<int32_t> sparseToSparse{ | ||||
| *this, "s2s-strategy", | *this, "s2s-strategy", | ||||
| desc("Set the strategy for sparse-to-sparse conversion"), init(0)}; | desc("Set the strategy for sparse-to-sparse conversion"), init(0)}; | ||||
| ▲ Show 20 Lines • Show All 75 Lines • Show Last 20 Lines | |||||
by the sparse compiler