Skip to content

Commit db1d40f

Browse files
authored
[mlir][sparse] refactor dim2lvl/lvl2dim passing into MapRef (#68649)
This revision refactors all "swiss army knife" entry points to pass dim2lvl/lvl2dim mapping, so that the callee can construct a MapRef (shown for SparseTensorStorage class). This is a next step towards completely centralizing mapping code into a single MapRef class.
1 parent 85d0fbe commit db1d40f

File tree

4 files changed

+99
-144
lines changed

4 files changed

+99
-144
lines changed

mlir/include/mlir/ExecutionEngine/SparseTensor/File.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ class SparseTensorReader final {
203203
MapRef map(dimRank, lvlRank, dim2lvl, lvl2dim);
204204
auto *coo = readCOO<V>(map, lvlSizes);
205205
auto *tensor = SparseTensorStorage<P, I, V>::newFromCOO(
206-
dimRank, getDimSizes(), lvlRank, lvlTypes, lvl2dim, *coo);
206+
dimRank, getDimSizes(), lvlRank, lvlTypes, dim2lvl, lvl2dim, *coo);
207207
delete coo;
208208
return tensor;
209209
}

0 commit comments

Comments
 (0)