From 94dafda4f0917d572074448369f79ab3931fad91 Mon Sep 17 00:00:00 2001 From: luyang Date: Fri, 10 Jan 2025 15:23:46 +0000 Subject: [PATCH] refine --- oneflow/core/graph/boxing/collective_boxing.proto | 5 ----- .../collective_communication/cuda/cuda_all_to_all.cpp | 2 -- 2 files changed, 7 deletions(-) diff --git a/oneflow/core/graph/boxing/collective_boxing.proto b/oneflow/core/graph/boxing/collective_boxing.proto index c024eb733a2..0f3d01d525c 100644 --- a/oneflow/core/graph/boxing/collective_boxing.proto +++ b/oneflow/core/graph/boxing/collective_boxing.proto @@ -15,11 +15,6 @@ enum OpType { kOpTypeAll2All = 6; } -enum CclCommType{ - kCommTypeCudaNccl = 0; - kCommTypeAscendHccl = 1; -} - enum ReduceMethod { kReduceMethodInvalid = 0; kReduceMethodSum = 1; diff --git a/oneflow/user/kernels/collective_communication/cuda/cuda_all_to_all.cpp b/oneflow/user/kernels/collective_communication/cuda/cuda_all_to_all.cpp index 1dd71fb4c98..9f9c3e31ef1 100644 --- a/oneflow/user/kernels/collective_communication/cuda/cuda_all_to_all.cpp +++ b/oneflow/user/kernels/collective_communication/cuda/cuda_all_to_all.cpp @@ -14,8 +14,6 @@ See the License for the specific language governing permissions and limitations under the License. */ #ifdef WITH_CUDA -#include "oneflow/user/kernels/collective_communication/include/send.h" -#include "oneflow/user/kernels/collective_communication/include/recv.h" #include "oneflow/user/kernels/collective_communication/include/all_to_all.h" #include "oneflow/user/kernels/collective_communication/cuda/cuda_communication_context.h" #include "oneflow/core/device/nccl_util.h"