fix lint problems

Author: luanyundu

csrc/deepep/deep_ep.cpp

@@ -86,19 +86,22 @@ Buffer::get_dispatch_layout(const torch::Tensor &topk_idx, int num_experts, std:
     auto is_token_in_rank = at::zeros({num_tokens, num_ranks}, at::dtype(at::kInt).device(device));
     const int notify_send_data_size =
         num_experts * EXPERT_DATA_SIZE + server_num + MAX_BATCH_SIZE * (1 + 2 * server_num + num_topk);
-    /*The output parameters are ordered as follows
-      1. the number of the tokens that every expert received from this NPU.                 size:[numExpert]
-      2. The number of tokens received by each server from this NPU (deduplicated).	        size:[serverNum]
-      3. The number of tokens sent from this NPU to each server (without deduplication).    size:[MAX_BS, serverNum]
-      4. The number of servers each token is sent to by this NPU.                           size:[MAX_BS]
-      5. The order in which each token of this NPU is sent to various servers.				size:[MAX_BS, serverNum]
-      6. The order in which each token is sent to the expert.							    size:[MAX_BS, numTopk]
-      7. The server offset of tokens received by each expert from this NPU.					size:[numExpert, MAX_BS]
-      8. The origin offset of the token received by each expert on the original NPU.		size:[numExpert, MAX_BS]
-      */
+    /*
+    The output parameters are ordered as follows:
+    1. the number of the tokens that every expert received from this NPU.                size:[numExpert]
+    2. The number of tokens received by each server from this NPU (deduplicated).        size:[serverNum]
+    3. The number of tokens sent from this NPU to each server (without deduplication).   size:[MAX_BS, serverNum]
+    4. The number of servers each token is sent to by this NPU.                          size:[MAX_BS]
+    5. The order in which each token of this NPU is sent to various servers.             size:[MAX_BS, serverNum]
+    6. The order in which each token is sent to the expert.                              size:[MAX_BS, numTopk]
+    7. The server offset of tokens received by each expert from this NPU.                size:[numExpert, MAX_BS]
+    8. The origin offset of the token received by each expert on the original NPU.       size:[numExpert, MAX_BS]
+    */
     auto notify_send_data = at::zeros({notify_send_data_size}, at::dtype(at::kInt).device(device));
-    notify_send_data.index({at::indexing::Slice(num_experts + server_num + MAX_BATCH_SIZE * (server_num + 1),
-                       num_experts + server_num + MAX_BATCH_SIZE * (server_num * 2 + 1))}).fill_(-1);
+    notify_send_data
+        .index({at::indexing::Slice(num_experts + server_num + MAX_BATCH_SIZE * (server_num + 1),
+                                    num_experts + server_num + MAX_BATCH_SIZE * (server_num * 2 + 1))})
+        .fill_(-1);
     // The order of each token sent to the server is set to -1.
     EXEC_NPU_CMD(aclnnDispatchLayout, new_topk_idx, num_tokens, num_ranks, num_experts, num_topk, local_ranksize,
                  num_tokens_per_rank, num_tokens_per_expert, is_token_in_rank, notify_send_data);

csrc/deepep/ops/op_kernel/dispatch_layout.h

@@ -67,11 +67,10 @@ class DispatchLayout
             isTokenOffset = (restNum + coreIdx_ * tempTokens_) * numRanks_ * sizeof(T);
         }
 
-            topkIdxGM_.SetGlobalBuffer((__gm__ int64_t *)(topkIdx + topkIdxOffset));
-            numTokensPerRankGM_.SetGlobalBuffer((__gm__ T *)numTokensPerRank);
-            numTokensPerExpertGM_.SetGlobalBuffer((__gm__ T *)numTokensPerExpert);
-            isTokenInRankGM_.SetGlobalBuffer((__gm__ T *)(isTokenInRank + isTokenOffset));
-        }
+        topkIdxGM_.SetGlobalBuffer((__gm__ int64_t *)(topkIdx + topkIdxOffset));
+        numTokensPerRankGM_.SetGlobalBuffer((__gm__ T *)numTokensPerRank);
+        numTokensPerExpertGM_.SetGlobalBuffer((__gm__ T *)numTokensPerExpert);
+        isTokenInRankGM_.SetGlobalBuffer((__gm__ T *)(isTokenInRank + isTokenOffset));
     }
 
     __aicore__ inline void Process()