diff options
Diffstat (limited to 'mlir/lib/Transforms/DmaGeneration.cpp')
| -rw-r--r-- | mlir/lib/Transforms/DmaGeneration.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/mlir/lib/Transforms/DmaGeneration.cpp b/mlir/lib/Transforms/DmaGeneration.cpp index 0437fb143e0..04eb38e9fc9 100644 --- a/mlir/lib/Transforms/DmaGeneration.cpp +++ b/mlir/lib/Transforms/DmaGeneration.cpp @@ -171,7 +171,8 @@ static bool getFullMemRefAsRegion(OperationInst *opInst, unsigned numSymbols, getLoopIVs(*opInst, &ivs); auto *regionCst = region->getConstraints(); - SmallVector<Value *, 4> symbols(ivs.begin(), ivs.end()); + + SmallVector<Value *, 8> symbols = extractForInductionVars(ivs); regionCst->reset(rank, numSymbols, 0, symbols); // Memref dim sizes provide the bounds. |
