9 #ifndef MLIR_CONVERSION_VECTORTOGPU_VECTORTOGPU_H 10 #define MLIR_CONVERSION_VECTORTOGPU_VECTORTOGPU_H 17 class RewritePatternSet;
24 bool useNvGpu =
false);
42 #endif // MLIR_CONVERSION_VECTORTOGPU_VECTORTOGPU_H Include the generated interface declarations.
void populatePrepareVectorToMMAPatterns(RewritePatternSet &patterns, bool useNvGpu=false)
Patterns to transform vector ops into a canonical form to convert to MMA matrix operations.
std::unique_ptr< Pass > createConvertVectorToGPUPass(bool useNvGpu=false)
Convert from vector to GPU ops.
void convertVectorToMMAOps(Operation *rootOp)
Convert vector ops to MMA matrix operations nested under rootOp.
LogicalResult convertVectorToNVVMCompatibleMMASync(Operation *rootOp)
Convert vector ops ops nested under rootOp to vector and GPU operaitons compatible with the nvvm...