diff options
Diffstat (limited to 'mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.h')
| -rw-r--r-- | mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.h | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.h b/mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.h index c6518de..320b397 100644 --- a/mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.h +++ b/mlir/lib/Dialect/SparseTensor/Transforms/LoopEmitter.h @@ -113,12 +113,11 @@ public: /// to `LoopId`. void initialize(ValueRange tensors, StringAttr loopTag = nullptr, bool hasOutput = false, bool isSparseOut = false, - ArrayRef<LoopId> topSort = {}, - DependentLvlGetter getter = nullptr); + unsigned numLoops = 0, DependentLvlGetter getter = nullptr); explicit LoopEmitter(ValueRange tensors, StringAttr loopTag = nullptr, bool hasOutput = false, bool isSparseOut = false, - ArrayRef<LoopId> topSort = {}, + unsigned numLoops = 0, DependentLvlGetter getter = nullptr); /// Starts a loop emitting session by generating all the buffers needed @@ -751,11 +750,6 @@ private: // TODO: maybe we should have a LoopSeqInfo std::vector<std::pair<Value, std::vector<std::tuple<TensorId, Level, bool>>>> loopSeqStack; - - /// Maps `LoopId` (used by `AffineDimExpr`) to `LoopOrd` (in the `loopStack`). - /// TODO: We should probably use a callback function here to make it more - /// general. - std::vector<LoopOrd> loopIdToOrd; }; } // namespace sparse_tensor |