modularize fetching the number of candidates

Author: kaushikcfd

pyop2/transforms/auto_tiling.py

@@ -1766,7 +1766,10 @@ def estimated_exec_time(self, tiling_config):
         )
         return (total_time, nsync)
 
-    def __call__(self) -> Tuple[ParametricTiling, ...]:
+    def _get_all_configurations(self) -> FrozenSet[ParametricTiling]:
+        """
+        Returns a :class:`frozenset` of all configurations in our search space.
+        """
         from itertools import product
 
         threads_to_cells = {}
@@ -1845,9 +1848,11 @@ def get_eta_shared_mem_alias(tiles):
                             False,
                         )
                     )
+        return frozenset(params)
 
-        # sort the parameters with highest occupancy.
-        params.sort(key=lambda P: self.estimated_exec_time(P))
+    def __call__(self) -> Tuple[ParametricTiling, ...]:
+        params = sorted(self._get_all_configurations(),
+                        key=self.estimated_exec_time)
 
         return tuple(params[: self.num_param_tiling_candidates])