Home
last modified time | relevance | path

Searched refs:OverheadType (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DCodegenUtils.cpp24 return OverheadType::kU64; in overheadTypeEncoding()
26 return OverheadType::kU32; in overheadTypeEncoding()
28 return OverheadType::kU16; in overheadTypeEncoding()
30 return OverheadType::kU8; in overheadTypeEncoding()
32 return OverheadType::kIndex; in overheadTypeEncoding()
47 case OverheadType::kIndex: in getOverheadType()
49 case OverheadType::kU64: in getOverheadType()
51 case OverheadType::kU32: in getOverheadType()
53 case OverheadType::kU16: in getOverheadType()
55 case OverheadType::kU8: in getOverheadType()
[all …]
H A DCodegenUtils.h34 OverheadType overheadTypeEncoding(unsigned width);
37 OverheadType overheadTypeEncoding(Type tp);
40 Type getOverheadType(Builder &builder, OverheadType ot);
43 OverheadType pointerOverheadTypeEncoding(const SparseTensorEncodingAttr &enc);
46 OverheadType indexOverheadTypeEncoding(const SparseTensorEncodingAttr &enc);
57 StringRef overheadTypeFunctionSuffix(OverheadType ot);
/llvm-project-15.0.7/mlir/lib/ExecutionEngine/
H A DSparseTensorUtils.cpp1557 OverheadType ptrTp, OverheadType indTp, in _mlir_ciface_newSparseTensor()
1582 CASE(OverheadType::kU64, OverheadType::kU8, PrimaryType::kF64, uint64_t, in _mlir_ciface_newSparseTensor()
1590 CASE(OverheadType::kU32, OverheadType::kU8, PrimaryType::kF64, uint32_t, in _mlir_ciface_newSparseTensor()
1600 CASE(OverheadType::kU8, OverheadType::kU64, PrimaryType::kF64, uint8_t, in _mlir_ciface_newSparseTensor()
1602 CASE(OverheadType::kU8, OverheadType::kU32, PrimaryType::kF64, uint8_t, in _mlir_ciface_newSparseTensor()
1604 CASE(OverheadType::kU8, OverheadType::kU16, PrimaryType::kF64, uint8_t, in _mlir_ciface_newSparseTensor()
1606 CASE(OverheadType::kU8, OverheadType::kU8, PrimaryType::kF64, uint8_t, in _mlir_ciface_newSparseTensor()
1634 CASE(OverheadType::kU8, OverheadType::kU64, PrimaryType::kF32, uint8_t, in _mlir_ciface_newSparseTensor()
1636 CASE(OverheadType::kU8, OverheadType::kU32, PrimaryType::kF32, uint8_t, in _mlir_ciface_newSparseTensor()
1638 CASE(OverheadType::kU8, OverheadType::kU16, PrimaryType::kF32, uint8_t, in _mlir_ciface_newSparseTensor()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/ExecutionEngine/
H A DSparseTensorUtils.h47 enum class OverheadType : uint32_t { enum
156 OverheadType ptrTp, OverheadType indTp,