8000 ENH: create and use indexed inner loops by mattip · Pull Request #23136 · numpy/numpy · GitHub
[go: up one dir, main page]
Skip to content

ENH: create and use indexed inner loops #23136

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 20 commits into from
Feb 7, 2023
Merged

ENH: create and use indexed inner loops #23136

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
90243f5
ENH: add indexed loops
mattip Jan 18, 2023
3ba259d
refactor ufunc_at to first get the inner loop, then create the iterators
mattip Jan 22, 2023
7910e2b
ENH: add indexed lower loops and connect them to ufuncs
mattip Jan 23, 2023
eb21b25
ENH: use an indexed loop if possible in ufunc_at
mattip Jan 30, 2023
888bba2
MAINT: fix linting
mattip Jan 31, 2023
be64173
ENH: update and sync the public API
mattip Jan 31, 2023
8bfaaf3
BENCH: fix benchmark
mattip Jan 31, 2023
28706af
MAINT, BUG: fixes from review and testing
mattip Feb 1, 2023
1db1ffe
MAINT: linting
mattip Feb 1, 2023
ec1a434
MAINT: update EXPERIMENTAL_DTYPE_API_VERSION too
mattip Feb 1, 2023
6541cf5
DOC: add a performance release note
mattip Feb 1, 2023
3802b16
MAINT: rework release note, changes from review
mattip Feb 5, 2023
f064552
BUG: Clean up reference handling in `ufunc.at` a little.
seberg Feb 6, 2023
6cf9cc9
BUG: Deal with casting and non-contig/1-D indices
seberg Feb 6, 2023
ff97723
BUG: fixes from review: use iter->nd and adapt to unary ufuncs
mattip Feb 6, 2023
163d3c5
BUG: improve check for 1d operands (from review)
mattip Feb 7, 2023
f79325e
TST: add test (from review)
mattip Feb 7, 2023
b80bc2d
ENH: fixes from review
mattip Feb 7, 2023
26978d7
DOC: improve release snippet (from review)
mattip Feb 7, 2023
790426e
TST, BUG: add a test for unary ufuncs, fix condition for indexed loops
mattip Feb 7, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
ENH: add indexed loops
  • Loading branch information
@mattip
mattip committed Jan 30, 2023
commit 90243f53f8efa200e22ee13ef1f0a8e1120b1c3a
12 changes: 12 additions & 0 deletions numpy/core/src/umath/loops.c.src
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -540,6 +540,18 @@ NPY_NO_EXPORT NPY_GCC_OPT_3 @ATTR@ void
BINARY_LOOP_FAST(@type@, @type@, *out = in1 @OP@ in2);
}
}

NPY_NO_EXPORT NPY_GCC_OPT_3 @ATTR@ void
@TYPE@_@kind@@isa@_indexed(char const *ip1, npy_intp const *indx, char *value,
npy_intp const *dimensions, npy_intp const *steps) {
npy_intp is1 = steps[0], is2 = steps[1], os1 = steps[2];
npy_intp n = dimensions[0];
npy_intp i;
for(i = 0; i < n; i++, indx += is2, value += os1) {
@type@ op1 = *(@type@ *)(ip1 + is1 * indx[0]) + value[0];
}
}

#endif

/**end repeat2**/
Expand Down
12 changes: 12 additions & 0 deletions numpy/core/src/umath/loops_arithm_fp.dispatch.c.src
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -546,6 +546,18 @@ NPY_NO_EXPORT void NPY_CPU_DISPATCH_CURFX(@TYPE@_@kind@)
}
}
}

NPY_NO_EXPORT void NPY_CPU_DISPATCH_CURFX(@TYPE@_@kind@_indexed)
(char const *ip1, npy_intp const *indx, char *value,
npy_intp const *dimensions, npy_intp const *steps) {
npy_intp is1 = steps[0], is2 = steps[1], os1 = steps[2];
npy_intp n = dimensions[0];
npy_intp i;
for(i = 0; i < n; i++, indx += is2, value += os1) {
@type@ op1 = *(@type@ *)(ip1 + is1 * indx[0]) + value[0];
}
}

/**end repeat1**/
/**end repeat**/

Expand Down
0