nikitinvv / tomocupy-stream

3 stars 2 forks source link

[feature request] more dynamic memory allocation #4

Open swelborn opened 3 months ago

swelborn commented 3 months ago

Hello! Nice work on this package.

I am using this inside a service, and scan parameters can change during a beamtime. It would be great if we did not have to reallocate GPU memory when this happens. For example, if ntheta changes from 180->179, or the image size changes, the way I am currently doing this is creating a new GPURecRAM object.

How hard would it be to use existing GPU memory for this? or does this already exist and I am not seeing it. I'm thinking of some kind of .update method in GPURecRAM that would reuse as much memory as possible.

Thanks in advance