Implement ScalarLoop in torch backend#958

Merged

Ch0ronomato merged 27 commits intopymc-devs:mainfrom

Ch0ronomato:scalarloop

Dec 8, 2024

Commits on Nov 29, 2024

Add for loop based scalar loop

Ch0ronomato
authored and
Ian Schweer
committed
Pass all loop tests
Ian Schweer
committed
Fetch constants from op
Ian Schweer
committed
Add while loop test

Ch0ronomato
authored and
Ian Schweer
committed
Fix while loop and nasty stack over dtypes
Ian Schweer
committed
Disable compile here based on CI result
Ian Schweer
committed
Fix mypy signature
Ian Schweer
committed
Remove unnecessary torch stack
Ian Schweer
committed
Only call .cpu when necessary
Ian Schweer
committed
Recursive false for torch compiler
Ian Schweer
committed
Add elemwise test
Ian Schweer
committed
Late import torch
Ian Schweer
committed
Do iteration instead of vmap for elemwise
Ian Schweer
committed
Clean up and add description

Ch0ronomato
authored and
Ian Schweer
committed
Add unit test to verify iteration

Ch0ronomato
authored and
Ian Schweer
committed
Refactor to ravel method
Ian Schweer
committed
Fix unpacking

authored and
Ian Schweer
committed
Fix comment
Ian Schweer
committed
Remove extra return
Ian Schweer
committed
Update test
Ian Schweer
committed
Add single carry test
Ian Schweer
committed
Remove compiler disable
Ian Schweer
committed
Better name
Ian Schweer
committed
Lint
Ian Schweer
committed
Better docstring
Ian Schweer
committed

Commits on Dec 3, 2024

Fix linker
Ian Schweer
committed

Commits on Dec 8, 2024

Pr comments
Ch0ronomato
committed