Skip to content

Simplify and speedup numba DimShuffle implementation

50801a3
Select commit
Loading
Failed to load commit list.
Merged

Speedup DimShuffle and Reshape in C and Numba backends #1226

Simplify and speedup numba DimShuffle implementation
50801a3
Select commit
Loading
Failed to load commit list.
Codecov / codecov/project succeeded Feb 25, 2025 in 1s

81.99% (+0.00%) compared to 53763f5

View this Pull Request on Codecov

81.99% (+0.00%) compared to 53763f5

Details

Codecov Report

Attention: Patch coverage is 96.55172% with 1 line in your changes missing coverage. Please review.

Project coverage is 81.99%. Comparing base (53763f5) to head (50801a3).

Files with missing lines Patch % Lines
pytensor/link/pytorch/dispatch/elemwise.py 0.00% 1 Missing ⚠️
Additional details and impacted files

Impacted file tree graph

@@           Coverage Diff           @@
##             main    #1226   +/-   ##
=======================================
  Coverage   81.98%   81.99%           
=======================================
  Files         188      188           
  Lines       48568    48523   -45     
  Branches     8677     8668    -9     
=======================================
- Hits        39819    39785   -34     
+ Misses       6584     6579    -5     
+ Partials     2165     2159    -6     
Files with missing lines Coverage Δ
pytensor/link/jax/dispatch/elemwise.py 83.82% <100.00%> (+2.13%) ⬆️
pytensor/link/numba/dispatch/elemwise.py 94.82% <100.00%> (+0.14%) ⬆️
pytensor/tensor/elemwise.py 89.59% <100.00%> (+0.59%) ⬆️
pytensor/tensor/shape.py 90.10% <100.00%> (-0.05%) ⬇️
pytensor/link/pytorch/dispatch/elemwise.py 66.16% <0.00%> (+1.45%) ⬆️

... and 1 file with indirect coverage changes