Merge pull request #88 from PanZezhong1725/issue/87/cpu

issue/87/cpu 重构cpu handle

src/infiniop/devices/cpu/cpu_handle.cc

 #include "cpu_handle.h"
 
-infiniStatus_t createCpuHandle(infiniopCpuHandle_t *handle_ptr) {
-    *handle_ptr = new InfiniopHandle{INFINI_DEVICE_CPU, 0};
-    return INFINI_STATUS_SUCCESS;
-}
+namespace infiniop::cpu {
+Handle::Handle() : InfiniopHandle{INFINI_DEVICE_CPU, 0} {}
 
-infiniStatus_t destroyCpuHandle(infiniopCpuHandle_t handle) {
-    delete handle;
+infiniStatus_t Handle::create(InfiniopHandle **handle_ptr) {
+    *handle_ptr = new Handle{};
     return INFINI_STATUS_SUCCESS;
 }
+
+} // namespace infiniop::cpu

src/infiniop/devices/cpu/cpu_handle.h

@@ -3,10 +3,13 @@
 
 #include "../../handle.h"
 
-typedef infiniopHandle_t infiniopCpuHandle_t;
-
-infiniStatus_t createCpuHandle(infiniopCpuHandle_t *handle_ptr);
-
-infiniStatus_t destroyCpuHandle(infiniopCpuHandle_t handle);
+namespace infiniop::cpu {
+class Handle : public InfiniopHandle {
+    Handle();
+
+public:
+    static infiniStatus_t create(InfiniopHandle **handle_ptr);
+};
+} // namespace infiniop::cpu
 
 #endif

src/infiniop/devices/handle.cc

@@ -24,7 +24,7 @@ __C infiniStatus_t infiniopCreateHandle(infiniopHandle_t *handle_ptr,
     switch (device) {
 #ifdef ENABLE_CPU_API
     case INFINI_DEVICE_CPU:
-        return createCpuHandle((infiniopCpuHandle_t *)handle_ptr);
+        return infiniop::cpu::Handle::create(handle_ptr);
 #endif
 #ifdef ENABLE_CUDA_API
     case INFINI_DEVICE_NVIDIA: {
@@ -51,10 +51,15 @@ __C infiniStatus_t infiniopCreateHandle(infiniopHandle_t *handle_ptr,
 }
 
 __C infiniStatus_t infiniopDestroyHandle(infiniopHandle_t handle) {
+
+#define DELETE(CASE, NAMESPACE)                                         \
+    case CASE:                                                          \
+        delete reinterpret_cast<infiniop::NAMESPACE::Handle *>(handle); \
+        return INFINI_STATUS_SUCCESS;
+
     switch (handle->device) {
 #ifdef ENABLE_CPU_API
-    case INFINI_DEVICE_CPU:
-        return destroyCpuHandle((infiniopCpuHandle_t)handle);
+        DELETE(INFINI_DEVICE_CPU, cpu)
 #endif
 #ifdef ENABLE_CUDA_API
     case INFINI_DEVICE_NVIDIA: {
@@ -76,6 +81,8 @@ __C infiniStatus_t infiniopDestroyHandle(infiniopHandle_t handle) {
         return destroyKunlunHandle((infiniopKunlunHandle_t)handle);
     }
 #endif
+    default:
+        return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
     }
-    return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
+#undef DELETE
 }

src/infiniop/ops/matmul/cpu/matmul_cpu.cc

@@ -12,7 +12,7 @@ infiniStatus_t Descriptor::create(
     infiniopTensorDescriptor_t c_desc,
     infiniopTensorDescriptor_t a_desc,
     infiniopTensorDescriptor_t b_desc) {
-    auto handle = reinterpret_cast<infiniopCpuHandle_t>(handle_);
+    auto handle = reinterpret_cast<infiniop::cpu::Handle *>(handle_);
     auto dtype = c_desc->dtype();
 
     if (dtype != INFINI_DTYPE_F16 && dtype != INFINI_DTYPE_F32) {