From cbe925bd7264c4e14681db16a14805670fd07c71 Mon Sep 17 00:00:00 2001
From: Tuowen Zhao <ztuowen@gmail.com>
Date: Tue, 11 Oct 2016 19:51:20 -0600
Subject: for loop index replacement

---
 src/transformations/loop.cc | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'src/transformations/loop.cc')

diff --git a/src/transformations/loop.cc b/src/transformations/loop.cc
index b3539a6..8f56be6 100644
--- a/src/transformations/loop.cc
+++ b/src/transformations/loop.cc
@@ -478,6 +478,9 @@ void Loop::align_loops(std::vector<ir_tree_node*> &ir_tree, std::vector<std::str
           CG_outputRepr *ivar = ocg->CreateIdent(iname);
           vars_to_be_replaced.push_back(clp->index()->name());
           vars_replacement.push_back(ivar);
+          ocg->CreateSubstitutedStmt(0,new CG_chillRepr(clp->chillforstmt->getInit()),vars_to_be_replaced,vars_replacement,false);
+          ocg->CreateSubstitutedStmt(0,new CG_chillRepr(clp->chillforstmt->getInc()),vars_to_be_replaced,vars_replacement,false);
+          ocg->CreateSubstitutedStmt(0,new CG_chillRepr(clp->chillforstmt->getCond()),vars_to_be_replaced,vars_replacement,false);
           // FIXME: this breaks abstraction
           if (clp->step_size()<0) {
             IR_CONDITION_TYPE cond = clp->conditionoperator;
-- 
cgit v1.2.3-70-g09d2