mlir::amdgpu Namespace Reference

Namespaces
namespace	impl

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

Functions
ParseResult	parseMNKDimensionList (OpAsmParser &parser, IntegerAttr &m, IntegerAttr &n, IntegerAttr &k)
	Parser for the custom<MNKDimensionList> custom assembly format used by WMMAOp.
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.
void	printMNKDimensionList (OpAsmPrinter &printer, Operation *, IntegerAttr m, IntegerAttr n, IntegerAttr k)
std::unique_ptr<::mlir::Pass >	createAmdgpuEmulateAtomicsPass ()
std::unique_ptr<::mlir::Pass >	createAmdgpuEmulateAtomicsPass (AmdgpuEmulateAtomicsPassOptions options)
std::unique_ptr<::mlir::Pass >	createAmdgpuFoldMemRefOpsPass ()
std::unique_ptr<::mlir::Pass >	createAmdgpuMaskedloadToLoadPass ()
std::unique_ptr<::mlir::Pass >	createAmdgpuResolveStridedMetadataPass ()
void	registerAmdgpuEmulateAtomicsPass ()
void	registerAmdgpuEmulateAtomicsPassPass ()
void	registerAmdgpuFoldMemRefOpsPass ()
void	registerAmdgpuFoldMemRefOpsPassPass ()
void	registerAmdgpuMaskedloadToLoadPass ()
void	registerAmdgpuMaskedloadToLoadPassPass ()
void	registerAmdgpuResolveStridedMetadataPass ()
void	registerAmdgpuResolveStridedMetadataPassPass ()
void	registerAMDGPUPasses ()
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

createAmdgpuEmulateAtomicsPass() [1/2]

std::unique_ptr<::mlir::Pass > mlir::amdgpu::createAmdgpuEmulateAtomicsPass

(

)

Definition at line 100 of file EmulateAtomics.cpp.

createAmdgpuEmulateAtomicsPass() [2/2]

std::unique_ptr<::mlir::Pass > mlir::amdgpu::createAmdgpuEmulateAtomicsPass

(

AmdgpuEmulateAtomicsPassOptions

options

)

Definition at line 104 of file EmulateAtomics.cpp.

createAmdgpuFoldMemRefOpsPass()

std::unique_ptr<::mlir::Pass > mlir::amdgpu::createAmdgpuFoldMemRefOpsPass

(

)

We declare an explicit private instantiation because Pass classes should only be visible by the current library.

Definition at line 179 of file FoldMemRefsOps.cpp.

createAmdgpuMaskedloadToLoadPass()

std::unique_ptr<::mlir::Pass > mlir::amdgpu::createAmdgpuMaskedloadToLoadPass

(

)

We declare an explicit private instantiation because Pass classes should only be visible by the current library.

Definition at line 255 of file MaskedloadToLoad.cpp.

createAmdgpuResolveStridedMetadataPass()

std::unique_ptr<::mlir::Pass > mlir::amdgpu::createAmdgpuResolveStridedMetadataPass

(

)

We declare an explicit private instantiation because Pass classes should only be visible by the current library.

Definition at line 331 of file ResolveStridedMetadata.cpp.

foldMemrefViewOp()

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::Value::getDefiningOp(), indices, mlir::RewriterBase::notifyMatchFailure(), mlir::affine::resolveIndicesIntoOpWithOffsetsAndStrides(), mlir::memref::resolveSourceIndicesCollapseShape(), mlir::memref::resolveSourceIndicesExpandShape(), and success().

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 390 of file AMDGPUDialect.cpp.

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

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::getElementTypeOrSelf(), mlir::amdgpu::Chipset::majorVersion, mlir::patterns, and target.

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.

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().

registerAmdgpuEmulateAtomicsPass()

void mlir::amdgpu::registerAmdgpuEmulateAtomicsPass ( )

inline

Definition at line 349 of file Passes.h.

registerAmdgpuEmulateAtomicsPassPass()

void mlir::amdgpu::registerAmdgpuEmulateAtomicsPassPass ( )

inline

Definition at line 356 of file Passes.h.

registerAmdgpuFoldMemRefOpsPass()

void mlir::amdgpu::registerAmdgpuFoldMemRefOpsPass ( )

inline

Definition at line 370 of file Passes.h.

registerAmdgpuFoldMemRefOpsPassPass()

void mlir::amdgpu::registerAmdgpuFoldMemRefOpsPassPass ( )

inline

Definition at line 377 of file Passes.h.

registerAmdgpuMaskedloadToLoadPass()

void mlir::amdgpu::registerAmdgpuMaskedloadToLoadPass ( )

inline

Definition at line 391 of file Passes.h.

registerAmdgpuMaskedloadToLoadPassPass()

void mlir::amdgpu::registerAmdgpuMaskedloadToLoadPassPass ( )

inline

Definition at line 398 of file Passes.h.

registerAMDGPUPasses()

void mlir::amdgpu::registerAMDGPUPasses ( )

inline

Definition at line 433 of file Passes.h.

Referenced by mlir::registerAllPasses().

registerAmdgpuResolveStridedMetadataPass()

void mlir::amdgpu::registerAmdgpuResolveStridedMetadataPass ( )

inline

Definition at line 412 of file Passes.h.

registerAmdgpuResolveStridedMetadataPassPass()

void mlir::amdgpu::registerAmdgpuResolveStridedMetadataPassPass ( )

inline

Definition at line 419 of file Passes.h.