pytensor icon indicating copy to clipboard operation
pytensor copied to clipboard

Numba lapack routines should use int64 for shapes

Open ricardoV94 opened this issue 9 months ago • 0 comments

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

ricardoV94 avatar Mar 21 '25 09:03 ricardoV94