Skip to content

Commit a016275

Browse files
authored
Revert "[Feature] Support noaux for eplb (#5143)"
This reverts commit 6ca2651.
1 parent 34f59d9 commit a016275

File tree

8 files changed

+23
-616
lines changed

8 files changed

+23
-616
lines changed

custom_ops/gpu_ops/cpp_extensions.cc

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -647,19 +647,6 @@ std::vector<paddle::Tensor> NoauxTc(paddle::Tensor& scores,
647647
bool renormalize,
648648
float routed_scaling_factor);
649649

650-
std::vector<paddle::Tensor> NoauxTcRedundant(
651-
paddle::Tensor& scores,
652-
paddle::Tensor& scores_with_bias,
653-
paddle::Tensor& expert_id_to_ep_rank_array,
654-
paddle::Tensor& expert_in_rank_num_list,
655-
paddle::Tensor& tokens_per_expert_stats_list,
656-
int n_group,
657-
int topk_group,
658-
int topk,
659-
bool renormalize,
660-
float routed_scaling_factor,
661-
int redundant_ep_rank_num_plus_one);
662-
663650
#ifdef ENABLE_FP8
664651
paddle::Tensor cutlass_fp8_fp8_half_gemm_func(
665652
const paddle::Tensor& x,
@@ -1498,10 +1485,6 @@ PYBIND11_MODULE(fastdeploy_ops, m) {
14981485

14991486
m.def("noaux_tc", &NoauxTc, "noaux_tc for Deepseekv3 MoE compute");
15001487

1501-
m.def("noaux_tc_redundant",
1502-
&NoauxTcRedundant,
1503-
"noaux_tc_redundant for MoE compute");
1504-
15051488
#ifdef ENABLE_FP8
15061489
m.def("cutlass_fp8_fp8_half_gemm_fused",
15071490
&cutlass_fp8_fp8_half_gemm_func,

custom_ops/gpu_ops/noaux_tc_redundant.cu

Lines changed: 0 additions & 103 deletions
This file was deleted.

0 commit comments

Comments
 (0)