Add CaKernel initialisation function to the schedule

ianhinder · Jul 25, 2012 · 7db43a1 · 7db43a1
1 parent 0e90f77
commit 7db43a1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/Tools/CodeGen/CaKernel.m b/Tools/CodeGen/CaKernel.m
@@ -91,8 +91,16 @@
     {},
     Map[kernelCCLBlock[#,OptionValue[TileSize]] &, Select[calcs, CalculationOnDevice]]]];
 
-DefFn[CaKernelSchedule[] :=
-      {}];
+DefFn[CaKernelSchedule[thornName_] :=
+{
+    {
+      Name          -> thornName <> "_Init",
+      SchedulePoint -> "in CCTK_BASEGRID after Accelerator_SetDevice",
+      Language      -> "C",
+      Options       -> "local",
+      Comment       -> "Initialize CUDA Device"
+    }
+}]
 
 CaKernelConfigurationCLL[] :=
   "REQUIRES CUDA MPI\n";

diff --git a/Tools/CodeGen/Schedule.m b/Tools/CodeGen/Schedule.m
@@ -303,7 +303,7 @@
            {scheduleMoLRegister}];
 
     If[OptionValue[UseCaKernel],
-       scheduledFunctions = Join[scheduledFunctions, CaKernelSchedule[]]];
+       scheduledFunctions = Join[scheduledFunctions, CaKernelSchedule[thornName]]];
 
     allParams = Union@@((lookup[#,Parameters] &) /@ calcs);
     schedule = CreateSchedule[globalStorageGroups,