Pull Request №447 RosettaCommons/RFdiffusion/main ← haoyu-haoyu/RFdiffusion/perf/reduce-gpu-memory-usage
Merge: 9535f1938203a24937d7dadf0cb831d02cb5fc0e←5735aff0e6d512ffa26a4d787613820c2c9bf16f
perf: reduce GPU memory usage during inference
----------------
Merge commit message:
perf: reduce GPU memory usage during inference
- Move trajectory data (px0, x_t, seq, plddt) to CPU immediately after
each denoising step instead of accumulating on GPU. This frees GPU
memory for the next forward pass, reducing peak memory usage
proportional to the number of diffusion steps.
- Remove two unused ComputeAllAtomCoords() instantiations in
get_next_ca() and get_next_pose() that were created every timestep
but never referenced, wasting memory and compute.