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
|
else
|
||||||
NVRTC_SPEC="nvidia-cuda-nvrtc-cu12"
|
NVRTC_SPEC="nvidia-cuda-nvrtc-cu12"
|
||||||
fi
|
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
|
# Install other test dependencies
|
||||||
if [ "$IS_BLACKWELL" != "1" ]; then
|
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()
|
ib_devices = get_rdma_devices_args()
|
||||||
|
|
||||||
|
|
||||||
@unittest.skipIf(is_in_ci(), "Skip since mooncake-ep is flaky.")
|
|
||||||
class TestTP(CustomTestCase):
|
class TestTP(CustomTestCase):
|
||||||
extra_args = []
|
extra_args = []
|
||||||
|
|
||||||
@@ -84,7 +83,6 @@ class TestTP(CustomTestCase):
|
|||||||
self.assertGreater(metrics["accuracy"], 0.60)
|
self.assertGreater(metrics["accuracy"], 0.60)
|
||||||
|
|
||||||
|
|
||||||
@unittest.skipIf(is_in_ci(), "Skip since mooncake-ep is flaky.")
|
|
||||||
class TestPureDP(TestTP):
|
class TestPureDP(TestTP):
|
||||||
extra_args = [
|
extra_args = [
|
||||||
"--enable-dp-attention",
|
"--enable-dp-attention",
|
||||||
|
|||||||
Reference in New Issue
Block a user