search

tianjuxue / jax-am

Additive manufacturing simulation with JAX.
https://jax-am.readthedocs.io/en/latest/
GNU General Public License v3.0
268 stars 56 forks source link

Implement guards against large matrices entering GPU memory #21

Open SNMS95 opened 1 year ago

SNMS95 commented 1 year ago

For now, we have used device_put() to put the global stiffness matrix into CPU. Need to create a transfer guard to ensure that this is indeed happening.