copilot fixes

Author: vchamarthi

benchmarks/benchmarks/__init__.py

@@ -26,7 +26,7 @@
 
 
 def _physical_cores():
-    """Return physical core count from /proc/cpuinfo; fall back to os.cpu_count()."""
+    """Return physical core count from /proc/cpuinfo; fall back to 1 (conservative)."""
     try:
         with open("/proc/cpuinfo") as f:
             content = f.read()
@@ -36,7 +36,7 @@ def _physical_cores():
         )
         return cpu_cores * sockets
     except Exception:
-        return os.cpu_count() or 1
+        return 1
 
 
 def _thread_count():

benchmarks/benchmarks/_utils.py

@@ -0,0 +1,16 @@
+"""Shared utilities for mkl_fft benchmarks."""
+
+import numpy as np
+
+
+def _make_input(rng, shape, dtype):
+    """Return an array of *shape* and *dtype*.
+
+    Complex dtypes get non-zero imaginary parts for a realistic signal.
+    *shape* may be an int (1-D) or a tuple.
+    """
+    dt = np.dtype(dtype)
+    s = (shape,) if isinstance(shape, int) else shape
+    if dt.kind == "c":
+        return (rng.standard_normal(s) + 1j * rng.standard_normal(s)).astype(dt)
+    return rng.standard_normal(s).astype(dt)

benchmarks/benchmarks/bench_fft1d.py

@@ -4,19 +4,9 @@
 
 import mkl_fft
 
-_RNG_SEED = 42
-
-
-def _make_input(rng, n, dtype):
-    """Return a 1-D array of length *n* with the given *dtype*.
+from ._utils import _make_input
 
-    Complex dtypes are populated with non-zero imaginary parts so the
-    benchmark exercises a genuine complex transform path.
-    """
-    dt = np.dtype(dtype)
-    if dt.kind == "c":
-        return (rng.standard_normal(n) + 1j * rng.standard_normal(n)).astype(dt)
-    return rng.standard_normal(n).astype(dt)
+_RNG_SEED = 42
 
 
 # ---------------------------------------------------------------------------

benchmarks/benchmarks/bench_fftnd.py

@@ -4,21 +4,9 @@
 
 import mkl_fft
 
-_RNG_SEED = 42
-
+from ._utils import _make_input
 
-def _make_input(rng, shape, dtype):
-    """Return an array of the given *shape* and *dtype*.
-
-    Complex dtypes are populated with non-zero imaginary parts so the
-    benchmark exercises a genuine complex transform path.
-    """
-    dt = np.dtype(dtype)
-    if dt.kind == "c":
-        return (
-            rng.standard_normal(shape) + 1j * rng.standard_normal(shape)
-        ).astype(dt)
-    return rng.standard_normal(shape).astype(dt)
+_RNG_SEED = 42
 
 
 # ---------------------------------------------------------------------------

benchmarks/benchmarks/bench_memory.py

@@ -8,15 +8,9 @@
 
 import mkl_fft
 
-_RNG_SEED = 42
-
+from ._utils import _make_input
 
-def _make_input(rng, shape, dtype):
-    dt = np.dtype(dtype)
-    s = (shape,) if isinstance(shape, int) else shape
-    if dt.kind == "c":
-        return (rng.standard_normal(s) + 1j * rng.standard_normal(s)).astype(dt)
-    return rng.standard_normal(s).astype(dt)
+_RNG_SEED = 42
 
 
 # ---------------------------------------------------------------------------

benchmarks/benchmarks/bench_numpy_fft.py

@@ -14,20 +14,9 @@
 
 from mkl_fft.interfaces import numpy_fft
 
-_RNG_SEED = 42
-
-
-def _make_input(rng, shape, dtype):
-    """Return an array of *shape* and *dtype*.
+from ._utils import _make_input
 
-    Complex dtypes get non-zero imaginary parts for a realistic signal.
-    *shape* may be an int (1-D) or a tuple.
-    """
-    dt = np.dtype(dtype)
-    s = (shape,) if isinstance(shape, int) else shape
-    if dt.kind == "c":
-        return (rng.standard_normal(s) + 1j * rng.standard_normal(s)).astype(dt)
-    return rng.standard_normal(s).astype(dt)
+_RNG_SEED = 42
 
 
 # ---------------------------------------------------------------------------
@@ -182,7 +171,7 @@ def time_irfft2(self, shape, dtype):
 # ---------------------------------------------------------------------------
 
 
-class TimeCCND:
+class TimeC2CND:
     """numpy_fft.fftn / ifftn — N-D."""
 
     params = [

benchmarks/benchmarks/bench_scipy_fft.py

@@ -16,20 +16,9 @@
 
 from mkl_fft.interfaces import scipy_fft
 
-_RNG_SEED = 42
-
-
-def _make_input(rng, shape, dtype):
-    """Return an array of *shape* and *dtype*.
+from ._utils import _make_input
 
-    Complex dtypes get non-zero imaginary parts for a realistic signal.
-    *shape* may be an int (1-D) or a tuple.
-    """
-    dt = np.dtype(dtype)
-    s = (shape,) if isinstance(shape, int) else shape
-    if dt.kind == "c":
-        return (rng.standard_normal(s) + 1j * rng.standard_normal(s)).astype(dt)
-    return rng.standard_normal(s).astype(dt)
+_RNG_SEED = 42
 
 
 # ---------------------------------------------------------------------------
@@ -218,7 +207,7 @@ def time_ihfft2(self, shape, dtype):
 # ---------------------------------------------------------------------------
 
 
-class TimeCCND:
+class TimeC2CND:
     """scipy_fft.fftn / ifftn — N-D."""
 
     params = [