SparseTensorDescriptor.cpp File Reference

#include "SparseTensorDescriptor.h"
#include "CodegenUtils.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/Dialect/SparseTensor/IR/SparseTensor.h"
#include "mlir/Dialect/SparseTensor/Transforms/Passes.h"
#include "mlir/Transforms/DialectConversion.h"

Go to the source code of this file.

Functions
static IntegerAttr	optionalLevelAttr (MLIRContext *ctx, std::optional< Level > lvl)
	Constructs a nullable LevelAttr from the std::optional<Level>.
static std::optional< LogicalResult >	convertSparseTensorType (RankedTensorType rtp, SmallVectorImpl< Type > &fields)
static Value	materializeTuple (OpBuilder &builder, RankedTensorType tp, ValueRange inputs, Location loc)

Function Documentation

convertSparseTensorType()

std::optional< LogicalResult > convertSparseTensorType ( RankedTensorType rtp, SmallVectorImpl< Type > & fields )

static

Definition at line 35 of file SparseTensorDescriptor.cpp.

References mlir::sparse_tensor::foreachFieldAndTypeInSparseTensor(), mlir::sparse_tensor::SparseTensorType::hasEncoding(), and success().

Referenced by mlir::SparseTensorTypeToBufferConverter::SparseTensorTypeToBufferConverter().

materializeTuple()

Value materializeTuple ( OpBuilder & builder, RankedTensorType tp, ValueRange inputs, Location loc )

static

Definition at line 58 of file SparseTensorDescriptor.cpp.

References mlir::sparse_tensor::genTuple(), and mlir::sparse_tensor::getSparseTensorEncoding().

Referenced by mlir::SparseTensorTypeToBufferConverter::SparseTensorTypeToBufferConverter().

optionalLevelAttr()

IntegerAttr optionalLevelAttr ( MLIRContext * ctx, std::optional< Level > lvl )

static

Constructs a nullable LevelAttr from the std::optional<Level>.

Definition at line 25 of file SparseTensorDescriptor.cpp.

Referenced by mlir::sparse_tensor::SparseTensorSpecifier::getSpecifierField(), and mlir::sparse_tensor::SparseTensorSpecifier::setSpecifierField().