diff options
Diffstat (limited to 'Tools/CodeGen/Vectorisation.m')
-rw-r--r-- | Tools/CodeGen/Vectorisation.m | 8 |
1 files changed, 7 insertions, 1 deletions
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[]; |