[PR #410] [MERGED] Canonicalize TMA usages #555

New issue

Closed

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

bsakaguchi commented

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

Owner

📋 Pull Request Information

Original PR: https://github.com/deepseek-ai/DeepEP/pull/410
Author: @LyricZhao
Created: 9/16/2025
Status: ✅ Merged
Merged: 9/16/2025
Merged by: @LyricZhao

Base: main ← Head: elect-one-sync

📝 Commits (8)

691803a Remove redundant TMA flushes
ff5ca40 Less barrier initialization overhead
0ba50a3 Simplify elect_one_sync
97d3b41 Use elect_one_sync instead of lanes
f73b88d Minor fix
2bd032c Polish testing prints
e408d1e Refactor for internode kernels
2cd957e Better performance

📊 Changes

6 files changed (+140 additions, -140 deletions)

View changed files

📝 csrc/kernels/internode.cu (+49 -44)
📝 csrc/kernels/internode_ll.cu (+9 -14)
📝 csrc/kernels/intranode.cu (+51 -59)
📝 csrc/kernels/utils.cuh (+16 -17)
📝 tests/test_internode.py (+13 -4)
📝 tests/test_intranode.py (+2 -2)

📄 Description

No description provided

_{🔄 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/410 **Author:** [@LyricZhao](https://github.com/LyricZhao) **Created:** 9/16/2025 **Status:** ✅ Merged **Merged:** 9/16/2025 **Merged by:** [@LyricZhao](https://github.com/LyricZhao) **Base:** `main` ← **Head:** `elect-one-sync` --- ### 📝 Commits (8) - [`691803a`](https://github.com/deepseek-ai/DeepEP/commit/691803ad2dc996659a29b94a79aadc813a12bcb3) Remove redundant TMA flushes - [`ff5ca40`](https://github.com/deepseek-ai/DeepEP/commit/ff5ca407f95f3b2be76953f2508c520da64cd20a) Less barrier initialization overhead - [`0ba50a3`](https://github.com/deepseek-ai/DeepEP/commit/0ba50a3c6c457f55a27b4d1da201e222920be93d) Simplify `elect_one_sync` - [`97d3b41`](https://github.com/deepseek-ai/DeepEP/commit/97d3b417cc30cf2e9a1286a9a245de23f283b847) Use `elect_one_sync` instead of lanes - [`f73b88d`](https://github.com/deepseek-ai/DeepEP/commit/f73b88d345362869ea974cc864a0a7a645e0dd00) Minor fix - [`2bd032c`](https://github.com/deepseek-ai/DeepEP/commit/2bd032cb6e045fa3ada241a5c44f51235b70c0bf) Polish testing prints - [`e408d1e`](https://github.com/deepseek-ai/DeepEP/commit/e408d1e00bd19010b1dd6ebe9f58dd1fc0d51c33) Refactor for internode kernels - [`2cd957e`](https://github.com/deepseek-ai/DeepEP/commit/2cd957ec3162cd3bd45b46946996979ae46c78c6) Better performance ### 📊 Changes **6 files changed** (+140 additions, -140 deletions) <details> <summary>View changed files</summary> 📝 `csrc/kernels/internode.cu` (+49 -44) 📝 `csrc/kernels/internode_ll.cu` (+9 -14) 📝 `csrc/kernels/intranode.cu` (+51 -59) 📝 `csrc/kernels/utils.cuh` (+16 -17) 📝 `tests/test_internode.py` (+13 -4) 📝 `tests/test_intranode.py` (+2 -2) </details> ### 📄 Description _No description provided_ --- <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:59 +00:00