diff --git a/fastdeploy/model_executor/layers/moe/fused_moe_deepgemm_backend.py b/fastdeploy/model_executor/layers/moe/fused_moe_deepgemm_backend.py index 5bbce2982..f04b35b7b 100644 --- a/fastdeploy/model_executor/layers/moe/fused_moe_deepgemm_backend.py +++ b/fastdeploy/model_executor/layers/moe/fused_moe_deepgemm_backend.py @@ -571,6 +571,6 @@ class DeepGemmFusedMoeMethod(MoEMethodBase): 1.0, )[0] if layer.tp_size > 1: - tensor_model_parallel_all_reduce(tmp_ffn_out, self.tp_group) + tensor_model_parallel_all_reduce(tmp_ffn_out) return tmp_ffn_out diff --git a/tests/cov_pytest.ini b/tests/cov_pytest.ini index ee3c7e915..79e84d165 100644 --- a/tests/cov_pytest.ini +++ b/tests/cov_pytest.ini @@ -22,7 +22,6 @@ addopts = --ignore=operators/test_flash_mask_attn.py --ignore=operators/test_w4afp8_gemm.py --ignore=operators/test_tree_mask.py - --ignore=model_loader/test_common_model.py --cov=fastdeploy --cov-branch --cov-append