You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi! I have a model that requires precomputing a large matrix that is a model constant. To avoid having to compute it at every function call, I thought I'd make it evaluate at compile time. However, this results in a segmentation fault. The same does not happen if I do not ask for compile-time evaluation. I'm not sure if this is a bug or some deeper issue I don't fully understand. This is related to the following question I posed on the Q&A #18830.
Here's a minimal (a bit absurdly minimal admittedly) reproducable example
This also results in a segmentation fault. Is there a better way of passing large constants? The key issue is that its matrix, so passing it as a static argument to jit also doesn't seem to work.
Description
Hi! I have a model that requires precomputing a large matrix that is a model constant. To avoid having to compute it at every function call, I thought I'd make it evaluate at compile time. However, this results in a segmentation fault. The same does not happen if I do not ask for compile-time evaluation. I'm not sure if this is a bug or some deeper issue I don't fully understand. This is related to the following question I posed on the Q&A #18830.
Here's a minimal (a bit absurdly minimal admittedly) reproducable example
the output is simply a segmentation fault. I'm on a Mac Studio M2 (no jax metal, because too many things are broken)
What jax/jaxlib version are you using?
0.4.20 0.4.20
Which accelerator(s) are you using?
CPU
Additional system info?
1.26.2 3.10.8 | packaged by conda-forge | (main, Nov 22 2022, 08:25:29) [Clang 14.0.6 ] uname_result(system='Darwin', node='Bvatter-Studio-23', release='23.1.0', version='Darwin Kernel Version 23.1.0: Mon Oct 9 21:28:45 PDT 2023; root:xnu-10002.41.9~6/RELEASE_ARM64_T6020', machine='arm64')
NVIDIA GPU info
No response
The text was updated successfully, but these errors were encountered: