Skip to content

Explicit copying of MPI buffers#102

Merged
haykh merged 9 commits intodev/pgensfrom
dev/cpu_mpi_pointers
Apr 25, 2025
Merged

Explicit copying of MPI buffers#102
haykh merged 9 commits intodev/pgensfrom
dev/cpu_mpi_pointers

Conversation

@LudwigBoess
Copy link
Collaborator

This PR adds explicit copying of MPI buffers from GPU to CPU and back to solve GPU pinning issues on DeltAI

@LudwigBoess LudwigBoess requested review from haykh and jmahlmann April 21, 2025 19:26
@LudwigBoess LudwigBoess self-assigned this Apr 21, 2025
Copy link
Collaborator

Choose a reason for hiding this comment

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

why this fix?

Copy link
Collaborator

Choose a reason for hiding this comment

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

i think with the quotes it should work ok: see 1952708

@haykh
Copy link
Collaborator

haykh commented Apr 25, 2025

I tested it this morning on 4 NVIDIA GPUs and it seems to perform great and also with no bugs. So merging for now.

@haykh haykh merged commit c375dce into dev/pgens Apr 25, 2025
2 checks passed
@haykh haykh deleted the dev/cpu_mpi_pointers branch April 25, 2025 12:37
@haykh haykh restored the dev/cpu_mpi_pointers branch April 25, 2025 22:37
@haykh haykh deleted the dev/cpu_mpi_pointers branch April 28, 2025 10:56
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants