mlir::xegpu::uArch::DPASInstruction Struct Reference

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

Inheritance diagram for mlir::xegpu::uArch::DPASInstruction:

Public Member Functions
	DPASInstruction ()
virtual llvm::SmallVector< std::pair< uint32_t, uint32_t >, 16 >	getSupportedShapes (Type dataType, MMAOpndKind matrixType) override
virtual llvm::SmallVector< Type, 8 >	getSupportedTypes (MLIRContext &context, MMAOpndKind matrixType) override
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) override
virtual bool	checkSupportedTypes (Type AType, Type BType, Type CType, Type DType) override
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) override
virtual llvm::SmallVector< uint32_t, 8 >	getSupportedM (Type type) override
virtual llvm::SmallVector< uint32_t, 8 >	getSupportedK (Type type) override
virtual llvm::SmallVector< uint32_t, 8 >	getSupportedN (Type type) override
Public Member Functions inherited from mlir::xegpu::uArch::Instruction
	Instruction (InstructionKind kind, InstructionScope scope)
virtual	~Instruction ()=default
InstructionKind	getInstructionKind ()
InstructionScope	getScope ()
Public Member Functions inherited from mlir::xegpu::uArch::MMAInstructionInterface
virtual	~MMAInstructionInterface ()=default

Additional Inherited Members
Static Public Member Functions inherited from mlir::xegpu::uArch::Instruction
static llvm::StringRef	toString (InstructionKind instKind)
static std::optional< InstructionKind >	parseInstructionKind (llvm::StringRef str)
Protected Attributes inherited from mlir::xegpu::uArch::Instruction
InstructionKind	instKind
InstructionScope	scope

Detailed Description

Definition at line 48 of file IntelGpuXe2.h.

Constructor & Destructor Documentation

DPASInstruction()

mlir::xegpu::uArch::DPASInstruction::DPASInstruction ( )

inline

Definition at line 49 of file IntelGpuXe2.h.

References mlir::xegpu::uArch::DPAS, and mlir::xegpu::uArch::Subgroup.

Member Function Documentation

checkSupportedShapesAndTypes()

bool DPASInstruction::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  )

inlineoverridevirtual

Implements mlir::xegpu::uArch::MMAInstructionInterface.

Definition at line 234 of file IntelGpuXe2.h.

References checkSupportedTypes(), getSupportedShapes(), mlir::xegpu::uArch::MatrixA, mlir::xegpu::uArch::MatrixB, mlir::xegpu::uArch::MatrixC, and mlir::xegpu::uArch::MatrixD.

Referenced by validate().

checkSupportedTypes()

bool DPASInstruction::checkSupportedTypes ( Type  AType, Type  BType, Type  CType, Type  DType  )

inlineoverridevirtual

Implements mlir::xegpu::uArch::MMAInstructionInterface.

Definition at line 203 of file IntelGpuXe2.h.

References mlir::Type::isBF16(), mlir::Type::isF16(), mlir::Type::isF32(), mlir::Type::isInteger(), and mlir::Type::isTF32().

Referenced by checkSupportedShapesAndTypes().

getSupportedK()

llvm::SmallVector< uint32_t, 8 > DPASInstruction::getSupportedK ( Type  type )

inlineoverridevirtual

Implements mlir::xegpu::uArch::MMAInstructionInterface.

Definition at line 265 of file IntelGpuXe2.h.

References mlir::Type::getIntOrFloatBitWidth(), and mlir::Type::isIntOrFloat().

Referenced by getSupportedShapes().

getSupportedM()

llvm::SmallVector< uint32_t, 8 > DPASInstruction::getSupportedM ( Type  type )

inlineoverridevirtual

Implements mlir::xegpu::uArch::MMAInstructionInterface.

Definition at line 260 of file IntelGpuXe2.h.

Referenced by getSupportedShapes().

getSupportedN()

llvm::SmallVector< uint32_t, 8 > DPASInstruction::getSupportedN ( Type  type )

inlineoverridevirtual

Implements mlir::xegpu::uArch::MMAInstructionInterface.

Definition at line 293 of file IntelGpuXe2.h.

Referenced by getSupportedShapes().

getSupportedShapes()

llvm::SmallVector< std::pair< uint32_t, uint32_t >, 16 > DPASInstruction::getSupportedShapes ( Type  dataType, MMAOpndKind  matrixType  )

inlineoverridevirtual

Implements mlir::xegpu::uArch::MMAInstructionInterface.

Definition at line 147 of file IntelGpuXe2.h.

References getSupportedK(), getSupportedM(), getSupportedN(), mlir::xegpu::uArch::MatrixA, mlir::xegpu::uArch::MatrixB, mlir::xegpu::uArch::MatrixC, and mlir::xegpu::uArch::MatrixD.

Referenced by checkSupportedShapesAndTypes().

getSupportedTypes()

llvm::SmallVector< Type, 8 > DPASInstruction::getSupportedTypes ( MLIRContext &  context, MMAOpndKind  matrixType  )

inlineoverridevirtual

Implements mlir::xegpu::uArch::MMAInstructionInterface.

Definition at line 183 of file IntelGpuXe2.h.

References mlir::get(), mlir::xegpu::uArch::MatrixA, mlir::xegpu::uArch::MatrixB, mlir::xegpu::uArch::MatrixC, and mlir::xegpu::uArch::MatrixD.

validate()

bool DPASInstruction::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  )

inlineoverridevirtual

Implements mlir::xegpu::uArch::MMAInstructionInterface.

Definition at line 249 of file IntelGpuXe2.h.

References checkSupportedShapesAndTypes().

---

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

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