mlir::xegpu::uArch::SubgroupMatrixMultiplyAcc Struct Reference

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

Inheritance diagram for mlir::xegpu::uArch::SubgroupMatrixMultiplyAcc:

Public Member Functions
	SubgroupMatrixMultiplyAcc (unsigned packedFormatBitSizeA, unsigned packedFormatBitSizeB)
llvm::SmallVector< std::pair< uint32_t, uint32_t >, 16 >	getSupportedShapes (Type dataType, MMAOpndKind matrixType) override
llvm::SmallVector< Type, 8 >	getSupportedTypes (MLIRContext &context, MMAOpndKind matrixType) override
llvm::SmallVector< uint32_t, 8 >	getSupportedM (Type type) const override
llvm::SmallVector< uint32_t, 8 >	getSupportedK (Type type) const override
llvm::SmallVector< uint32_t, 8 >	getSupportedN (Type type) const override
unsigned	getPackedFormatBitSizeA () const
unsigned	getPackedFormatBitSizeB () const
bool	isLaneLayoutRowMajorOrder () const override
Public Member Functions inherited from mlir::xegpu::uArch::Instruction
	Instruction (InstructionKind kind, InstructionScope scope)
	~Instruction ()=default
InstructionKind	getInstructionKind () const
InstructionScope	getScope () const
Public Member Functions inherited from mlir::xegpu::uArch::MMAInstructionInterface
virtual	~MMAInstructionInterface ()=default

Static Public Member Functions
static bool	classof (const Instruction *B)
Static Public Member Functions inherited from mlir::xegpu::uArch::Instruction
static llvm::StringRef	toString (InstructionKind instKind)

Protected Attributes
const unsigned	packedFormatBitSizeA
const unsigned	packedFormatBitSizeB
Protected Attributes inherited from mlir::xegpu::uArch::Instruction
const InstructionKind	instKind
const InstructionScope	scope

Detailed Description

Definition at line 372 of file uArchBase.h.

Constructor & Destructor Documentation

SubgroupMatrixMultiplyAcc()

mlir::xegpu::uArch::SubgroupMatrixMultiplyAcc::SubgroupMatrixMultiplyAcc ( unsigned packedFormatBitSizeA, unsigned packedFormatBitSizeB )

inline

Definition at line 374 of file uArchBase.h.

References mlir::xegpu::uArch::Instruction::Instruction(), packedFormatBitSizeA, packedFormatBitSizeB, mlir::xegpu::uArch::Subgroup, and SubgroupMatrixMultiplyAcc().

Referenced by SubgroupMatrixMultiplyAcc().

Member Function Documentation

classof()

bool mlir::xegpu::uArch::SubgroupMatrixMultiplyAcc::classof ( const Instruction * B )

inlinestatic

Definition at line 380 of file uArchBase.h.

References mlir::xegpu::uArch::Instruction::Instruction(), and mlir::xegpu::uArch::SubgroupMatrixMultiplyAcc.

getPackedFormatBitSizeA()

unsigned mlir::xegpu::uArch::SubgroupMatrixMultiplyAcc::getPackedFormatBitSizeA ( ) const

inline

Definition at line 396 of file uArchBase.h.

References packedFormatBitSizeA.

getPackedFormatBitSizeB()

unsigned mlir::xegpu::uArch::SubgroupMatrixMultiplyAcc::getPackedFormatBitSizeB ( ) const

inline

Definition at line 397 of file uArchBase.h.

References packedFormatBitSizeB.

getSupportedK()

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

inlineoverridevirtual

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

Definition at line 497 of file uArchBase.h.

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

Referenced by getSupportedShapes().

getSupportedM()

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

inlineoverridevirtual

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

Definition at line 492 of file uArchBase.h.

Referenced by getSupportedShapes().

getSupportedN()

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

inlineoverridevirtual

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

Definition at line 521 of file uArchBase.h.

Referenced by getSupportedShapes().

getSupportedShapes()

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

inlineoverridevirtual

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

Definition at line 455 of file uArchBase.h.

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

getSupportedTypes()

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

inlineoverridevirtual

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

Definition at line 473 of file uArchBase.h.

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

isLaneLayoutRowMajorOrder()

bool mlir::xegpu::uArch::SubgroupMatrixMultiplyAcc::isLaneLayoutRowMajorOrder ( ) const

inlineoverridevirtual

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

Definition at line 398 of file uArchBase.h.

Member Data Documentation

packedFormatBitSizeA

const unsigned mlir::xegpu::uArch::SubgroupMatrixMultiplyAcc::packedFormatBitSizeA

protected

Definition at line 401 of file uArchBase.h.

Referenced by getPackedFormatBitSizeA(), and SubgroupMatrixMultiplyAcc().

packedFormatBitSizeB

const unsigned mlir::xegpu::uArch::SubgroupMatrixMultiplyAcc::packedFormatBitSizeB

protected

Definition at line 402 of file uArchBase.h.

Referenced by getPackedFormatBitSizeB(), and SubgroupMatrixMultiplyAcc().

---

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

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