KrancThorn.m: Accumulate calculation sources in sources variable

ianhinder · Sep 12, 2013 · 3567168 · 3567168
1 parent 665ca92
commit 3567168
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/Tools/CodeGen/KrancThorn.m b/Tools/CodeGen/KrancThorn.m
@@ -362,13 +362,16 @@ Thorn generation (main entry point for non-tensorial thorns)
     InfoMessage[Terse, "Creating calculation source files"];
 
     hostCalcs = Select[calcs, !CalculationOnDevice[#] &];
-    deviceCalcs = Select[calcs, CalculationOnDevice];
-
-    calcSources = Join[Map[CreateSetterSource[{#}, False, {}, opts] &, hostCalcs],
-                       Map[CaKernelCode[#,opts] &, deviceCalcs]];
+    hostSources = Map[{Filename -> lookup[#, Name] <> ext,
+                       Contents -> CreateSetterSource[{#}, False, {}, opts]} &,
+                      hostCalcs];
+    sources = Join[sources, hostSources];
 
-    calcFilenames = Join[Map[lookup[#, Name] <> ext &, hostCalcs],
-                         Map["CaKernel__"<>lookup[#, Name] <> ".code" &, deviceCalcs]];
+    deviceCalcs = Select[calcs, CalculationOnDevice];
+    deviceSources = Map[{Filename -> "CaKernel__"<>lookup[#, Name] <> ".code",
+                         Contents -> CaKernelCode[#,opts]} &,
+                        deviceCalcs];
+    sources = Join[sources, deviceSources];
 
     incFilenames = Map[lookup[#, Name] <> ext &, hostCalcs];
 
@@ -396,8 +399,6 @@ Thorn generation (main entry point for non-tensorial thorns)
                  CaKernel      -> cakernel,
                  Makefile      -> make,
                  Sources       -> Join[sources,
-                  MapThread[{Filename -> #1, Contents -> #2} &, 
-                            {calcFilenames, calcSources}],
                   If[Length[OptionValue[ParameterConditions]] > 0,
                      {{Filename -> "ParamCheck.cc",
                       Contents -> ParameterCheckSource[thornName, OptionValue[ParameterConditions]]}},