InferIntRangeInterfaceImpls.cpp File Reference

#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/IR/Matchers.h"
#include "mlir/Interfaces/FunctionInterfaces.h"
#include "mlir/Interfaces/InferIntRangeInterface.h"
#include <optional>

Go to the source code of this file.

Functions
static ConstantIntRanges	getIndexRange (uint64_t umin, uint64_t umax)
static uint64_t	zext (uint32_t arg)
static Value	valueByDim (KernelDim3 dims, Dimension dim)
static std::optional< uint32_t >	getKnownLaunchAttr (GPUFuncOp func, DimensionKind dims, Dimension dim)
static std::optional< uint32_t >	getKnownLaunchAttr (FunctionOpInterface func, StringRef attrName, Dimension dim)

Variables
static constexpr uint64_t	kMaxDim = std::numeric_limits<uint32_t>::max()
static constexpr uint64_t	kMaxClusterDim = 16
static constexpr uint64_t	kMaxSubgroupSize = 128

Function Documentation

getIndexRange()

ConstantIntRanges getIndexRange ( uint64_t umin, uint64_t umax )

static

Definition at line 25 of file InferIntRangeInterfaceImpls.cpp.

References mlir::ConstantIntRanges::fromUnsigned().

getKnownLaunchAttr() [1/2]

std::optional< uint32_t > getKnownLaunchAttr ( FunctionOpInterface func, StringRef attrName, Dimension dim )

static

Definition at line 68 of file InferIntRangeInterfaceImpls.cpp.

getKnownLaunchAttr() [2/2]

std::optional< uint32_t > getKnownLaunchAttr ( GPUFuncOp func, DimensionKind dims, Dimension dim )

static

Definition at line 46 of file InferIntRangeInterfaceImpls.cpp.

Referenced by mlir::gpu::getKnownDimensionSizeAround().

valueByDim()

Value valueByDim ( KernelDim3 dims, Dimension dim )

static

Definition at line 33 of file InferIntRangeInterfaceImpls.cpp.

References mlir::gpu::KernelDim3::x, mlir::gpu::KernelDim3::y, and mlir::gpu::KernelDim3::z.

Referenced by mlir::gpu::getKnownDimensionSizeAround().

zext()

uint64_t zext ( uint32_t arg )

static

Definition at line 31 of file InferIntRangeInterfaceImpls.cpp.

Referenced by CanonicalizeContractMatmulToMMT::matchAndRewrite().

Variable Documentation

kMaxClusterDim

uint64_t kMaxClusterDim = 16

staticconstexpr

Definition at line 21 of file InferIntRangeInterfaceImpls.cpp.

kMaxDim

uint64_t kMaxDim = std::numeric_limits<uint32_t>::max()

staticconstexpr

Definition at line 19 of file InferIntRangeInterfaceImpls.cpp.

kMaxSubgroupSize

uint64_t kMaxSubgroupSize = 128

staticconstexpr

Definition at line 23 of file InferIntRangeInterfaceImpls.cpp.