diff options
| author | mattip <matti.picus@gmail.com> | 2022-12-29 14:16:43 +0200 |
|---|---|---|
| committer | mattip <matti.picus@gmail.com> | 2022-12-29 14:23:03 +0200 |
| commit | c95932eecccad8d3c8e77ae1f7b8238be95d3e78 (patch) | |
| tree | 024ef4d057fedcb96526cdb82bd1ed75fe41e70e | |
| parent | 845a327f4ff81bb55d6a15906bd65de375a34958 (diff) | |
| download | numpy-c95932eecccad8d3c8e77ae1f7b8238be95d3e78.tar.gz | |
use 4 for float, 2 for double in fast-path count
Signed-off-by: mattip <matti.picus@gmail.com>
| -rw-r--r-- | numpy/core/src/umath/loops_arithm_fp.dispatch.c.src | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/numpy/core/src/umath/loops_arithm_fp.dispatch.c.src b/numpy/core/src/umath/loops_arithm_fp.dispatch.c.src index 3b44e16f8..c1bfaa63a 100644 --- 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 (dimensions[0] < 2 || !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; |