mlir::amdgpu Namespace Reference

Classes
struct	Chipset
	Represents the amdgpu gfx chipset version, e.g., gfx90a, gfx942, gfx1103. More...
struct	AmdgpuFoldMemRefOpsPass
struct	FoldMemRefOpsIntoGatherToLDSOp

Functions
ParseResult	parseMNKDimensionList (OpAsmParser &parser, IntegerAttr &m, IntegerAttr &n, IntegerAttr &k)
	Parser for the custom<MNKDimensionList> custom assembly format used by WMMAOp. More...
ParseResult	parseMNKDimensionList (OpAsmParser &parser, Operation *, IntegerAttr &m, IntegerAttr &n, IntegerAttr &k)
void	printMNKDimensionList (OpAsmPrinter &printer, IntegerAttr m, IntegerAttr n, IntegerAttr k)
	Printer for the custom<MNKDimensionList> custom assembly format used by WMMAOp. More...
void	printMNKDimensionList (OpAsmPrinter &printer, Operation *, IntegerAttr m, IntegerAttr n, IntegerAttr k)
void	populateAmdgpuEmulateAtomicsPatterns (ConversionTarget &target, RewritePatternSet &patterns, Chipset chipset, PatternBenefit benefit=1)
void	populateAmdgpuResolveStridedMetadataPatterns (RewritePatternSet &patterns, PatternBenefit benefit=1)
void	populateAmdgpuMaskedloadToLoadPatterns (RewritePatternSet &patterns, PatternBenefit benefit=1)
void	populateAmdgpuFoldMemRefOpsPatterns (RewritePatternSet &patterns, PatternBenefit benefit=1)
bool	hasOcpFp8 (const Chipset &chipset)
static LogicalResult	foldMemrefViewOp (PatternRewriter &rewriter, Location loc, Value view, mlir::OperandRange indices, SmallVectorImpl< Value > &resolvedIndices, Value &memrefBase, StringRef role)

Function Documentation

foldMemrefViewOp()

static LogicalResult mlir::amdgpu::foldMemrefViewOp ( PatternRewriter &  rewriter, Location  loc, Value  view, mlir::OperandRange  indices, SmallVectorImpl< Value > &  resolvedIndices, Value &  memrefBase, StringRef  role  )

static

Definition at line 31 of file FoldMemRefsOps.cpp.

References mlir::remark::failed(), mlir::Value::getDefiningOp(), mlir::RewriterBase::notifyMatchFailure(), mlir::affine::resolveIndicesIntoOpWithOffsetsAndStrides(), mlir::memref::resolveSourceIndicesCollapseShape(), and mlir::memref::resolveSourceIndicesExpandShape().

Referenced by mlir::amdgpu::FoldMemRefOpsIntoGatherToLDSOp::matchAndRewrite().

hasOcpFp8()

bool mlir::amdgpu::hasOcpFp8 ( const Chipset &  chipset )

inline

Definition at line 52 of file Chipset.h.

References mlir::amdgpu::Chipset::majorVersion, and mlir::amdgpu::Chipset::minorVersion.

Referenced by isSupportedF8(), typeIsExpectedBf8ForChipset(), and typeIsExpectedFp8ForChipset().

parseMNKDimensionList() [1/2]

ParseResult mlir::amdgpu::parseMNKDimensionList	(	OpAsmParser &	parser,
		IntegerAttr &	m,
		IntegerAttr &	n,
		IntegerAttr &	k
	)

Parser for the custom<MNKDimensionList> custom assembly format used by WMMAOp.

Definition at line 364 of file AMDGPUDialect.cpp.

References mlir::AsmParser::emitError(), mlir::AsmParser::getBuilder(), mlir::AsmParser::getCurrentLocation(), mlir::Builder::getI32IntegerAttr(), and mlir::AsmParser::parseDimensionList().

Referenced by parseMNKDimensionList().

parseMNKDimensionList() [2/2]

ParseResult mlir::amdgpu::parseMNKDimensionList ( OpAsmParser &  parser, Operation *  , IntegerAttr &  m, IntegerAttr &  n, IntegerAttr &  k  )

inline

Definition at line 34 of file AMDGPUDialect.h.

References parseMNKDimensionList().

populateAmdgpuEmulateAtomicsPatterns()

void mlir::amdgpu::populateAmdgpuEmulateAtomicsPatterns	(	ConversionTarget &	target,
		RewritePatternSet &	patterns,
		Chipset	chipset,
		PatternBenefit	benefit = 1
	)

Definition at line 166 of file EmulateAtomics.cpp.

References mlir::ConversionTarget::addDynamicallyLegalOp(), mlir::ConversionTarget::addIllegalOp(), mlir::getElementTypeOrSelf(), mlir::amdgpu::Chipset::majorVersion, and mlir::patterns.

populateAmdgpuFoldMemRefOpsPatterns()

void mlir::amdgpu::populateAmdgpuFoldMemRefOpsPatterns	(	RewritePatternSet &	patterns,
		PatternBenefit	benefit = 1
	)

Definition at line 110 of file FoldMemRefsOps.cpp.

References mlir::patterns.

Referenced by mlir::amdgpu::AmdgpuFoldMemRefOpsPass::runOnOperation().

populateAmdgpuMaskedloadToLoadPatterns()

void mlir::amdgpu::populateAmdgpuMaskedloadToLoadPatterns	(	RewritePatternSet &	patterns,
		PatternBenefit	benefit = 1
	)

Definition at line 226 of file MaskedloadToLoad.cpp.

References mlir::patterns.

Referenced by AmdgpuMaskedloadToLoadPass::runOnOperation().

populateAmdgpuResolveStridedMetadataPatterns()

void mlir::amdgpu::populateAmdgpuResolveStridedMetadataPatterns	(	RewritePatternSet &	patterns,
		PatternBenefit	benefit = 1
	)

Definition at line 68 of file ResolveStridedMetadata.cpp.

References mlir::patterns.

printMNKDimensionList() [1/2]

void mlir::amdgpu::printMNKDimensionList ( OpAsmPrinter &  printer, IntegerAttr  m, IntegerAttr  n, IntegerAttr  k  )

inline

Printer for the custom<MNKDimensionList> custom assembly format used by WMMAOp.

Definition at line 42 of file AMDGPUDialect.h.

References mlir::AsmPrinter::printDimensionList().

Referenced by printMNKDimensionList().

printMNKDimensionList() [2/2]

void mlir::amdgpu::printMNKDimensionList ( OpAsmPrinter &  printer, Operation *  , IntegerAttr  m, IntegerAttr  n, IntegerAttr  k  )

inline

Definition at line 46 of file AMDGPUDialect.h.

References printMNKDimensionList().