| /llvm-project/mlir/include/mlir/Dialect/SparseTensor/IR/ |
| H A D | Enums.h | 154 enum class LevelFormat : uint64_t { enum
164 constexpr bool encPowOfTwo(LevelFormat fmt) { in encPowOfTwo()
165 auto enc = static_cast<std::underlying_type_t<LevelFormat>>(fmt); in encPowOfTwo()
170 static_assert(encPowOfTwo(LevelFormat::Dense) &&
171 encPowOfTwo(LevelFormat::Batch) &&
172 encPowOfTwo(LevelFormat::Compressed) &&
173 encPowOfTwo(LevelFormat::Singleton) &&
174 encPowOfTwo(LevelFormat::LooseCompressed) &&
175 encPowOfTwo(LevelFormat::NOutOfM));
177 template <LevelFormat... targets>
[all …]
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/IR/Detail/ |
| H A D | LvlTypeParser.cpp | 64 properties |= static_cast<uint64_t>(LevelFormat::Dense); in parseLvlType()
66 properties |= static_cast<uint64_t>(LevelFormat::Batch); in parseLvlType()
68 properties |= static_cast<uint64_t>(LevelFormat::Compressed); in parseLvlType()
70 properties |= static_cast<uint64_t>(LevelFormat::NOutOfM); in parseLvlType()
73 properties |= static_cast<uint64_t>(LevelFormat::LooseCompressed); in parseLvlType()
75 properties |= static_cast<uint64_t>(LevelFormat::Singleton); in parseLvlType()
|
| /llvm-project/mlir/lib/CAPI/Dialect/ |
| H A D | SparseTensor.cpp | 25 static_cast<int>(LevelFormat::Dense) &&
27 static_cast<int>(LevelFormat::Compressed) &&
29 static_cast<int>(LevelFormat::Singleton) &&
31 static_cast<int>(LevelFormat::LooseCompressed) &&
33 static_cast<int>(LevelFormat::NOutOfM),
34 "MlirSparseTensorLevelFormat (C-API) and LevelFormat (C++) mismatch");
117 *buildLevelType(static_cast<LevelFormat>(lvlFmt), props, n, m));
|
| /llvm-project/mlir/unittests/Dialect/SparseTensor/ |
| H A D | MergerTest.cpp | 124 class MergerTestBase : public ::testing::TestWithParam<LevelFormat> {
318 merger.setLevelAndType(tid(0), lid(0), 0, LevelFormat::Compressed); in MergerTest3T1L()
320 merger.setLevelAndType(tid(1), lid(0), 0, LevelFormat::Compressed); in MergerTest3T1L()
327 ::testing::Values(LevelFormat::Dense,
328 LevelFormat::Batch));
336 merger.setLevelAndType(tid(0), lid(0), 0, LevelFormat::Compressed); in MergerTest4T1L()
338 merger.setLevelAndType(tid(1), lid(0), 0, LevelFormat::Compressed); in MergerTest4T1L()
340 merger.setLevelAndType(tid(2), lid(0), 0, LevelFormat::Compressed); in MergerTest4T1L()
347 ::testing::Values(LevelFormat::Dense,
348 LevelFormat::Batch));
[all …]
|
| /llvm-project/mlir/test/python/dialects/sparse_tensor/ |
| H A D | dialect.py | 93 # CHECK: lvl_formats_enum: [{{65536|LevelFormat.dense}}, {{65536|LevelFormat.dense}}, {{2097152|LevelFormat.n_out_of_m}}]
117 st.LevelFormat.n_out_of_m, [], 2, 4
119 built_dense = st.EncodingAttr.build_level_type(st.LevelFormat.dense)
236 levels = [st.LevelFormat.compressed]
|
| /llvm-project/mlir/test/Integration/Dialect/SparseTensor/python/ |
| H A D | test_stress.py | 209 dense_lvl = st.EncodingAttr.build_level_type(st.LevelFormat.dense)
210 sparse_lvl = st.EncodingAttr.build_level_type(st.LevelFormat.compressed)
|
| H A D | test_output.py | 128 fmt = st.LevelFormat
|
| H A D | test_SpMM.py | 129 fmt = st.LevelFormat
|
| H A D | test_SDDMM.py | 143 fmt = st.LevelFormat
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 141 LevelFormat::Undef))) in foreachField()
145 LevelFormat::Undef))) in foreachField()
456 return LevelFormat::Batch; in getLvlType()
797 !(it - 1)->isa<LevelFormat::Compressed, LevelFormat::LooseCompressed>()) in verify()
826 return !lt.isa<LevelFormat::Singleton>(); in verify()
969 if (lt.isa<LevelFormat::Compressed, LevelFormat::LooseCompressed>()) { in getCOOSegments()
972 return !lt.isa<LevelFormat::Singleton>(); in getCOOSegments()
1017 *buildLevelType(LevelFormat in getCOOType()
[all...] |
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | SparseTensorIterator.cpp | 89 : SparseTensorLevel(tid, lvl, LevelFormat::Dense, lvlSize) {} in DenseLevel()
110 : SparseTensorLevel(tid, lvl, LevelFormat::Batch, lvlSize) {} in BatchLevel()
311 return stl.getLT().isa<LevelFormat::Batch>(); in isBatchIterator()
1565 case LevelFormat::Dense: in makeSparseTensorLevel()
1567 case LevelFormat::Batch: in makeSparseTensorLevel()
1569 case LevelFormat::Compressed: in makeSparseTensorLevel()
1571 case LevelFormat::LooseCompressed: in makeSparseTensorLevel()
1573 case LevelFormat::Singleton: in makeSparseTensorLevel()
1575 case LevelFormat::NOutOfM: in makeSparseTensorLevel()
1577 case LevelFormat::Undef: in makeSparseTensorLevel()
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseTensorCodegen.cpp | 1371 if (lt.isa<LevelFormat::Dense>()) { in matchAndRewrite()
1376 if (lt.isa<LevelFormat::Batch>()) { in matchAndRewrite()
|
| H A D | SparseTensorRewriting.cpp | 51 [](auto lt) { return lt == LevelFormat::Dense; }); in isSparseTensor()
|
| /llvm-project/mlir/lib/Dialect/SparseTensor/Utils/ |
| H A D | Merger.cpp | 231 std::vector<LevelType>(numLoops, LevelFormat::Undef)), in Merger()
|