MLIR  16.0.0git
Functions
SparseTensorRewriting.cpp File Reference
#include "CodegenUtils.h"
#include "mlir/Dialect/Arithmetic/IR/Arithmetic.h"
#include "mlir/Dialect/Bufferization/IR/Bufferization.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/SparseTensor/IR/SparseTensor.h"
#include "mlir/Dialect/SparseTensor/Transforms/Passes.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/IR/AffineMap.h"
#include "mlir/IR/Matchers.h"
#include "mlir/Support/LLVM.h"
+ Include dependency graph for SparseTensorRewriting.cpp:

Go to the source code of this file.

Functions

static bool isSparseTensor (OpOperand *op)
 
static bool isAlloc (OpOperand *op, bool isZero)
 
static bool isSampling (GenericOp op)
 
static bool isMulChain (Value val, Value x)
 
static bool isSumOfMul (GenericOp op)
 
static bool isZeroYield (GenericOp op)
 

Function Documentation

◆ isAlloc()

static bool isAlloc ( OpOperand op,
bool  isZero 
)
static

◆ isMulChain()

static bool isMulChain ( Value  val,
Value  x 
)
static

Definition at line 75 of file SparseTensorRewriting.cpp.

References mlir::Value::dyn_cast(), and mlir::Value::getDefiningOp().

Referenced by isSumOfMul().

◆ isSampling()

static bool isSampling ( GenericOp  op)
static

Definition at line 60 of file SparseTensorRewriting.cpp.

Referenced by isZeroYield().

◆ isSparseTensor()

static bool isSparseTensor ( OpOperand op)
static

◆ isSumOfMul()

static bool isSumOfMul ( GenericOp  op)
static

Definition at line 87 of file SparseTensorRewriting.cpp.

References isMulChain().

Referenced by isZeroYield().

◆ isZeroYield()

static bool isZeroYield ( GenericOp  op)
static