Skip to content

Commit b37a8a9

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

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
@@ -145,10 +145,6 @@ void moeSetSignalForCpuStageForTest(MoeLoadBalanceSingleLayerSignal* signal)
145145
template <typename TYPE>
146146
__global__ void zeroExpertTokenCountKernel(MoeLoadBalanceMetaInfo metaInfo, int* const enabled, int* expertTokenCount)
147147
{
148-
if (*enabled == 0)
149-
{
150-
return;
151-
}
152148
TYPE oldExpertTokenCount = {0};
153149
int* expertTokenCountPtr = expertTokenCount + metaInfo.expertCount * blockIdx.x;
154150
TYPE* typedExpertTokenCountPtr = reinterpret_cast<TYPE*>(expertTokenCountPtr);

0 commit comments

Comments
 (0)