Skip to content

ufunc.at #9239

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
sklam merged 18 commits into from
Mar 29, 2024
Merged

ufunc.at #9239

Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
f188770
[wip] ufunc.at
guilhermeleobas Oct 12, 2023
adc2934
upd
guilhermeleobas Oct 24, 2023
806a8f7
update tests
guilhermeleobas Oct 26, 2023
c706afa
fix CI bug
guilhermeleobas Oct 26, 2023
7c91799
fix CI bug (take 2)
guilhermeleobas Oct 26, 2023
e740181
add release notes
guilhermeleobas Oct 26, 2023
d04c33f
remove comented code
guilhermeleobas Oct 30, 2023
00e4ed2
Add missing test_ufunc_at_inner_loop test
guilhermeleobas Oct 31, 2023
ab87c5d
skip unsupported typecodes
guilhermeleobas Oct 31, 2023
2db339e
enable boolean array indexing
guilhermeleobas Nov 14, 2023
e4ca86b
rewrite ufunc.at to support multi dimension indexing
guilhermeleobas Nov 14, 2023
9dcb2bf
Fix isinstance check
guilhermeleobas Nov 15, 2023
f49e05b
fix test_import error
guilhermeleobas Nov 15, 2023
7ff257e
TypeError -> TypingError
guilhermeleobas Mar 8, 2024
a5f9549
Merge remote-tracking branch 'upstream/main' into guilhermeleobas/ufu…
guilhermeleobas Mar 8, 2024
1aac0f9
dynamically compile ufunc on calling 'at'
guilhermeleobas Mar 8, 2024
137d175
adjust release notes
guilhermeleobas Mar 8, 2024
a539aa2
Update 9239.new_feature.rst
guilhermeleobas Mar 20, 2024
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
Prev Previous commit
Next Next commit
fix test_import error
  • Loading branch information
@guilhermeleobas
guilhermeleobas committed Nov 15, 2023
commit f49e05b0fd65e27b467ac01becb79e3aec8e7de4
8 changes: 6 additions & 2 deletions numba/np/ufunc/dufunc.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,6 @@
from numba.parfors import array_analysis
from numba.np.ufunc import ufuncbuilder
from numba.np import numpy_support
from numba.np.arrayobj import (load_item, store_item, normalize_indices,
FancyIndexer, make_array)
from typing import Callable
from llvmlite import ir

Expand All @@ -43,6 +41,8 @@ def need_advanced_indexing(self):
return isinstance(self.indices_ty, types.BaseTuple)

def _prepare(self, context, builder):
from numba.np.arrayobj import normalize_indices, FancyIndexer

a, indices = self.a, self.indices
a_ty, indices_ty = self.a_ty, self.indices_ty

Expand All @@ -69,6 +69,7 @@ def _prepare(self, context, builder):
self.cres = self._compile_ufunc(context, builder)

def _load_val(self, context, builder, loop_indices, array, array_ty):
from numba.np.arrayobj import load_item
shapes = cgutils.unpack_tuple(builder, array.shape)
strides = cgutils.unpack_tuple(builder, array.strides)
data = array.data
Expand All @@ -92,6 +93,7 @@ def _load_flat(self, context, builder, indices, array, array_ty):
return None, val

def _store_val(self, context, builder, array, array_ty, ptr, val):
from numba.np.arrayobj import store_item
fromty = self.cres.signature.return_type
toty = array_ty.dtype
val = context.cast(builder, val, fromty, toty)
Expand Down Expand Up @@ -433,6 +435,8 @@ def ol_at(ufunc, a, indices, b=None):
self.add((a.dtype, b.dtype))

def apply_ufunc_codegen(context, builder, sig, args):
from numba.np.arrayobj import make_array

if len(args) == 4:
_, aty, idxty, bty = sig.args
_, a, indices, b = args
Expand Down