Merge pull request #706 from InfiniTensor/issue/704

issue/704 - Add ccl supprt for maca with mc api

src/infiniccl/metax/infiniccl_metax.cc

@@ -2,8 +2,13 @@
 
 #include "../../utils.h"
 
+#ifdef ENABLE_METAX_MC_API
+#include <mccl.h>
+#include <mcr/mc_runtime_api.h>
+#else
 #include <hccl.h>
 #include <hcr/hc_runtime_api.h>
+#endif
 
 #include <iostream>
 #include <vector>

src/infiniccl/metax/infiniccl_metax.h

 #ifndef INFINICCL_METAX_H_
 #define INFINICCL_METAX_H_
 
+#include "../../infiniop/devices/metax/metax_ht2mc.h"
 #include "../infiniccl_impl.h"
 
 #if defined(ENABLE_METAX_API) && defined(ENABLE_CCL)

src/infiniop/devices/metax/metax_ht2mc.h

@@ -70,4 +70,19 @@
 #define HCBLAS_GEMM_DEFAULT_TENSOR_OP MCBLAS_GEMM_DEFAULT_TENSOR_OP
 #define hcblasGemmStridedBatchedEx mcblasGemmStridedBatchedEx
 #define hcLaunchKernel mcLaunchKernel
+#define hcclDataType_t mcclDataType_t
+#define hcclRedOp_t mcclRedOp_t
+#define hcclComm_t mcclComm_t
+#define hcclFloat mcclFloat
+#define hcclHalf mcclHalf
+#define hcclBfloat16 mcclBfloat16
+#define hcclSum mcclSum
+#define hcclProd mcclProd
+#define hcclMax mcclMax
+#define hcclMin mcclMin
+#define hcclAvg mcclAvg
+#define hcclCommInitAll mcclCommInitAll
+#define hcclSuccess mcclSuccess
+#define hcclCommDestroy mcclCommDestroy
+#define hcclAllReduce mcclAllReduce
 #endif