SparseTensorDescriptor.cpp File Reference

#include "SparseTensorDescriptor.h"
#include "CodegenUtils.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.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>. More...
static std::optional< LogicalResult >	convertSparseTensorType (RankedTensorType rtp, SmallVectorImpl< Type > &fields)

Function Documentation

convertSparseTensorType()

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

static

Definition at line 36 of file SparseTensorDescriptor.cpp.

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

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

optionalLevelAttr()

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

static

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

Definition at line 26 of file SparseTensorDescriptor.cpp.

References mlir::get().