Skip to content

Cleanup numba dispatches #1648

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

Open
wants to merge 16 commits into
base: main
Choose a base branch
from
Open

Cleanup numba dispatches #1648

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
6c437ba
Remove unused numba_vectorize
ricardoV94 Oct 9, 2025
3b359fc
Remove numba__vectorize_target config
ricardoV94 Oct 14, 2025
6660ad7
Remove unused global_numba_func
ricardoV94 Oct 9, 2025
182abd0
Remove duplicated Solve dispatch
ricardoV94 Oct 9, 2025
0609870
Move slice dispatcher functionality to subtensor.py
ricardoV94 Oct 9, 2025
f523fd4
Move shape Ops dispatchers to their own file
ricardoV94 Oct 9, 2025
6c1c9da
Move sort Ops dispatchers to their own file
ricardoV94 Oct 9, 2025
8066720
Move NonZero Op dispatcher to tensor_basic
ricardoV94 Oct 9, 2025
91cb2c0
Move dot Op dispatchers to Elemwise
ricardoV94 Oct 9, 2025
0512b32
Don't use overload for deepcopy
ricardoV94 Oct 9, 2025
a458be6
Remove `to_scalar` helper
ricardoV94 Oct 9, 2025
9650630
Reorder functions in numba/dispatch/basic.py
ricardoV94 Oct 9, 2025
33522f3
Add error message in Numba implementation of SpecifyShape
ricardoV94 Oct 9, 2025
165520d
Cleanup sort Op dispatchers
ricardoV94 Oct 9, 2025
ea08864
Move TypeCastingOp dispatcher to basic.py
ricardoV94 Oct 9, 2025
c594eb4
Test numba slice boxing and fix representation of None stop with nega…
ricardoV94 Oct 14, 2025
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
6 changes: 0 additions & 6 deletions pytensor/configdefaults.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1099,12 +1099,6 @@ def add_scan_configvars():


def add_numba_configvars():
config.add(
"numba__vectorize_target",
("Default target for numba.vectorize."),
EnumStr("cpu", ["parallel", "cuda"], mutable=True),
in_c_key=False,
)
config.add(
"numba__fastmath",
("If True, use Numba's fastmath mode."),
Expand Down
1 change: 0 additions & 1 deletion pytensor/configparser.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -157,7 +157,6 @@ class PyTensorConfigParser:
scan__allow_gc: bool
scan__allow_output_prealloc: bool
# add_numba_configvars
numba__vectorize_target: str
numba__fastmath: bool
numba__cache: bool
# add_caching_dir_configvars
Expand Down
2 changes: 2 additions & 0 deletions pytensor/link/numba/dispatch/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,10 @@
import pytensor.link.numba.dispatch.random
import pytensor.link.numba.dispatch.scan
import pytensor.link.numba.dispatch.scalar
import pytensor.link.numba.dispatch.shape
import pytensor.link.numba.dispatch.signal
import pytensor.link.numba.dispatch.slinalg
import pytensor.link.numba.dispatch.sort
import pytensor.link.numba.dispatch.sparse
import pytensor.link.numba.dispatch.subtensor
import pytensor.link.numba.dispatch.tensor_basic
Expand Down
Loading