[Parser] Fixed an issue where register variables initialized to value…

…s in shared were being migrated in OpenCL mode (#167)
libocca · Jul 21, 2018 · 86efdaf · 86efdaf
1 parent 405fb35
commit 86efdaf
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/lang/modes/opencl.cpp b/src/lang/modes/opencl.cpp
@@ -242,8 +242,13 @@ namespace occa {
         statementExprMap::iterator it = exprMap.begin();
         while (it != exprMap.end()) {
           declarationStatement &declSmnt = *((declarationStatement*) it->first);
-          declSmnt.removeFromParent();
-          kernelSmnt.addFirst(declSmnt);
+
+          variable_t *var = declSmnt.declarations[0].variable;
+
+          if (var->hasAttribute("shared")) {
+            declSmnt.removeFromParent();
+            kernelSmnt.addFirst(declSmnt);
+          }
           ++it;
         }
       }