numpy · seberg · Jan 3, 2023 · Dec 10, 2022 · Dec 11, 2022 · Dec 25, 2022
diff --git a/benchmarks/benchmarks/bench_ufunc.py b/benchmarks/benchmarks/bench_ufunc.py
@@ -34,6 +34,16 @@ def time_broadcast(self):
         self.d - self.e
 
 
+class At(Benchmark):
+    def setup(self):
+        self.vals = np.random.rand(1_000_000)
+        self.idx = np.random.randint(1000, size=1_000_000).astype(np.intp)
+        self.res = np.zeros(1000, dtype=self.vals.dtype)
+
+    def time_sum_at(self):
+        np.add.at(self.res, self.idx, self.vals)
+
+
 class UFunc(Benchmark):
     params = [ufuncs]
     param_names = ['ufunc']

diff --git a/numpy/core/src/umath/legacy_array_method.c b/numpy/core/src/umath/legacy_array_method.c
@@ -91,7 +91,6 @@ generic_wrapped_legacy_loop(PyArrayMethod_Context *NPY_UNUSED(context),
     return 0;
 }
 
-
 /*
  * Signal that the old type-resolution function must be used to resolve
  * the descriptors (mainly/only used for datetimes due to the unit).

diff --git a/numpy/core/src/umath/loops_arithm_fp.dispatch.c.src b/numpy/core/src/umath/loops_arithm_fp.dispatch.c.src
@@ -514,6 +514,7 @@ run_binary_simd_@kind@_@TYPE@(char **args, npy_intp const *dimensions, npy_intp
  *  #TYPE = FLOAT, DOUBLE#
  *  #c = f, #
  *  #C = F, #
+ *  #count = 4,2#
  */
 /**begin repeat1
  * Arithmetic
@@ -537,7 +538,7 @@ NPY_NO_EXPORT void NPY_CPU_DISPATCH_CURFX(@TYPE@_@kind@)
         *((@type@ *)iop1) = io1;
 #endif
     }
-    else if (!run_binary_simd_@kind@_@TYPE@(args, dimensions, steps)) {
+    else if (dimensions[0] < @count@ || !run_binary_simd_@kind@_@TYPE@(args, dimensions, steps)) {
         BINARY_LOOP {
             const @type@ in1 = *(@type@ *)ip1;
             const @type@ in2 = *(@type@ *)ip2;
-Original file line number
+Diff line change
@@ Expand Up @@
         return 0;
     }
     /*
      * Signal that the old type-resolution function must be used to resolve
      * the descriptors (mainly/only used for datetimes due to the unit).
@@ Expand Down @@