From baac37233f45ea307555425cd0b44c09faab6bb2 Mon Sep 17 00:00:00 2001
From: Ian Hinder <ian.hinder@aei.mpg.de>
Date: Fri, 6 Sep 2013 15:30:47 +0200
Subject: CodeGenCactus.m: Move vectorisation for AssignVariableInLoop into
 Vectorisation.m

---
 Tools/CodeGen/Vectorisation.m | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'Tools/CodeGen/Vectorisation.m')

diff --git a/Tools/CodeGen/Vectorisation.m b/Tools/CodeGen/Vectorisation.m
index 842282b..ab32421 100644
--- a/Tools/CodeGen/Vectorisation.m
+++ b/Tools/CodeGen/Vectorisation.m
@@ -18,12 +18,14 @@
     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *)
 
-BeginPackage["Vectorisation`", {"Errors`", "Helpers`", "Kranc`", "CodeGenC`"}];
+BeginPackage["Vectorisation`", {"Errors`", "Helpers`", "Kranc`", "CodeGenC`",
+                                "CodeGen`"}];
 
 VectoriseExpression;
 VectorisationLocalsToGridFunctions;
 OpenCLLocalsToGridFunctions;
 VectorisationSimpleAssignEquationList;
+VectorisationAssignVariableInLoop;
 
 Begin["`Private`"];
 
@@ -278,6 +280,10 @@ DefFn[
   {prepareStorePartialVariableInLoop["i", "vecimin", "vecimax"],
    MapThread[storePartialVariableInLoop, {lhss, rhss}]}];
 
+DefFn[
+  VectorisationAssignVariableInLoop[dest:(_String|_Symbol), src:CodeGenBlock] :=
+  {dest, " = ", "vec_load(", src, ")", EOL[]}];
+
 End[];
 
 EndPackage[];
-- 
cgit v1.2.3