XeGPUUtils.cpp File Reference

#include "mlir/Dialect/XeGPU/Utils/XeGPUUtils.h"
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/Dialect/LLVMIR/XeVMDialect.h"
#include "mlir/Dialect/SCF/Transforms/Patterns.h"
#include "mlir/Dialect/Utils/IndexingUtils.h"
#include "mlir/Dialect/XeGPU/IR/XeGPU.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/Operation.h"
#include "mlir/IR/ValueRange.h"
#include "mlir/Interfaces/LoopLikeInterface.h"
#include "mlir/Transforms/DialectConversion.h"
#include "llvm/Support/FormatVariadic.h"
#include <cstdint>
#include <numeric>

Go to the source code of this file.

Functions
xegpu::DistributeLayoutAttr	maybePickPermanentLayout (xegpu::DistributeLayoutAttr layout, const OpResult &result, mlir::Operation *owner, const std::string &name)
xegpu::DistributeLayoutAttr	maybePickPermanentLayout (xegpu::DistributeLayoutAttr layout, const OpOperand &operand, mlir::Operation *owner, const std::string &name)
template xegpu::DistributeLayoutAttr	xegpu::getTemporaryLayout< mlir::OpResult > (const OpResult &result)
template xegpu::DistributeLayoutAttr	xegpu::getTemporaryLayout< mlir::OpOperand > (const OpOperand &operand)
template void	xegpu::setTemporaryLayout< mlir::OpResult > (const mlir::OpResult &result, const mlir::xegpu::DistributeLayoutAttr layout)
template void	xegpu::setTemporaryLayout< mlir::OpOperand > (const mlir::OpOperand &operand, const mlir::xegpu::DistributeLayoutAttr layout)
template void	xegpu::removeLayoutAttr< mlir::OpResult > (const mlir::OpResult &result)
template void	xegpu::removeLayoutAttr< mlir::OpOperand > (const mlir::OpOperand &operand)

Function Documentation

maybePickPermanentLayout() [1/2]

xegpu::DistributeLayoutAttr maybePickPermanentLayout	(	xegpu::DistributeLayoutAttr	layout,
		const OpOperand &	operand,
		mlir::Operation *	owner,
		const std::string &	name )

Definition at line 212 of file XeGPUUtils.cpp.

maybePickPermanentLayout() [2/2]

xegpu::DistributeLayoutAttr maybePickPermanentLayout	(	xegpu::DistributeLayoutAttr	layout,
		const OpResult &	result,
		mlir::Operation *	owner,
		const std::string &	name )

Definition at line 196 of file XeGPUUtils.cpp.

References result.

xegpu::getTemporaryLayout< mlir::OpOperand >()

template xegpu::DistributeLayoutAttr xegpu::getTemporaryLayout< mlir::OpOperand >

(

const OpOperand &

operand

)

References mlir::xegpu::getTemporaryLayout().

xegpu::getTemporaryLayout< mlir::OpResult >()

template xegpu::DistributeLayoutAttr xegpu::getTemporaryLayout< mlir::OpResult >

(

const OpResult &

result

)

References mlir::xegpu::getTemporaryLayout(), and result.

xegpu::removeLayoutAttr< mlir::OpOperand >()

template void xegpu::removeLayoutAttr< mlir::OpOperand >

(

const mlir::OpOperand &

operand

)

References mlir::xegpu::removeLayoutAttr().

xegpu::removeLayoutAttr< mlir::OpResult >()

template void xegpu::removeLayoutAttr< mlir::OpResult >

(

const mlir::OpResult &

result

)

References mlir::xegpu::removeLayoutAttr(), and result.

xegpu::setTemporaryLayout< mlir::OpOperand >()

template void xegpu::setTemporaryLayout< mlir::OpOperand >	(	const mlir::OpOperand &	operand,
		const mlir::xegpu::DistributeLayoutAttr	layout )

References mlir::xegpu::setTemporaryLayout().

xegpu::setTemporaryLayout< mlir::OpResult >()

template void xegpu::setTemporaryLayout< mlir::OpResult >	(	const mlir::OpResult &	result,
		const mlir::xegpu::DistributeLayoutAttr	layout )

References result, and mlir::xegpu::setTemporaryLayout().