MLIR
22.0.0git
|
#include "mlir/Dialect/AMDGPU/Transforms/Passes.h"
#include "mlir/Dialect/AMDGPU/IR/AMDGPUDialect.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/Dialect/MemRef/Utils/MemRefUtils.h"
#include "mlir/Dialect/SCF/IR/SCF.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/Dialect/Vector/Transforms/VectorTransforms.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/OpDefinition.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/IR/TypeUtilities.h"
#include "mlir/Support/LogicalResult.h"
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
#include "llvm/Support/MathExtras.h"
#include "mlir/Dialect/AMDGPU/Transforms/Passes.h.inc"
Go to the source code of this file.
Classes | |
struct | AmdgpuMaskedloadToLoadPass |
Namespaces | |
mlir | |
Include the generated interface declarations. | |
mlir::amdgpu | |
Macros | |
#define | GEN_PASS_DEF_AMDGPUMASKEDLOADTOLOADPASS |
Functions | |
static LogicalResult | baseInBufferAddrSpace (PatternRewriter &rewriter, vector::MaskedLoadOp maskedOp) |
This pattern supports lowering of: vector.maskedload to vector.load and arith.select if the memref is in buffer address space. More... | |
static Value | createVectorLoadForMaskedLoad (OpBuilder &builder, Location loc, vector::MaskedLoadOp maskedOp, bool passthru) |
static FailureOr< Value > | matchFullMask (OpBuilder &b, Value val) |
Check if the given value comes from a broadcasted i1 condition. More... | |
Variables | |
static constexpr char | kMaskedloadNeedsMask [] |
#define GEN_PASS_DEF_AMDGPUMASKEDLOADTOLOADPASS |
Definition at line 27 of file MaskedloadToLoad.cpp.
|
static |
This pattern supports lowering of: vector.maskedload
to vector.load
and arith.select
if the memref is in buffer address space.
Definition at line 36 of file MaskedloadToLoad.cpp.
References mlir::RewriterBase::notifyMatchFailure().
|
static |
Definition at line 53 of file MaskedloadToLoad.cpp.
Check if the given value comes from a broadcasted i1 condition.
Definition at line 66 of file MaskedloadToLoad.cpp.
References mlir::Value::getDefiningOp().
|
staticconstexpr |
Definition at line 75 of file MaskedloadToLoad.cpp.