From 0e73dc86b01619827f495a566c82108d31a9a0b8 Mon Sep 17 00:00:00 2001
From: jiarundong <j30079061@china.huawei.com>
Date: Fri, 15 Aug 2025 10:20:48 +0800
Subject: [PATCH 1/5] build deepep

---
 .github/workflows/pr-test-npu.yml | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/.github/workflows/pr-test-npu.yml b/.github/workflows/pr-test-npu.yml
index b9e37a29e..4395f7a2c 100644
--- a/.github/workflows/pr-test-npu.yml
+++ b/.github/workflows/pr-test-npu.yml
@@ -39,6 +39,9 @@ jobs:
       - name: Start CI container
         run: bash scripts/npu_ci_a3_start_container.sh
 
+      - name: Prepare Deepep
+        run: bash scripts/prepare_deepep_in_container.sh
+
       - name: Run test
         timeout-minutes: 10
         run: |
@@ -69,6 +72,9 @@ jobs:
       - name: Start CI container
         run: bash scripts/npu_ci_a3_start_container.sh
 
+      - name: Prepare Deepep
+        run: bash scripts/prepare_deepep_in_container.sh
+
       - name: Run test
         timeout-minutes: 10
         run: |

From a15ecf0777c256f11debd1e1a423d6eb6812d12d Mon Sep 17 00:00:00 2001
From: jiarundong <j30079061@china.huawei.com>
Date: Fri, 15 Aug 2025 10:21:10 +0800
Subject: [PATCH 2/5] build deepep

---
 scripts/prepare_deepep_in_container.sh | 4 ++++
 1 file changed, 4 insertions(+)
 create mode 100644 scripts/prepare_deepep_in_container.sh

diff --git a/scripts/prepare_deepep_in_container.sh b/scripts/prepare_deepep_in_container.sh
new file mode 100644
index 000000000..831bc1377
--- /dev/null
+++ b/scripts/prepare_deepep_in_container.sh
@@ -0,0 +1,4 @@
+ENV_DIR="/home/runner/.cache/env_sglang.sh"
+docker exec \
+  sglang_kernel_ci_a3 \
+  /bin/bash -c "source \"$ENV_DIR\" && cd ${GITHUB_WORKSPACE} &&  bash build.sh && pip install ${GITHUB_WORKSPACE}/output/deep_ep*.whl --no-cache-dir"

From 2954d373ae74ea717b435a7c63f7ff5d6873d3cb Mon Sep 17 00:00:00 2001
From: jia-rundong <74397815+jia-rundong@users.noreply.github.com>
Date: Fri, 15 Aug 2025 14:45:00 +0800
Subject: [PATCH 3/5] test build deepep

---
 python/deep_ep/deep_ep/buffer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/deep_ep/deep_ep/buffer.py b/python/deep_ep/deep_ep/buffer.py
index 5890fb234..1f36e0cfb 100644
--- a/python/deep_ep/deep_ep/buffer.py
+++ b/python/deep_ep/deep_ep/buffer.py
@@ -18,6 +18,8 @@ def __init__(self, group: dist.ProcessGroup,
         """
         Initialize the communication buffer.
 
+        ############ test build
+
         Arguments:
             group: the communication group.
             num_nvl_bytes: the buffer size for intranode NVLink communication.

From bf7734147386b5cedbc9e82e059c09bb4bfc8150 Mon Sep 17 00:00:00 2001
From: jia-rundong <74397815+jia-rundong@users.noreply.github.com>
Date: Fri, 15 Aug 2025 14:50:55 +0800
Subject: [PATCH 4/5] revert test code

---
 python/deep_ep/deep_ep/buffer.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/python/deep_ep/deep_ep/buffer.py b/python/deep_ep/deep_ep/buffer.py
index 1f36e0cfb..5890fb234 100644
--- a/python/deep_ep/deep_ep/buffer.py
+++ b/python/deep_ep/deep_ep/buffer.py
@@ -18,8 +18,6 @@ def __init__(self, group: dist.ProcessGroup,
         """
         Initialize the communication buffer.
 
-        ############ test build
-
         Arguments:
             group: the communication group.
             num_nvl_bytes: the buffer size for intranode NVLink communication.

From 4587175bb5c294daf118b2f608407b7894f1642b Mon Sep 17 00:00:00 2001
From: jia-rundong <74397815+jia-rundong@users.noreply.github.com>
Date: Tue, 19 Aug 2025 09:31:17 +0800
Subject: [PATCH 5/5] link deepep so

---
 scripts/prepare_deepep_in_container.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/prepare_deepep_in_container.sh b/scripts/prepare_deepep_in_container.sh
index 831bc1377..190055d14 100644
--- a/scripts/prepare_deepep_in_container.sh
+++ b/scripts/prepare_deepep_in_container.sh
@@ -1,4 +1,4 @@
 ENV_DIR="/home/runner/.cache/env_sglang.sh"
 docker exec \
   sglang_kernel_ci_a3 \
-  /bin/bash -c "source \"$ENV_DIR\" && cd ${GITHUB_WORKSPACE} &&  bash build.sh && pip install ${GITHUB_WORKSPACE}/output/deep_ep*.whl --no-cache-dir"
+  /bin/bash -c "source \"$ENV_DIR\" && cd ${GITHUB_WORKSPACE} && bash build.sh && pip install ${GITHUB_WORKSPACE}/output/deep_ep*.whl --no-cache-dir" && cd "$(pip show deep-ep | awk '/^Location:/ {print $2}')" && ln -s deep_ep/deep_ep_cpp*.so