#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/IR/Matchers.h"
#include "mlir/Interfaces/InferIntRangeInterface.h"
#include "llvm/ADT/STLForwardCompat.h"
#include "llvm/Support/ErrorHandling.h"
#include "llvm/Support/MathExtras.h"
#include <optional>
Go to the source code of this file.
|
static ConstantIntRanges | getIndexRange (uint64_t umin, uint64_t umax) |
|
static Value | valueByDim (KernelDim3 dims, Dimension dim) |
| If the operation op is in a context that is annotated with maximum launch dimensions (a launch op with constant block or grid sizes or a launch_func op with the appropriate dimensions), return the bound on the maximum size of the dimension that the op is querying. More...
|
|
static uint64_t | zext (uint32_t arg) |
|
template<typename Op > |
static std::optional< uint64_t > | getKnownLaunchDim (Op op, LaunchDims type) |
|
◆ getIndexRange()
◆ getKnownLaunchDim()
template<typename Op >
static std::optional<uint64_t> getKnownLaunchDim |
( |
Op |
op, |
|
|
LaunchDims |
type |
|
) |
| |
|
static |
◆ valueByDim()
◆ zext()
static uint64_t zext |
( |
uint32_t |
arg | ) |
|
|
static |
◆ kMaxClusterDim
constexpr uint64_t kMaxClusterDim = 8 |
|
staticconstexpr |
◆ kMaxDim
constexpr uint64_t kMaxDim = std::numeric_limits<uint32_t>::max() |
|
staticconstexpr |
◆ kMaxSubgroupSize
constexpr uint64_t kMaxSubgroupSize = 128 |
|
staticconstexpr |