pytensor
pytensor copied to clipboard
Numba lapack routines should use int64 for shapes
Description
Unless this is some lapack constraint?
int64 is the default dtype for numpy arrays, downcasting to int32 could lead to surprising errors when working with huge arrays?
https://github.com/pymc-devs/pytensor/blob/a149f6c996bd0499406150853bef517e7d7a3c4f/pytensor/link/numba/dispatch/slinalg.py#L125