Skip to content

Optionally turn off GPU-aware MPI#105

Merged
haykh merged 2 commits intodev/pgensfrom
dev/gpu-aware-mpi
May 1, 2025
Merged

Optionally turn off GPU-aware MPI#105
haykh merged 2 commits intodev/pgensfrom
dev/gpu-aware-mpi

Conversation

@haykh
Copy link
Collaborator

@haykh haykh commented Apr 28, 2025

Flag -D gpu_aware_mpi added which is ON by default. when not compiled with a GPU support, this flag has no effect.

Also added a more verbose .info which prints all compilation flags.

@haykh haykh added the enhancement New feature or request label Apr 28, 2025
@haykh
Copy link
Collaborator Author

haykh commented Apr 28, 2025

@LudwigBoess could you test this on Delta? On stellar it works great! The problem with previous implementation was that we were adding offsets to CPU pointers which is not allowed (since their shape is weird even when you explicitly mirror it).

@LudwigBoess
Copy link
Collaborator

It works great on DeltaAI too, so I think we can merge this

@haykh haykh merged commit af74825 into dev/pgens May 1, 2025
2 checks passed
@haykh haykh deleted the dev/gpu-aware-mpi branch May 1, 2025 05:59
@haykh haykh mentioned this pull request May 1, 2025
36 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants