Skip to content

Commit b10c1d9

Browse files
dongxuy04syuoni
authored andcommitted
[None][fix] fix EPLB init hang (#8649)
Signed-off-by: Dongxu Yang <[email protected]>
1 parent b334102 commit b10c1d9

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

cpp/tensorrt_llm/kernels/moeLoadBalance/moeLoadBalanceKernels.cu

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -131,10 +131,6 @@ void moeSetSignalForCpuStageForTest(MoeLoadBalanceSingleLayerSignal* signal)
131131
template <typename TYPE>
132132
__global__ void zeroExpertTokenCountKernel(MoeLoadBalanceMetaInfo metaInfo, int* const enabled, int* expertTokenCount)
133133
{
134-
if (*enabled == 0)
135-
{
136-
return;
137-
}
138134
TYPE oldExpertTokenCount = {0};
139135
int* expertTokenCountPtr = expertTokenCount + metaInfo.expertCount * blockIdx.x;
140136
TYPE* typedExpertTokenCountPtr = reinterpret_cast<TYPE*>(expertTokenCountPtr);

0 commit comments

Comments
 (0)