diff options
Diffstat (limited to 'mlir/lib/Transforms/DmaGeneration.cpp')
| -rw-r--r-- | mlir/lib/Transforms/DmaGeneration.cpp | 7 |
1 files changed, 1 insertions, 6 deletions
diff --git a/mlir/lib/Transforms/DmaGeneration.cpp b/mlir/lib/Transforms/DmaGeneration.cpp index e60f3531b62..df4aa84b039 100644 --- a/mlir/lib/Transforms/DmaGeneration.cpp +++ b/mlir/lib/Transforms/DmaGeneration.cpp @@ -223,13 +223,8 @@ bool DmaGeneration::generateDma(const MemRefRegion ®ion, ForInst *forInst, // on; this would correspond to loop IVs surrounding the level at which the // DMA generation is being done. const FlatAffineConstraints *cst = region.getConstraints(); - auto ids = cst->getIds(); SmallVector<Value *, 8> outerIVs; - for (unsigned i = rank, e = ids.size(); i < e; i++) { - auto id = cst->getIds()[i]; - assert(id.hasValue() && "Value id expected"); - outerIVs.push_back(id.getValue()); - } + cst->getIdValues(rank, cst->getNumIds(), &outerIVs); // Construct the index expressions for the fast memory buffer. The index // expression for a particular dimension of the fast buffer is obtained by |
