mlir::sparse_tensor::ir_detail::DimLvlExpr Class Reference

#include "Dialect/SparseTensor/IR/Detail/DimLvlMap.h"

Inheritance diagram for mlir::sparse_tensor::ir_detail::DimLvlExpr:

Public Member Functions
constexpr	DimLvlExpr (ExprKind ek, AffineExpr expr)
constexpr bool	operator== (DimLvlExpr other) const
constexpr bool	operator!= (DimLvlExpr other) const
	operator bool () const
template<typename U >
constexpr bool	isa () const
template<typename U >
constexpr U	cast () const
template<typename U >
constexpr U	dyn_cast () const
constexpr ExprKind	getExprKind () const
constexpr VarKind	getAllowedVarKind () const
constexpr AffineExpr	getAffineExpr () const
AffineExprKind	getAffineKind () const
MLIRContext *	tryGetContext () const
SymVar	castSymVar () const
std::optional< SymVar >	dyn_castSymVar () const
Var	castDimLvlVar () const
std::optional< Var >	dyn_castDimLvlVar () const
std::tuple< DimLvlExpr, AffineExprKind, DimLvlExpr >	unpackBinop () const
bool	isValid (Ranks const &ranks) const
	Checks whether the variables bound/used by this spec are valid with respect to the given ranks. More...

Protected Types
enum class	BindingStrength : bool { Weak = false , Strong = true }

Detailed Description

Definition at line 32 of file DimLvlMap.h.

Member Enumeration Documentation

BindingStrength

enum mlir::sparse_tensor::ir_detail::DimLvlExpr::BindingStrength : bool

strongprotected

Enumerator
Weak
Strong

Definition at line 92 of file DimLvlMap.h.

Constructor & Destructor Documentation

DimLvlExpr()

constexpr mlir::sparse_tensor::ir_detail::DimLvlExpr::DimLvlExpr ( ExprKind  ek, AffineExpr  expr  )

inlineconstexpr

Definition at line 38 of file DimLvlMap.h.

References kind.

Member Function Documentation

cast()

template<typename U >

constexpr U mlir::sparse_tensor::ir_detail::DimLvlExpr::cast

constexpr

Definition at line 143 of file DimLvlMap.h.

castDimLvlVar()

Var DimLvlExpr::castDimLvlVar

(

)

const

Definition at line 29 of file DimLvlMap.cpp.

References getAllowedVarKind().

Referenced by mlir::sparse_tensor::ir_detail::VarSet::add(), mlir::sparse_tensor::ir_detail::LvlExpr::castDimVar(), and mlir::sparse_tensor::ir_detail::DimExpr::castLvlVar().

castSymVar()

SymVar DimLvlExpr::castSymVar

(

)

const

Definition at line 19 of file DimLvlMap.cpp.

Referenced by mlir::sparse_tensor::ir_detail::VarSet::add().

dyn_cast()

template<typename U >

constexpr U mlir::sparse_tensor::ir_detail::DimLvlExpr::dyn_cast

constexpr

Definition at line 149 of file DimLvlMap.h.

dyn_castDimLvlVar()

std::optional< Var > DimLvlExpr::dyn_castDimLvlVar

(

)

const

Definition at line 33 of file DimLvlMap.cpp.

References getAllowedVarKind().

Referenced by mlir::sparse_tensor::ir_detail::LvlExpr::dyn_castDimVar(), and mlir::sparse_tensor::ir_detail::DimExpr::dyn_castLvlVar().

dyn_castSymVar()

std::optional< SymVar > DimLvlExpr::dyn_castSymVar

(

)

const

Definition at line 23 of file DimLvlMap.cpp.

getAffineExpr()

constexpr AffineExpr mlir::sparse_tensor::ir_detail::DimLvlExpr::getAffineExpr ( ) const

inlineconstexpr

Definition at line 68 of file DimLvlMap.h.

Referenced by mlir::sparse_tensor::ir_detail::Ranks::isValid().

getAffineKind()

AffineExprKind mlir::sparse_tensor::ir_detail::DimLvlExpr::getAffineKind ( ) const

inline

Definition at line 69 of file DimLvlMap.h.

References mlir::AffineExpr::getKind().

Referenced by mlir::sparse_tensor::ir_detail::VarSet::add(), and unpackBinop().

getAllowedVarKind()

constexpr VarKind mlir::sparse_tensor::ir_detail::DimLvlExpr::getAllowedVarKind ( ) const

inlineconstexpr

Definition at line 65 of file DimLvlMap.h.

References mlir::sparse_tensor::ir_detail::getVarKindAllowedInExpr(), and kind.

Referenced by castDimLvlVar(), dyn_castDimLvlVar(), and mlir::sparse_tensor::ir_detail::Ranks::isValid().

getExprKind()

constexpr ExprKind mlir::sparse_tensor::ir_detail::DimLvlExpr::getExprKind ( ) const

inlineconstexpr

Definition at line 64 of file DimLvlMap.h.

References kind.

Referenced by mlir::sparse_tensor::ir_detail::DimExpr::classof(), mlir::sparse_tensor::ir_detail::LvlExpr::classof(), and isa().

isa()

template<typename U >

constexpr bool mlir::sparse_tensor::ir_detail::DimLvlExpr::isa

constexpr

Definition at line 135 of file DimLvlMap.h.

References mlir::sparse_tensor::ir_detail::Dimension, getExprKind(), and mlir::sparse_tensor::ir_detail::Level.

isValid()

bool mlir::sparse_tensor::ir_detail::DimLvlExpr::isValid

(

Ranks const &

ranks

)

const

Checks whether the variables bound/used by this spec are valid with respect to the given ranks.

operator bool()

mlir::sparse_tensor::ir_detail::DimLvlExpr::operator bool ( ) const

inlineexplicit

Definition at line 49 of file DimLvlMap.h.

operator!=()

constexpr bool mlir::sparse_tensor::ir_detail::DimLvlExpr::operator!= ( DimLvlExpr  other ) const

inlineconstexpr

Definition at line 46 of file DimLvlMap.h.

operator==()

constexpr bool mlir::sparse_tensor::ir_detail::DimLvlExpr::operator== ( DimLvlExpr  other ) const

inlineconstexpr

Definition at line 43 of file DimLvlMap.h.

References kind.

tryGetContext()

MLIRContext* mlir::sparse_tensor::ir_detail::DimLvlExpr::tryGetContext ( ) const

inline

Definition at line 73 of file DimLvlMap.h.

References mlir::AffineExpr::getContext().

Referenced by mlir::sparse_tensor::ir_detail::LvlSpec::getContext(), and mlir::sparse_tensor::ir_detail::DimSpec::tryGetContext().

unpackBinop()

std::tuple< DimLvlExpr, AffineExprKind, DimLvlExpr > DimLvlExpr::unpackBinop

(

)

const

Definition at line 40 of file DimLvlMap.cpp.

References getAffineKind().

Referenced by mlir::sparse_tensor::ir_detail::VarSet::add().

---

The documentation for this class was generated from the following files:

• lib/Dialect/SparseTensor/IR/Detail/DimLvlMap.h
• lib/Dialect/SparseTensor/IR/Detail/DimLvlMap.cpp