Differential D102962 Diff 350344 mlir/include/mlir/Conversion/VectorToGPU/VectorToGPU.h

Changeset View

Standalone View

mlir/include/mlir/Conversion/VectorToGPU/VectorToGPU.h

This file was added.

				#ifndef MLIR_INCLUDE_MLIR_CONVERSION_VECTORTOSCF_VECTORTOGPU_H_
				Lint: Pre-merge checks Inline Actions clang-tidy: warning: header guard does not follow preferred style [llvm-header-guard] not useful Lint: Pre-merge checks: clang-tidy: warning: header guard does not follow preferred style [llvm-header-guard] [[https…
				#define MLIR_INCLUDE_MLIR_CONVERSION_VECTORTOSCF_VECTORTOGPU_H_

				#include "mlir/IR/PatternMatch.h"

				namespace mlir {
				class MLIRContext;
				class Pass;
				class FuncOp;
				class RewritePatternSet;

				/// Patterns to transform vector ops into a canonical form to convert to MMA
				/// matrix operations.
				void populatePrepareVectorToMMAPatterns(RewritePatternSet &patterns);

				/// Convert vector ops to MMA matrix operations. This will convert slice of
				/// operations that can be legally converted to MMA operations. The rest of the
				/// vector operations are left untouched.
				void convertVectorToMMAOps(FuncOp funcOp);

				/// Convert from vector to GPU ops.
				std::unique_ptr<Pass> createConvertVectorToGPUPass();
				bondhugulaUnsubmitted Not Done Reply Inline Actions Pass to test convertion -> Convert I guess this is not a test pass but the conversion itself. bondhugula: Pass to test convertion -> Convert I guess this is not a test pass but the conversion itself.
				ThomasRaouxAuthorUnsubmitted Done Reply Inline Actions Done, you're right, I was originally thinking about having this only for test but at this point it is really a conversion pass. ThomasRaoux: Done, you're right, I was originally thinking about having this only for test but at this point…

				} // namespace mlir

				#endif // MLIR_INCLUDE_MLIR_CONVERSION_VECTORTOSCF_VECTORTOGPU_H_