compiler: Remember to further improve block size

devitocodes · Jul 22, 2024 · 8c62ae9 · 8c62ae9
1 parent 8a0d4d1
commit 8c62ae9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/devito/arch/archinfo.py b/devito/arch/archinfo.py
@@ -670,8 +670,8 @@ def limits(self, compiler=None, language=None):
         language.
         """
         return {
-            'max-par-dims': np.inf,
-            'max-block-dims': np.inf,
+            'max-par-dims': sys.maxsize,
+            'max-block-dims': sys.maxsize,
         }
 
 

diff --git a/devito/types/dimension.py b/devito/types/dimension.py
@@ -1327,6 +1327,8 @@ def _arg_values(self, interval, grid=None, args=None, **kwargs):
             # no value supplied -> the sub-block will span the entire block
             return {name: args[self.parent.step.name]}
         else:
+            # TODO": Check the args for space order and apply heuristics (e.g.,
+            # `2*space_order`?) for even better block sizes
             value = self._arg_defaults()[name]
             if value <= args[self.root.max_name] - args[self.root.min_name] + 1:
                 return {name: value}