[PR #365] [MERGED] Init buffer with mpi4py.MPI.Comm #536

New issue

Closed

opened 2026-10-27 21:03:46 +00:00 by bsakaguchi · 0 comments

bsakaguchi commented

2026-10-27 21:03:46 +00:00

Owner

📋 Pull Request Information

Original PR: https://github.com/deepseek-ai/DeepEP/pull/365
Author: @yuantailing
Created: 8/18/2025
Status: ✅ Merged
Merged: 8/19/2025
Merged by: @sphish

Base: main ← Head: mpi_comm

📝 Commits (1)

bfd39bb Init buffer with mpi4py.MPI.Comm

📊 Changes

1 file changed (+23 additions, -11 deletions)

View changed files

📝 deep_ep/buffer.py (+23 -11)

📄 Description

Buffers can be initialized with either torch.distributed.ProcessGroup or mpi4py.MPI.Comm communicator.

There is no type hint for mpi4py.MPI.Comm because it's not a requirement.

_{🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.}

## 📋 Pull Request Information **Original PR:** https://github.com/deepseek-ai/DeepEP/pull/365 **Author:** [@yuantailing](https://github.com/yuantailing) **Created:** 8/18/2025 **Status:** ✅ Merged **Merged:** 8/19/2025 **Merged by:** [@sphish](https://github.com/sphish) **Base:** `main` ← **Head:** `mpi_comm` --- ### 📝 Commits (1) - [`bfd39bb`](https://github.com/deepseek-ai/DeepEP/commit/bfd39bb47a162326ff466d22876530fa88bf30d4) Init buffer with mpi4py.MPI.Comm ### 📊 Changes **1 file changed** (+23 additions, -11 deletions) <details> <summary>View changed files</summary> 📝 `deep_ep/buffer.py` (+23 -11) </details> ### 📄 Description Buffers can be initialized with either torch.distributed.ProcessGroup or mpi4py.MPI.Comm communicator. There is no type hint for `mpi4py.MPI.Comm` because it's not a requirement. --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>

bsakaguchi

2026-10-27 21:03:46 +00:00