Skip to content

Update GPU memory footprint estimation for shared primary context#120

Merged
lohedges merged 5 commits intodevelfrom
feature_primary_context
Feb 4, 2026
Merged

Update GPU memory footprint estimation for shared primary context#120
lohedges merged 5 commits intodevelfrom
feature_primary_context

Conversation

@lohedges
Copy link
Contributor

@lohedges lohedges commented Feb 4, 2026

This PR updates the way that the GPU memory footprint on each device is estimated during repex simulations. This is needed since both openmm and loch use a shared primary context on each device, meaning that the memory footprint for the first context on a device is larger than the subsequent ones. The log now reports the primary and marginal footprint.

I've also updated the code to log version information for ghostly and loch. These will only be displayed when the appropriate configuration options are set.

@lohedges lohedges added enhancement New feature or request cresset Related to work with Cresset labels Feb 4, 2026
@lohedges lohedges merged commit 1896f34 into devel Feb 4, 2026
4 checks passed
@lohedges lohedges deleted the feature_primary_context branch February 4, 2026 10:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cresset Related to work with Cresset enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant