Differential D138915 Diff 479618 mlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.h

Changeset View

Standalone View

mlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.h

	Show First 20 Lines • Show All 156 Lines • ▼ Show 20 Lines
	std::unique_ptr<Pass>			std::unique_ptr<Pass>
	createPostSparsificationRewritePass(bool enableRT, bool enableForeach = true,			createPostSparsificationRewritePass(bool enableRT, bool enableForeach = true,
	bool enableConvert = true);			bool enableConvert = true);

	//===----------------------------------------------------------------------===//			//===----------------------------------------------------------------------===//
	// Other rewriting rules and passes.			// Other rewriting rules and passes.
	//===----------------------------------------------------------------------===//			//===----------------------------------------------------------------------===//

	std::unique_ptr<Pass> createDenseBufferizationPass(			std::unique_ptr<Pass> createSparsificationAndBufferizationPass(
	const bufferization::OneShotBufferizationOptions &options);			const bufferization::OneShotBufferizationOptions &bufferizationOptions,
				const SparsificationOptions &sparsificationOptions,
				const SparseTensorConversionOptions &sparseTensorConversionOptions,
				bool enableRuntimeLibrary, bool enableBufferInitialization);

	void populateSparseBufferRewriting(RewritePatternSet &patterns,			void populateSparseBufferRewriting(RewritePatternSet &patterns,
	bool enableBufferInitialization);			bool enableBufferInitialization);

	std::unique_ptr<Pass> createSparseBufferRewritePass();			std::unique_ptr<Pass> createSparseBufferRewritePass();
	std::unique_ptr<Pass>			std::unique_ptr<Pass>
	createSparseBufferRewritePass(bool enableBufferInitialization);			createSparseBufferRewritePass(bool enableBufferInitialization);

	Show All 21 Lines