Skip to content

Extend cholesky of triangular dot rewrite to matmul Ops - #459

Merged
ricardoV94 merged 2 commits into
pymc-devs:mainfrom
ricardoV94:tweak_chol_dot_rewrite
Sep 28, 2023
Merged

ricardoV94 merged 2 commits into
pymc-devs:mainfrom
ricardoV94:tweak_chol_dot_rewrite

Conversation

@ricardoV94

@ricardoV94 ricardoV94 commented Sep 26, 2023 •

Copy link
Copy Markdown
Member

Also restrict to 2D Dot cases

Also reverts #455

@ricardoV94 ricardoV94 added bug Something isn't working enhancement New feature or request graph rewriting linalg Linear algebra labels Sep 26, 2023

@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.

Looks good pending that you check the typo on ndim in the unittest

Comment thread pytensor/tensor/rewriting/linalg.py
Comment thread tests/tensor/rewriting/test_linalg.py Outdated

@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.

LGTM

@ricardoV94
ricardoV94 merged commit 39aa123 into pymc-devs:main Sep 28, 2023
@ricardoV94
ricardoV94 deleted the tweak_chol_dot_rewrite branch September 28, 2023 10:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working enhancement New feature or request graph rewriting linalg Linear algebra

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants