MLIR  16.0.0git
Namespaces | Functions
VectorToGPU.h File Reference
#include "mlir/IR/PatternMatch.h"
+ Include dependency graph for VectorToGPU.h:
+ This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

 mlir
 Include the generated interface declarations.
 

Functions

void mlir::populatePrepareVectorToMMAPatterns (RewritePatternSet &patterns, bool useNvGpu=false)
 Patterns to transform vector ops into a canonical form to convert to MMA matrix operations. More...
 
void mlir::convertVectorToMMAOps (Operation *rootOp)
 Convert vector ops to MMA matrix operations nested under rootOp. More...
 
LogicalResult mlir::convertVectorToNVVMCompatibleMMASync (Operation *rootOp)
 Convert vector ops ops nested under rootOp to vector and GPU operaitons compatible with the nvvm.mma.sync lowering path. More...
 
std::unique_ptr< Passmlir::createConvertVectorToGPUPass (bool useNvGpu=false)
 Convert from vector to GPU ops. More...