Skip to content

Remove Unbroadcast Op #1286

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.

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
merged 2 commits into from
Mar 18, 2025
Merged

Conversation

ricardoV94
Copy link
Member

@ricardoV94 ricardoV94 commented Mar 11, 2025

Follow up to #1281, remove weird Op that was only used in Scan graphs. Removing revealed a bug in the scan save memory that I attempted to fix a while ago in #236


📚 Documentation preview 📚: https://pytensor--1286.org.readthedocs.build/en/1286/

Copy link

codecov bot commented Mar 14, 2025

Codecov Report

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

Project coverage is 82.01%. Comparing base (c822a8e) to head (ad9f405).
Report is 6 commits behind head on main.

Files with missing lines Patch % Lines
pytensor/compile/function/pfunc.py 0.00% 1 Missing ⚠️
Additional details and impacted files

Impacted file tree graph

@@            Coverage Diff             @@
##             main    #1286      +/-   ##
==========================================
- Coverage   82.03%   82.01%   -0.02%     
==========================================
  Files         188      188              
  Lines       48567    48442     -125     
  Branches     8675     8659      -16     
==========================================
- Hits        39841    39730     -111     
+ Misses       6574     6566       -8     
+ Partials     2152     2146       -6     
Files with missing lines Coverage Δ
pytensor/ifelse.py 52.27% <100.00%> (ø)
pytensor/link/jax/dispatch/shape.py 91.22% <100.00%> (-0.71%) ⬇️
pytensor/link/numba/dispatch/tensor_basic.py 88.79% <ø> (-0.64%) ⬇️
pytensor/link/pytorch/dispatch/shape.py 86.48% <100.00%> (+0.77%) ⬆️
pytensor/scan/basic.py 84.34% <100.00%> (ø)
pytensor/scan/op.py 84.96% <ø> (+0.13%) ⬆️
pytensor/scan/rewriting.py 82.89% <100.00%> (+0.27%) ⬆️
pytensor/tensor/basic.py 91.14% <100.00%> (-0.03%) ⬇️
pytensor/tensor/rewriting/shape.py 85.92% <ø> (+0.13%) ⬆️
pytensor/tensor/rewriting/subtensor.py 89.93% <ø> (-0.20%) ⬇️
... and 2 more

... and 1 file with indirect coverage changes

🚀 New features to boost your workflow:
  • Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@ricardoV94 ricardoV94 marked this pull request as ready for review March 14, 2025 11:55
Copy link
Member

@jessegrabowski jessegrabowski left a comment

Choose a reason for hiding this comment

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

-700 lines, you love to see it

else:
# FIXME: This is never used
nw_input = nw_inputs[in_idx][: (initl + nw_steps)]
# Number of steps in the initial state
Copy link
Member

Choose a reason for hiding this comment

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

is that FIXME in the old code still relevant?

Copy link
Member Author

Choose a reason for hiding this comment

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

Fixed with these changes

@ricardoV94 ricardoV94 merged commit 42a7adb into pymc-devs:main Mar 18, 2025
71 checks passed
@ricardoV94 ricardoV94 deleted the remove_unbroadcast branch March 18, 2025 07:06
@ricardoV94 ricardoV94 changed the title Remove Unbroadcast Op Remove Unbroadcast Op Mar 18, 2025
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