[https://nvbugs/5422621][fix] fix EPLB init hang (cherry-pick #8649) (#8727)

Signed-off-by: Dongxu Yang <[email protected]>
Signed-off-by: Enwei Zhu <[email protected]>
Co-authored-by: dongxuy04 <[email protected]>

Author: syuoni

cpp/tensorrt_llm/kernels/moeLoadBalance/moeLoadBalanceKernels.cu

@@ -131,10 +131,6 @@ void moeSetSignalForCpuStageForTest(MoeLoadBalanceSingleLayerSignal* signal)
 template <typename TYPE>
 __global__ void zeroExpertTokenCountKernel(MoeLoadBalanceMetaInfo metaInfo, int* const enabled, int* expertTokenCount)
 {
-    if (*enabled == 0)
-    {
-        return;
-    }
     TYPE oldExpertTokenCount = {0};
     int* expertTokenCountPtr = expertTokenCount + metaInfo.expertCount * blockIdx.x;
     TYPE* typedExpertTokenCountPtr = reinterpret_cast<TYPE*>(expertTokenCountPtr);