Skip to content

Numba: Speedup RV eval by casting float parameters to float64#2223

Merged
ricardoV94 merged 1 commit into
pymc-devs:mainfrom
ricardoV94:numba_rv_dtype_autocast
Jun 15, 2026
Merged

Numba: Speedup RV eval by casting float parameters to float64#2223
ricardoV94 merged 1 commit into
pymc-devs:mainfrom
ricardoV94:numba_rv_dtype_autocast

Conversation

@ricardoV94

@ricardoV94 ricardoV94 commented Jun 14, 2026

Copy link
Copy Markdown
Member

It was slowing us down, specially in conjunction with #1917

The MvNormal actually fails in float32, which I was intending to fix only in #2186

A bit more comment verbosity than I'd like, but nit-fine imo

@ricardoV94 ricardoV94 force-pushed the numba_rv_dtype_autocast branch from 6ffc6fd to eb35888 Compare June 15, 2026 09:15
@ricardoV94 ricardoV94 changed the title Numba: cast float RV parameters to float64 to speed up sampling Numba: Speedup RV eval by casting float parameters to float64 Jun 15, 2026

@jessegrabowski jessegrabowski left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

would prefer a commentary cleanup pass before this goes in

Comment thread pytensor/tensor/random/rewriting/numba.py Outdated
Comment thread pytensor/tensor/random/rewriting/numba.py
@ricardoV94 ricardoV94 force-pushed the numba_rv_dtype_autocast branch from eb35888 to 7a6610c Compare June 15, 2026 14:17
@ricardoV94

Copy link
Copy Markdown
Member Author

Cleanup done

@jessegrabowski

Copy link
Copy Markdown
Member

I double approve then

@ricardoV94 ricardoV94 merged commit e3c3e2d into pymc-devs:main Jun 15, 2026
64 checks passed
@ricardoV94 ricardoV94 deleted the numba_rv_dtype_autocast branch June 15, 2026 14:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants