Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

VecMem Update, main branch (2024.10.29.) #757

Merged
merged 2 commits into from
Oct 29, 2024

Conversation

krasznaa
Copy link
Member

Updated the project to using vecmem-1.11.0.

This update, most importantly, includes: acts-project/vecmem#299 Which was made by @stephenswat to get the memory usage of our code under control. It also includes acts-project/vecmem#286, which is the reason for all the actual code changes in the PR...

Currently traccc_throughput_st_cuda uses the following amount of memory to reconstruct 100 mu=140 events:

current_mu140

So, about 3.85 GB at its peak.

With this update included, this looks like:

newvecmem_mu140

About 2.77 GB at its peak.

To complete the picture, this is what I get when using vecmem::pool_memory_resource instead of vecmem::binary_page_memory_resource:

pool_mu140

About 3.3 GB at its peak.

And finally, if I turn off memory caching completely, I get:

nocache_mu140

I.e. 200-700 MB of memory usage.

In all these cases, on the RTX 2060 GPU of my desktop machine, the throughput of reconstructing mu=140 events in a single thread/stream remained in +-10%. Without caching being the slowest of course, and vecmem::pool_memory_resource (which is based on CUB and std::pmr::unsynchronized_pool_resource, being the fastest. But only by little. Apparently with the full chain running, the caching is not dominating the performance at the moment. 🤔

@krasznaa krasznaa added build This relates to the build system tests Make sure the code keeps working cuda Changes related to CUDA sycl Changes related to SYCL cpu Changes related to CPU code kokkos Changes related to Kokkos alpaka Changes related to Alpaka labels Oct 29, 2024
Copy link

@beomki-yeo
Copy link
Contributor

Great! I will let @stephenswat take a look into this for detailed review

@stephenswat
Copy link
Member

Closes #645.

Copy link
Member

@stephenswat stephenswat left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Happy to see the memory usage drop, and happy to approve this PR.

@krasznaa krasznaa merged commit 8dd31a4 into acts-project:main Oct 29, 2024
26 checks passed
@krasznaa krasznaa deleted the VecMemUpdate-main-20241029 branch October 29, 2024 14:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
alpaka Changes related to Alpaka build This relates to the build system cpu Changes related to CPU code cuda Changes related to CUDA kokkos Changes related to Kokkos sycl Changes related to SYCL tests Make sure the code keeps working
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants