compiler: fix tile input

devitocodes · Oct 7, 2023 · 41457f1 · 41457f1
1 parent 732b470
commit 41457f1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/devito/passes/iet/parpragma.py b/devito/passes/iet/parpragma.py
@@ -614,7 +614,7 @@ def _make_partree(self, candidates, nthreads=None, index=None):
         if self._is_offloadable(root):
             body = self.DeviceIteration(gpu_fit=self.gpu_fit,
                                         ncollapsed=len(collapsable) + 1,
-                                        tile=self.par_tile,
+                                        tile=self.par_tile.next(),
                                         **root.args)
             partree = ParallelTree([], body, nthreads=nthreads)
 

diff --git a/devito/tools/data_structures.py b/devito/tools/data_structures.py
@@ -626,6 +626,8 @@ def __init__(self, items):
         self.current = 0
 
     def next(self):
+        if self.last == 0:
+            return None
         item = self.items[self.current]
         self.current = min(self.last - 1, self.current+1)
         return item