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/Transforms/Passes.td
Show First 20 Lines • Show All 214 Lines • ▼ Show 20 Lines | let dependentDialects = [ | ||||
"linalg::LinalgDialect", | "linalg::LinalgDialect", | ||||
"memref::MemRefDialect", | "memref::MemRefDialect", | ||||
"scf::SCFDialect", | "scf::SCFDialect", | ||||
"sparse_tensor::SparseTensorDialect", | "sparse_tensor::SparseTensorDialect", | ||||
]; | ]; | ||||
let options = [ | let options = [ | ||||
Option<"enableBufferInitialization", "enable-buffer-initialization", "bool", | Option<"enableBufferInitialization", "enable-buffer-initialization", "bool", | ||||
"false", "Enable zero-initialization of the memory buffers">, | "false", "Enable zero-initialization of the memory buffers">, | ||||
Option<"createSparseDeallocs", "create-sparse-deallocs", "bool", | |||||
"true", "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 " | |||||
"the memory storage for sparse tensors are managed by the runtime " | |||||
"library. See also create-deallocs for BufferizationOption.">, | |||||
]; | ]; | ||||
} | } | ||||
def SparseBufferRewrite : Pass<"sparse-buffer-rewrite", "ModuleOp"> { | def SparseBufferRewrite : Pass<"sparse-buffer-rewrite", "ModuleOp"> { | ||||
let summary = "Rewrite sparse primitives on buffers to actual code"; | let summary = "Rewrite sparse primitives on buffers to actual code"; | ||||
let description = [{ | let description = [{ | ||||
A pass that rewrites sparse primitives on buffers to the MLIR implementation | A pass that rewrites sparse primitives on buffers to the MLIR implementation | ||||
of the primitives. For example, sparse_tensor.sort operator is implemented | of the primitives. For example, sparse_tensor.sort operator is implemented | ||||
▲ Show 20 Lines • Show All 101 Lines • Show Last 20 Lines |
by the sparse compiler