Skip to content

Fix SDPA output dtype following config.floatX instead of the input dtype - #165

Open
cetagostini wants to merge 2 commits into
pymc-devs:mainfrom
cetagostini:fix-sdpa-dtype-promotion
Open

cetagostini wants to merge 2 commits into
pymc-devs:mainfrom
cetagostini:fix-sdpa-dtype-promotion

Address SDPA dtype review feedback

7e8f24d
Select commit
Loading
Failed to load commit list.
Sign in for the full log view

Annotations

2 warnings and 1 notice
documentation-links
succeeded Sep 19, 2026 in 7s