mlir::xegpu::uArch::MMAInstructionInterface Struct Reference

#include "mlir/Dialect/XeGPU/uArch/uArchBase.h"

Inheritance diagram for mlir::xegpu::uArch::MMAInstructionInterface:

Public Member Functions
virtual llvm::SmallVector< std::pair< uint32_t, uint32_t >, 16 >	getSupportedShapes (Type dataType, MMAOpndKind matrixType)=0
virtual llvm::SmallVector< Type, 8 >	getSupportedTypes (MLIRContext &context, MMAOpndKind matrixType)=0
virtual bool	checkSupportedShapesAndTypes (std::pair< uint32_t, uint32_t > AShape, std::pair< uint32_t, uint32_t > BShape, std::pair< uint32_t, uint32_t > CShape, std::pair< uint32_t, uint32_t > DShape, Type AType, Type BType, Type CType, Type DType)=0
virtual bool	checkSupportedTypes (Type AType, Type BType, Type CType, Type DType)=0
virtual bool	validate (std::pair< uint32_t, uint32_t > AShape, std::pair< uint32_t, uint32_t > BShape, std::pair< uint32_t, uint32_t > CShape, std::pair< uint32_t, uint32_t > DShape, Type AType, Type BType, Type CType, Type DType)=0
virtual llvm::SmallVector< uint32_t, 8 >	getSupportedM (Type type)=0
virtual llvm::SmallVector< uint32_t, 8 >	getSupportedK (Type type)=0
virtual llvm::SmallVector< uint32_t, 8 >	getSupportedN (Type type)=0
virtual	~MMAInstructionInterface ()=default

Detailed Description

Definition at line 222 of file uArchBase.h.

Constructor & Destructor Documentation

~MMAInstructionInterface()

virtual mlir::xegpu::uArch::MMAInstructionInterface::~MMAInstructionInterface ( )

virtualdefault

Member Function Documentation

checkSupportedShapesAndTypes()

virtual bool mlir::xegpu::uArch::MMAInstructionInterface::checkSupportedShapesAndTypes ( std::pair< uint32_t, uint32_t >  AShape, std::pair< uint32_t, uint32_t >  BShape, std::pair< uint32_t, uint32_t >  CShape, std::pair< uint32_t, uint32_t >  DShape, Type  AType, Type  BType, Type  CType, Type  DType  )

pure virtual

Implemented in mlir::xegpu::uArch::DPASInstruction.

checkSupportedTypes()

virtual bool mlir::xegpu::uArch::MMAInstructionInterface::checkSupportedTypes ( Type  AType, Type  BType, Type  CType, Type  DType  )

pure virtual

Implemented in mlir::xegpu::uArch::DPASInstruction.

getSupportedK()

virtual llvm::SmallVector<uint32_t, 8> mlir::xegpu::uArch::MMAInstructionInterface::getSupportedK ( Type  type )

pure virtual

Implemented in mlir::xegpu::uArch::DPASInstruction.

getSupportedM()

virtual llvm::SmallVector<uint32_t, 8> mlir::xegpu::uArch::MMAInstructionInterface::getSupportedM ( Type  type )

pure virtual

Implemented in mlir::xegpu::uArch::DPASInstruction.

getSupportedN()

virtual llvm::SmallVector<uint32_t, 8> mlir::xegpu::uArch::MMAInstructionInterface::getSupportedN ( Type  type )

pure virtual

Implemented in mlir::xegpu::uArch::DPASInstruction.

getSupportedShapes()

virtual llvm::SmallVector<std::pair<uint32_t, uint32_t>, 16> mlir::xegpu::uArch::MMAInstructionInterface::getSupportedShapes ( Type  dataType, MMAOpndKind  matrixType  )

pure virtual

Implemented in mlir::xegpu::uArch::DPASInstruction.

getSupportedTypes()

virtual llvm::SmallVector<Type, 8> mlir::xegpu::uArch::MMAInstructionInterface::getSupportedTypes ( MLIRContext &  context, MMAOpndKind  matrixType  )

pure virtual

Implemented in mlir::xegpu::uArch::DPASInstruction.

validate()

virtual bool mlir::xegpu::uArch::MMAInstructionInterface::validate ( std::pair< uint32_t, uint32_t >  AShape, std::pair< uint32_t, uint32_t >  BShape, std::pair< uint32_t, uint32_t >  CShape, std::pair< uint32_t, uint32_t >  DShape, Type  AType, Type  BType, Type  CType, Type  DType  )

pure virtual

Implemented in mlir::xegpu::uArch::DPASInstruction.

---

The documentation for this struct was generated from the following file:

• include/mlir/Dialect/XeGPU/uArch/uArchBase.h