Add new "Default" schedule type for dace <> gt4py schedule mapping

GridTools · Nov 28, 2023 · 808acb4 · 808acb4
1 parent ceef1d6
commit 808acb4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/gt4py/cartesian/gtc/daceir.py b/src/gt4py/cartesian/gtc/daceir.py
@@ -101,6 +101,7 @@ def from_dace_schedule(cls, schedule):
             dace.ScheduleType.Default: MapSchedule.Default,
             dace.ScheduleType.Sequential: MapSchedule.Sequential,
             dace.ScheduleType.CPU_Multicore: MapSchedule.CPU_Multicore,
+            dace.ScheduleType.GPU_Default: MapSchedule.GPU_Device,
             dace.ScheduleType.GPU_Device: MapSchedule.GPU_Device,
             dace.ScheduleType.GPU_ThreadBlock: MapSchedule.GPU_ThreadBlock,
         }[schedule]