Add positionContext save/restore around recursion into sub computation …

…#27
BoiseState-AdaptLab · Nov 25, 2021 · 8d6f76e · 8d6f76e
1 parent 63b493d
commit 8d6f76e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/ComputationBuilder.cpp b/src/ComputationBuilder.cpp
@@ -110,8 +110,10 @@ void ComputationBuilder::processSingleStmt(clang::Stmt *stmt) {
     std::string calleeName = calleeDefinition->getNameAsString();
     if (!subComputations.count(calleeName)) {
       // build Computation from calleeDefinition, if we haven't done so already
+      PositionContext oldContext = *positionContext;
       auto builder = new ComputationBuilder();
       subComputations[calleeName] = builder->buildComputationFromFunction(calleeDefinition);
+      *positionContext = oldContext;
     }
     std::vector<std::string> callArgStrings;
     for (unsigned int i = 0; i < asCallExpr->getNumArgs(); ++i) {

diff --git a/src/ComputationBuilderTest.cpp b/src/ComputationBuilderTest.cpp
@@ -307,7 +307,11 @@ TEST_F(ComputationBuilderTest, basic_nesting) {
                                                   "{[i]->[1,i,1]}",
                                                   {{"x", "{[i]->[0]}"}},
                                                   {{"_iegen_0x", "{[i]->[0]}"}}));
-  expectedComputation->addStmt(new iegenlib::Stmt("_iegen_0x*=5;", "{[i]: 0<=i<N}", "{[i]->[1,i,2]}", {}, {}));
+  expectedComputation->addStmt(new iegenlib::Stmt("_iegen_0x*=5;",
+                                                  "{[i]: 0<=i<N}",
+                                                  "{[i]->[1,i,2]}",
+                                                  {{"_iegen_0x", "{[i]->[0]}"}},
+                                                  {{"_iegen_0x", "{[i]->[0]}"}}));
 
   expectedComputation->addReturnValue("3", false);