| /llvm-project-15.0.7/mlir/lib/CAPI/Dialect/ |
| H A D | SparseTensor.cpp | 25 static_cast<int>(SparseTensorEncodingAttr::DimLevelType::Dense) &&
28 SparseTensorEncodingAttr::DimLevelType::Compressed) &&
30 static_cast<int>(SparseTensorEncodingAttr::DimLevelType::Singleton),
41 SmallVector<SparseTensorEncodingAttr::DimLevelType> cppDimLevelTypes; in mlirSparseTensorEncodingAttrGet()
45 static_cast<SparseTensorEncodingAttr::DimLevelType>(dimLevelTypes[i]); in mlirSparseTensorEncodingAttrGet()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | CodegenUtils.cpp | 144 DimLevelType mlir::sparse_tensor::dimLevelTypeEncoding( in dimLevelTypeEncoding()
145 SparseTensorEncodingAttr::DimLevelType dlt) { in dimLevelTypeEncoding()
147 case SparseTensorEncodingAttr::DimLevelType::Dense: in dimLevelTypeEncoding()
148 return DimLevelType::kDense; in dimLevelTypeEncoding()
149 case SparseTensorEncodingAttr::DimLevelType::Compressed: in dimLevelTypeEncoding()
150 return DimLevelType::kCompressed; in dimLevelTypeEncoding()
151 case SparseTensorEncodingAttr::DimLevelType::Singleton: in dimLevelTypeEncoding()
152 return DimLevelType::kSingleton; in dimLevelTypeEncoding()
|
| H A D | CodegenUtils.h | 72 DimLevelType dimLevelTypeEncoding(SparseTensorEncodingAttr::DimLevelType dlt);
191 SparseTensorEncodingAttr::DimLevelType dlt) { in constantDimLevelTypeEncoding()
|
| H A D | SparseTensorRewriting.cpp | 35 ArrayRef<SparseTensorEncodingAttr::DimLevelType> dimTypes = in isSparseTensor()
38 if (dimType == SparseTensorEncodingAttr::DimLevelType::Compressed) in isSparseTensor()
|
| H A D | SparseTensorConversion.cpp | 202 ArrayRef<SparseTensorEncodingAttr::DimLevelType> dlt = enc.getDimLevelType(); in newParams()
368 ArrayRef<SparseTensorEncodingAttr::DimLevelType> dimTypes) { in canUseDirectConversion()
372 case SparseTensorEncodingAttr::DimLevelType::Compressed: in canUseDirectConversion()
377 case SparseTensorEncodingAttr::DimLevelType::Dense: in canUseDirectConversion()
381 case SparseTensorEncodingAttr::DimLevelType::Singleton: in canUseDirectConversion()
744 SmallVector<SparseTensorEncodingAttr::DimLevelType>( in matchAndRewrite()
745 rank, SparseTensorEncodingAttr::DimLevelType::Dense), in matchAndRewrite()
|
| H A D | Sparsification.cpp | 147 SparseTensorEncodingAttr::DimLevelType tp = enc.getDimLevelType()[d]; in toDim()
148 if (tp == SparseTensorEncodingAttr::DimLevelType::Compressed) in toDim()
150 if (tp == SparseTensorEncodingAttr::DimLevelType::Singleton) in toDim()
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/python/ |
| H A D | test_SpMM.py | 130 levels = [[st.DimLevelType.dense, st.DimLevelType.dense],
131 [st.DimLevelType.dense, st.DimLevelType.compressed],
132 [st.DimLevelType.compressed, st.DimLevelType.dense],
133 [st.DimLevelType.compressed, st.DimLevelType.compressed]]
|
| H A D | test_SDDMM.py | 135 levels = [[st.DimLevelType.dense, st.DimLevelType.dense],
136 [st.DimLevelType.dense, st.DimLevelType.compressed],
137 [st.DimLevelType.compressed, st.DimLevelType.dense],
138 [st.DimLevelType.compressed, st.DimLevelType.compressed]]
|
| H A D | test_output.py | 80 levels = [[st.DimLevelType.dense, st.DimLevelType.compressed],
81 [st.DimLevelType.compressed, st.DimLevelType.compressed]]
|
| H A D | test_stress.py | 210 [st.DimLevelType.dense, st.DimLevelType.compressed], rank)))
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 50 SmallVector<SparseTensorEncodingAttr::DimLevelType, 4> dlt; in parse()
71 dlt.push_back(SparseTensorEncodingAttr::DimLevelType::Dense); in parse()
73 dlt.push_back(SparseTensorEncodingAttr::DimLevelType::Compressed); in parse()
75 dlt.push_back(SparseTensorEncodingAttr::DimLevelType::Singleton); in parse()
123 case DimLevelType::Dense: in print()
126 case DimLevelType::Compressed: in print()
129 case DimLevelType::Singleton: in print()
145 ArrayRef<DimLevelType> dimLevelType, AffineMap dimOrdering, in verify()
|
| /llvm-project-15.0.7/mlir/lib/ExecutionEngine/ |
| H A D | SparseTensorUtils.cpp | 276 assert((dimTypes[r] == DimLevelType::kDense || in SparseTensorStorageBase()
371 const std::vector<DimLevelType> dimTypes;
580 const DimLevelType *sparsity, in newSparseTensor()
921 case DimLevelType::kCompressed: in SparseTensorNNZ()
927 case DimLevelType::kDense: in SparseTensorNNZ()
931 case DimLevelType::kSingleton: in SparseTensorNNZ()
983 if (dimTypes[r] == DimLevelType::kCompressed) in add()
1006 const std::vector<DimLevelType> &dimTypes;
1430 const DimLevelType *sparsity = (DimLevelType *)(sparse); in toMLIRSparseTensor()
1441 if (sparsity[i] != DimLevelType::kDense && in toMLIRSparseTensor()
[all …]
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/tools/ |
| H A D | mlir_pytaco_utils.py | 236 sparsity_codes: Sequence[sparse_tensor.DimLevelType], type: str) -> str: argument
270 sparsity: Sequence[sparse_tensor.DimLevelType], argument
314 sparsity_codes: Sequence[sparse_tensor.DimLevelType], argument
344 sparsity: Sequence[sparse_tensor.DimLevelType], argument
|
| H A D | mlir_pytaco.py | 189 DENSE = sparse_tensor.DimLevelType.dense
190 COMPRESSED = sparse_tensor.DimLevelType.compressed
|
| /llvm-project-15.0.7/mlir/include/mlir/ExecutionEngine/ |
| H A D | SparseTensorUtils.h | 118 enum class DimLevelType : uint8_t { enum
153 _mlir_ciface_newSparseTensor(StridedMemRefType<DimLevelType, 1> *aref, // NOLINT
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorAttrDefs.td | 57 "SparseTensorEncodingAttr::DimLevelType",
90 enum class DimLevelType {
|
| /llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/taco/ |
| H A D | unit_test_tensor_utils.py | 67 sparsity_codes: Sequence[sparse_tensor.DimLevelType]) -> int: argument
|