|
| SmallVector< Value > | mlir::xegpu::flattenValues (ArrayRef< ValueRange > values) |
| | Flatten a set of ValueRange into a single SmallVector<Value> More...
|
| |
| FailureOr< VectorType > | mlir::xegpu::getDistributedVectorType (xegpu::TensorDescType tdescTy) |
| | If tensor descriptor has a layout attribute it is used in SIMT mode. More...
|
| |
| FailureOr< VectorType > | mlir::xegpu::getDistributedVectorType (VectorType originalType, LayoutAttr layout) |
| | Helper to get the distributed vector type for a given vector type according to a given LayoutAttr. More...
|
| |
| std::string | mlir::xegpu::getLayoutName (const OpOperand &operand) |
| | Return the attribute name for the OpOperand to attach LayoutAttr. More...
|
| |
| std::string | mlir::xegpu::getLayoutName (const OpResult result) |
| | Return the attribute name for the OpResult to attach LayoutAttr. More...
|
| |
| LayoutAttr | mlir::xegpu::getLayoutAttr (const Value value) |
| | Retrieves the LayoutAttr associated with a given Value. More...
|
| |
| LayoutAttr | mlir::xegpu::getLayoutAttr (const OpOperand &opr) |
| | Retrieves the LayoutAttr associated with a given OpOperand. More...
|
| |
| template<typename T , typename = std::enable_if_t<std::is_same_v<T, OpOperand> || std::is_same_v<T, OpResult>>> |
| void | mlir::xegpu::setLayoutAttr (const T &operandOrResult, const LayoutAttr layout) |
| | Sets the LayoutAttr for a given OpOperand or OpResult by attaching it to the owner's dictionary attributes. More...
|
| |
| void | mlir::xegpu::setLayoutAttrs (Operation *op, function_ref< LayoutAttr(Value)> getLayoutImpl) |
| | Set the LayoutAttr for each OpOperand and OpResult of the given operation. More...
|
| |
| SmallVector< Value > | mlir::xegpu::extractVectorsWithShapeFromValue (OpBuilder &builder, Location loc, Value value, ArrayRef< int64_t > shape) |
| | Extract a set of small vectors from a value with a given shape using vector.extract_stride_slice. More...
|
| |
| Value | mlir::xegpu::createVectorWithShapeFromValues (OpBuilder &builder, Location loc, ValueRange values, ArrayRef< int64_t > shape) |
| | Create a vector of shape from a set of values using vector.insert_stride_slice. More...
|
| |
| void | mlir::xegpu::doSCFStructuralTypeConversionWithTensorType (Operation *op, TypeConverter converter) |
| | Do type conversion for SCF structural ops, e.g., scf.for using SCF structure type convertion patterns. More...
|
| |
