Skip to content

Refactor BLAS code #2049

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
jessegrabowski wants to merge 13 commits into
base: main
Choose a base branch
from
Open

Refactor BLAS code #2049

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
534b4fb
Refactor blas.py into a module, split out classes by Op
jessegrabowski Apr 12, 2026
984f077
Move blas c-code templates and codegen utils to tensor/blas/c_code an…
jessegrabowski Apr 12, 2026
cd41fe6
Move static C-code strings to header files for readability
jessegrabowski Apr 12, 2026
565bb56
Remove macOS 10.6 sdot bug workaround
jessegrabowski Apr 12, 2026
e17164e
Remove unused blas_c_headers
jessegrabowski Apr 12, 2026
5aea945
Refactor blas_headers.py to use external .h files
jessegrabowski Apr 12, 2026
2607a49
Restore macOS sdot bug, move test code and patch to separate files
jessegrabowski Apr 13, 2026
f009725
Move GEMM c code to header files where possible
jessegrabowski Apr 13, 2026
9cbc8ba
Move GEMV c code to header files where possible
jessegrabowski Apr 13, 2026
693fb57
Move GER c code to header files where possible
jessegrabowski Apr 13, 2026
2d5fcf2
Extract GER codegen into a C function, codegen becomes a function call
jessegrabowski Apr 13, 2026
26909a1
Don't cache blas_header_text
jessegrabowski Apr 19, 2026
8b69f18
mypy problems? IGNORE IGNORE IGNORE
jessegrabowski Apr 19, 2026
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
1 change: 0 additions & 1 deletion pytensor/tensor/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,6 @@ def _get_vector_length_Constant(op: Op | Variable, var: Constant) -> int:
# adds shared-variable constructors
from pytensor.tensor import (
blas,
blas_c,
sharedvar,
xlogx,
)
Expand Down
4 changes: 2 additions & 2 deletions pytensor/tensor/basic.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1774,9 +1774,9 @@ def do_constant_folding(self, fgraph, node):
| pytensor.tensor.subtensor.AdvancedIncSubtensor1
| pytensor.tensor.subtensor.AdvancedIncSubtensor
| pytensor.tensor.blas.Gemv
| pytensor.tensor.blas_c.CGemv
| pytensor.tensor.blas.CGemv
| pytensor.tensor.blas.Ger
| pytensor.tensor.blas_c.CGer,
| pytensor.tensor.blas.CGer,
)
):
# Ops that will work inplace on the Alloc. So if they
Expand Down
Loading
Loading