diff options
| author | mattip <matti.picus@gmail.com> | 2022-12-27 23:49:48 +0200 |
|---|---|---|
| committer | mattip <matti.picus@gmail.com> | 2022-12-28 00:09:28 +0200 |
| commit | cfa2b176956b9632020747556b8665d163d3beec (patch) | |
| tree | 793948313d82e77fdb23f53ce9a9b9f36e86049e | |
| parent | 154c293786a29cfbf976f08cab48698166d99399 (diff) | |
| download | numpy-cfa2b176956b9632020747556b8665d163d3beec.tar.gz | |
MAINT: do not try SIMD arithmetic loops if count < 4
| -rw-r--r-- | numpy/core/src/umath/loops_arithm_fp.dispatch.c.src | 2 |
1 files changed, 1 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 f637ecfe4..86f062b93 100644 --- a/numpy/core/src/umath/loops_arithm_fp.dispatch.c.src +++ b/numpy/core/src/umath/loops_arithm_fp.dispatch.c.src @@ -537,7 +537,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] < 4 || !run_binary_simd_@kind@_@TYPE@(args, dimensions, steps)) { BINARY_LOOP { const @type@ in1 = *(@type@ *)ip1; const @type@ in2 = *(@type@ *)ip2; |