PyTorch inline constants in dispatch to avoid graph breaks#1118

Open

ricardoV94 wants to merge 7 commits intopymc-devs:mainfrom ricardoV94:torch_constant_dispatch

+146-11

Commits on Jan 26, 2025

Split and inverse

ricardoV94
authored and
Ch0ronomato
committed
PyTorch inline constants in dispatch to avoid graph breaks

ricardoV94
authored and
Ch0ronomato
committed
Don't check runtime broadcast

ricardoV94
authored and
Ch0ronomato
committed
Readd bcast check

Ian Schweer
authored and
Ch0ronomato
committed
Add compiler flags to help hint more to torch

Ian Schweer
authored and
Ch0ronomato
committed
Wrong flags
Ch0ronomato
committed
Add split test
Ch0ronomato
committed