xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-16 06:15:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Refactor] Rename commnication utils (#20091)

Browse Source 
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Wentao Ye 2025-06-26 18:19:32 -04:00 committed by GitHub
parent 04e1642e32
commit 562308816c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/kernels/moe/test_deepep_deepgemm_moe.py
View File

@ -22,7 +22,7 @@ from vllm.model_executor.layers.quantization.utils.fp8_utils import (
per_token_group_quant_fp8) per_token_group_quant_fp8)
from vllm.platforms import current_platform from vllm.platforms import current_platform
from .deepep_utils import ProcessGroupInfo, parallel_launch from .utils import ProcessGroupInfo, parallel_launch
has_deep_ep = importlib.util.find_spec("deep_ep") is not None has_deep_ep = importlib.util.find_spec("deep_ep") is not None

2
tests/kernels/moe/test_deepep_moe.py
View File

@ -23,7 +23,7 @@ from vllm.model_executor.layers.quantization.utils.fp8_utils import (
per_token_group_quant_fp8) per_token_group_quant_fp8)
from vllm.platforms import current_platform from vllm.platforms import current_platform
from .deepep_utils import ProcessGroupInfo, parallel_launch from .utils import ProcessGroupInfo, parallel_launch
has_deep_ep = importlib.util.find_spec("deep_ep") is not None has_deep_ep = importlib.util.find_spec("deep_ep") is not None

2
tests/kernels/moe/test_pplx_cutlass_moe.py
View File

@ -15,7 +15,7 @@ from vllm.model_executor.layers.fused_moe.modular_kernel import (
FusedMoEModularKernel) FusedMoEModularKernel)
from vllm.platforms import current_platform from vllm.platforms import current_platform
from .deepep_utils import ProcessGroupInfo, parallel_launch from .utils import ProcessGroupInfo, parallel_launch
try: try:
from pplx_kernels import AllToAll from pplx_kernels import AllToAll

2
tests/kernels/moe/test_pplx_moe.py
View File

@ -29,7 +29,7 @@ from vllm.model_executor.layers.fused_moe.modular_kernel import (
FusedMoEModularKernel) FusedMoEModularKernel)
from vllm.platforms import current_platform from vllm.platforms import current_platform
from .deepep_utils import ProcessGroupInfo, parallel_launch from .utils import ProcessGroupInfo, parallel_launch
requires_pplx = pytest.mark.skipif( requires_pplx = pytest.mark.skipif(
not has_pplx, not has_pplx,

0
tests/kernels/moe/deepep_utils.py → tests/kernels/moe/utils.py
View File