rebase metax maca support to main

Author: crapromer

src/infiniop/ops/causal_softmax/metax/causal_softmax_metax.maca

@@ -1,7 +1,11 @@
 #include "../../../devices/metax/metax_common.h"
 #include "causal_softmax_metax.h"
 
-#include <hccub/block/block_reduce.cuh>
+#ifdef ENABLE_METAX_MC_API
+    #include <cub/block/block_reduce.cuh>
+#else
+    #include <hccub/block/block_reduce.cuh>
+#endif
 #include "../../../devices/metax/metax_kernel_common.h"
 
 #include "../../../reduce/cuda/reduce.cuh"

src/infiniop/ops/random_sample/metax/random_sample_kernel.h

@@ -1,8 +1,15 @@
 #include "../../../devices/metax/metax_kernel_common.h"
 #include "infinicore.h"
-#include <hccub/device/device_radix_sort.cuh>
-#include <hccub/device/device_reduce.cuh>
-#include <hccub/device/device_scan.cuh>
+
+#ifdef ENABLE_METAX_MC_API
+    #include <cub/device/device_radix_sort.cuh>
+    #include <cub/device/device_reduce.cuh>
+    #include <cub/device/device_scan.cuh>
+#else
+    #include <cub/device/device_radix_sort.cuh>
+    #include <cub/device/device_reduce.cuh>
+    #include <cub/device/device_scan.cuh>
+#endif
 
 namespace op::random_sample::metax {
 

src/infiniop/ops/random_sample/metax/random_sample_metax.h

@@ -1,6 +1,5 @@
 #ifndef __RANDOM_SAMPLE_METAX_H__
 #define __RANDOM_SAMPLE_METAX_H__
-
 #include "../random_sample.h"
 
 DESCRIPTOR(metax)

src/infiniop/ops/random_sample/metax/random_sample_metax.maca

@@ -1,3 +1,4 @@
+#include "../../../devices/metax/metax_ht2mc.h"
 #include "../../../devices/metax/metax_common.h"
 #include "../../../devices/metax/metax_handle.h"
 #include "../info.h"

src/infiniop/ops/softplus/operator.cc

@@ -8,9 +8,6 @@
 #if defined(ENABLE_NVIDIA_API) || defined(ENABLE_ILUVATAR_API)
 #include "nvidia/softplus_nvidia.cuh"
 #endif
-#ifdef ENABLE_METAX_API
-#include "metax/softplus_metax.h"
-#endif
 
 __C infiniStatus_t infiniopCreateSoftplusDescriptor(
     infiniopHandle_t handle,