GPUUtils.h File Reference

#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/Support/LLVM.h"
#include <string>

Go to the source code of this file.

Classes
struct	mlir::gpu::SubgroupReduceClusterInfo
	Lane layout of the clusters reduced by a gpu.subgroup_reduce op. More...

Namespaces
namespace	mlir
	Include the generated interface declarations.
namespace	mlir::gpu

Functions
vector::CombiningKind	mlir::gpu::convertReductionKind (gpu::AllReduceOperation mode)
	Returns the matching vector combining kind.
FailureOr< SubgroupReduceClusterInfo >	mlir::gpu::getSubgroupReduceClusterInfo (SubgroupReduceOp op, unsigned subgroupSize)
	Returns the cluster layout of op in a subgroup of subgroupSize lanes, which must be a power of two.
gpu::GPUFuncOp	mlir::outlineKernelFunc (gpu::LaunchOp launchOp, StringRef kernelFnName, SmallVectorImpl< Value > &operands)
	Get a gpu.func created from outlining the region of a gpu.launch op with the given kernelFnName.
LogicalResult	mlir::sinkOperationsIntoLaunchOp (gpu::LaunchOp launchOp, llvm::function_ref< bool(Operation *)> isSinkingBeneficiary)
	Sink operations into the launchOp to reduce the number of values that are used within the region of the operation, but defined outside of the region.