v1.6.43; adds some logging to RCMG to better understand the effect of the size argument

Author: simon-bachhuber

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "imt-ring"
-version = "1.6.42"
+version = "1.6.43"
 authors = [
   { name="Simon Bachhuber", email="simon.bachhuber@fau.de" },
 ]
@@ -57,3 +57,6 @@ where = ["src"]
 [tool.setuptools.package-data]
 "ring.io.examples" = ["**/*.xml"]
 "ring.ml.params" = ["**/*.pickle"]
+
+[project.scripts]
+ring-view = "ring.extras.interactive_viewer:main"

src/ring/algorithms/generator/base.py

@@ -1,5 +1,6 @@
 from dataclasses import replace
 from functools import partial
+import logging
 import random
 from typing import Callable, Optional
 import warnings
@@ -20,6 +21,8 @@
 from ring.algorithms.generator import setup_fns
 from ring.algorithms.generator import types
 
+logger = logging.getLogger(__name__)
+
 
 class RCMG:
     def __init__(
@@ -237,7 +240,9 @@ def _number_of_executions_required(size: int) -> int:
     def _generators_ncalls(self, sizes: int | list[int] = 1):
         "Returns list of unbatched sequences as numpy arrays."
         repeats = self._compute_repeats(sizes)
+        logger.info(f"`repeats` = {repeats}")
         sizes = list(jnp.array(repeats) * jnp.array(self._size_of_generators))
+        logger.info(f"`sizes` = {sizes}")
 
         reduced_repeats = []
         n_calls = []
@@ -246,6 +251,9 @@ def _generators_ncalls(self, sizes: int | list[int] = 1):
             gcd = utils.gcd(n_call, repeat)
             n_calls.append(gcd)
             reduced_repeats.append(repeat // gcd)
+        logger.info(f"`reduced_repeats` = {reduced_repeats}")
+        logger.info(f"`n_calls` = {n_calls}")
+
         jits = [N > 1 for N in n_calls]
 
         gens = []