chore: bump mooncake version to 0.3.10 (#20942)
Signed-off-by: Shangming Cai <csmthu@gmail.com>
This commit is contained in:
@@ -298,7 +298,7 @@ if [ "$CU_VERSION" = "cu130" ]; then
|
||||
else
|
||||
NVRTC_SPEC="nvidia-cuda-nvrtc-cu12"
|
||||
fi
|
||||
$PIP_CMD install mooncake-transfer-engine==0.3.9 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX
|
||||
$PIP_CMD install mooncake-transfer-engine==0.3.10 "${NVRTC_SPEC}" py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX
|
||||
|
||||
# Install other test dependencies
|
||||
if [ "$IS_BLACKWELL" != "1" ]; then
|
||||
|
||||
@@ -20,7 +20,6 @@ register_cuda_ci(est_time=660, suite="stage-c-test-deepep-4-gpu-h100")
|
||||
ib_devices = get_rdma_devices_args()
|
||||
|
||||
|
||||
@unittest.skipIf(is_in_ci(), "Skip since mooncake-ep is flaky.")
|
||||
class TestTP(CustomTestCase):
|
||||
extra_args = []
|
||||
|
||||
@@ -84,7 +83,6 @@ class TestTP(CustomTestCase):
|
||||
self.assertGreater(metrics["accuracy"], 0.60)
|
||||
|
||||
|
||||
@unittest.skipIf(is_in_ci(), "Skip since mooncake-ep is flaky.")
|
||||
class TestPureDP(TestTP):
|
||||
extra_args = [
|
||||
"--enable-dp-attention",
|
||||
|
||||
Reference in New Issue
Block a user