6565 run : |
6666 sudo apt update -y
6767 sudo bash -x dependencies.sh -y
68- pip install --upgrade pip pybind11
68+ pip install pybind11
6969 mkdir build
7070 cd build
7171 cmake .. -DUSE_HTTP=ON -DUSE_ETCD=ON -DSTORE_USE_ETCD=ON -DENABLE_ASAN=ON -DENABLE_SCCACHE=ON
@@ -104,15 +104,18 @@ jobs:
104104 run : |
105105 cd build
106106 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib
107- # Preload AddressSanitizer library to fix ASan runtime warning
108- export LD_PRELOAD=$(gcc -print-file-name=libasan.so):$LD_PRELOAD
109- # Configure ASan with suppressions for system tools
110- export ASAN_OPTIONS="detect_leaks=1:abort_on_error=1:halt_on_error=1"
111- export LSAN_OPTIONS="suppressions=../.github/asan_suppressions.txt:print_suppressions=0"
112107 ldconfig -v || echo "always continue"
113108 MC_METADATA_SERVER=http://127.0.0.1:8080/metadata DEFAULT_KV_LEASE_TTL=500 make test -j ARGS="-V"
114109 shell : bash
115110
111+ - name : Test CoroRPC Performance
112+ run : |
113+ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib
114+ cd mooncake-transfer-engine/tests
115+ pip install torch numpy
116+ python test_coro_rpc_performance.py
117+ shell : bash
118+
116119 - name : Stop Mooncake Master Service
117120 run : |
118121 pkill mooncake_master || true
@@ -130,34 +133,8 @@ jobs:
130133 PYTHON_VERSION=${{ matrix.python-version }} OUTPUT_DIR=dist-py${{ steps.generate_tag_build.outputs.python_version_tag }} ./scripts/build_wheel.sh
131134 shell : bash
132135
133- - name : Test CoroRPC Performance with built wheel
134- run : |
135- export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib
136- # Preload AddressSanitizer library for Python processes
137- export LD_PRELOAD=$(gcc -print-file-name=libasan.so):$LD_PRELOAD
138- # Configure ASan with suppressions for system tools
139- export ASAN_OPTIONS="detect_leaks=1:abort_on_error=0:halt_on_error=0"
140- export LSAN_OPTIONS="suppressions=.github/asan_suppressions.txt:print_suppressions=0"
141- # Create test environment and install wheel
142- python -m venv coro_test_env
143- source coro_test_env/bin/activate
144- pip install --upgrade pip --quiet
145- pip install torch numpy
146- pip install mooncake-wheel/dist-py${{ steps.generate_tag_build.outputs.python_version_tag }}/*.whl
147- # Copy test file and run
148- cp mooncake-transfer-engine/tests/test_coro_rpc_performance.py .
149- python test_coro_rpc_performance.py
150- deactivate
151- shell : bash
152-
153- - name : Upload Python wheel artifact from build
154- uses : actions/upload-artifact@v4
155- with :
156- name : mooncake-wheel-build-py${{ steps.generate_tag_build.outputs.python_version_tag }}
157- path : mooncake-wheel/dist-py${{ steps.generate_tag_build.outputs.python_version_tag }}/*.whl
158-
159136 test-wheel-ubuntu :
160- needs : build
137+ needs : build-flags
161138 strategy :
162139 matrix :
163140 ubuntu-version : [ubuntu-22.04, ubuntu-24.04]
@@ -180,7 +157,7 @@ jobs:
180157 - name : Download wheel artifact
181158 uses : actions/download-artifact@v4
182159 with :
183- name : mooncake-wheel-build -py${{ steps.generate_tag_test.outputs.python_version_tag }}
160+ name : mooncake-wheel-ubuntu -py${{ steps.generate_tag_test.outputs.python_version_tag }}
184161 path : mooncake-wheel/dist
185162
186163 - name : Verify wheel file exists
@@ -210,17 +187,6 @@ jobs:
210187 deactivate
211188 shell : bash
212189
213- - name : Test CoroRPC Performance in wheel environment
214- run : |
215- source test_env/bin/activate
216- # Upgrade pip to avoid version warnings
217- pip install --upgrade pip --quiet
218- pip install torch numpy
219- cp mooncake-transfer-engine/tests/test_coro_rpc_performance.py .
220- python test_coro_rpc_performance.py
221- deactivate
222- shell : bash
223-
224190 build-flags :
225191 runs-on : ubuntu-22.04
226192 strategy :
@@ -269,7 +235,7 @@ jobs:
269235 run : |
270236 sudo apt update -y
271237 sudo bash -x dependencies.sh -y
272- pip install --upgrade pip pybind11
238+ pip install pybind11
273239 shell : bash
274240
275241 - name : Build transfer engine only
0 commit comments