XeGPUUtils.cpp File Reference

#include "mlir/Dialect/XeGPU/Utils/XeGPUUtils.h"
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/Dialect/Index/IR/IndexOps.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 void	xegpu::setDistributeLayoutAttr< mlir::OpResult > (const mlir::OpResult &result, const mlir::xegpu::DistributeLayoutAttr layout, bool respectPermLayout)
template void	xegpu::setDistributeLayoutAttr< mlir::OpOperand > (const mlir::OpOperand &operand, const mlir::xegpu::DistributeLayoutAttr layout, bool respectPermLayout)
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 207 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 191 of file XeGPUUtils.cpp.

References result.

Referenced by mlir::xegpu::setDistributeLayoutAttr().

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::setDistributeLayoutAttr< mlir::OpOperand >()

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

References mlir::xegpu::setDistributeLayoutAttr().

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

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

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