From 0d307d11835be0450d1c10a44e5d959e3a7ef791 Mon Sep 17 00:00:00 2001
From: Gheorghe-Teodor Bercea <gheorghe-teod.bercea@ibm.com>
Date: Tue, 25 Feb 2020 09:47:42 -0500
Subject: [PATCH] Set flag to true when definition is emitted. (#97)

---
 src/dialect/krnl/krnl_helper.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/dialect/krnl/krnl_helper.cpp b/src/dialect/krnl/krnl_helper.cpp
index 4f75a43..87a5f80 100644
--- a/src/dialect/krnl/krnl_helper.cpp
+++ b/src/dialect/krnl/krnl_helper.cpp
@@ -168,6 +168,8 @@ void BuildKrnlLoop::createDefineAndOptimizeOp(bool withEmptyOptimization) {
   originalLoops.reserve(originalLoopNum);
   for (auto result : loopsOp.getResults())
     originalLoops.push_back(result);
+  createdDefineOp = true;
+
   // inserte optimize loop op.
   auto optimizedLoopsOp =
       rewriter.create<KrnlOptimizeLoopsOp>(loc, originalLoopNum);
@@ -182,9 +184,10 @@ void BuildKrnlLoop::createDefineAndOptimizeOp(bool withEmptyOptimization) {
     rewriter.create<KrnlReturnLoopsOp>(loc, originalLoops);
     rewriter.restoreInsertionPoint(ip);
   }
+  createdOptimizeOp = true;
+
   // prepare data structure to push bounds
   pack = new KrnlIterateOperandPack(rewriter, originalLoops, optLoops);
-  createdOptimizeOp = true;
 }
 
 // push bounds (lower and upper) and return index for loop info