[PR #356] [MERGED] fix ll test script when --num-tokens == 1 #531

New issue

Closed

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

bsakaguchi commented

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

Owner

📋 Pull Request Information

Original PR: https://github.com/deepseek-ai/DeepEP/pull/356
Author: @alpha-baby
Created: 8/8/2025
Status: ✅ Merged
Merged: 8/8/2025
Merged by: @sphish

Base: main ← Head: fujh/fix_test_low_latency

📝 Commits (1)

5f44caa fix when --num-tokens == 1

📊 Changes

1 file changed (+2 additions, -0 deletions)

View changed files

📝 tests/test_low_latency.py (+2 -0)

📄 Description

test command:

NCCL_DEBUG=WARN NVSHMEM_DEBUG=WARN python3 ./tests/test_low_latency_bak.py --num-tokens 20 --hidden 2048 --num-topk 1 --num-experts 8

assert : calc_diff(recv_x[:, -1], recv_src_info.view(-1)) is nan when --num-tokens == 1

_{🔄 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/356 **Author:** [@alpha-baby](https://github.com/alpha-baby) **Created:** 8/8/2025 **Status:** ✅ Merged **Merged:** 8/8/2025 **Merged by:** [@sphish](https://github.com/sphish) **Base:** `main` ← **Head:** `fujh/fix_test_low_latency` --- ### 📝 Commits (1) - [`5f44caa`](https://github.com/deepseek-ai/DeepEP/commit/5f44caa457d546053e4f17a4c7beacd169c51651) fix when --num-tokens == 1 ### 📊 Changes **1 file changed** (+2 additions, -0 deletions) <details> <summary>View changed files</summary> 📝 `tests/test_low_latency.py` (+2 -0) </details> ### 📄 Description test command: ``` NCCL_DEBUG=WARN NVSHMEM_DEBUG=WARN python3 ./tests/test_low_latency_bak.py --num-tokens 20 --hidden 2048 --num-topk 1 --num-experts 8 ``` assert : `calc_diff(recv_x[:, -1], recv_src_info.view(-1))` is nan when `--num-tokens == 1` --- <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:44 +00:00