issue/69 增加获得全设备数量接口

Author: PanZezhong1725

include/infinicore.h

@@ -1,6 +1,5 @@
 #ifndef __INFINICORE_H__
 #define __INFINICORE_H__
-#include <stdint.h>
 
 #if defined(_WIN32)
 #define __export __declspec(dllexport)
@@ -15,7 +14,7 @@
 #include <cstddef>
 #else
 #define __C
-#include <stddef>
+#include <stddef.h>
 #endif
 
 typedef enum {
@@ -45,6 +44,7 @@ typedef enum {
     INFINI_DEVICE_ILUVATAR = 6,
     INFINI_DEVICE_KUNLUN = 7,
     INFINI_DEVICE_SUGON = 8,
+    INFINI_DEVICE_TYPE_COUNT
 } infiniDevice_t;
 
 typedef enum {

include/infinirt.h

@@ -9,9 +9,10 @@ typedef void *infinirtEvent_t;
 __C __export infiniStatus_t infinirtInit(infiniDevice_t device);
 
 // Device
+__C __export infiniStatus_t infinirtGetAllDeviceCount(int *count_array);
+__C __export infiniStatus_t infinirtGetDeviceCount(infiniDevice_t device, int *count);
 __C __export infiniStatus_t infinirtSetDevice(infiniDevice_t device, int device_id);
 __C __export infiniStatus_t infinirtGetDevice(infiniDevice_t *device_ptr, int *device_id_ptr);
-__C __export infiniStatus_t infinirtGetDeviceCount(infiniDevice_t device, int *count);
 __C __export infiniStatus_t infinirtDeviceSynchronize();
 
 // Stream
@@ -24,11 +25,11 @@ __C __export infiniStatus_t infinirtStreamWaitEvent(infinirtStream_t stream, inf
 typedef enum {
     INFINIRT_EVENT_COMPLETE = 0,
     INFINIRT_EVENT_NOT_READY = 1,
-} infinirtEventStatus;
+} infinirtEventStatus_t;
 
 __C __export infiniStatus_t infinirtEventCreate(infinirtEvent_t *event_ptr);
 __C __export infiniStatus_t infinirtEventRecord(infinirtEvent_t event, infinirtStream_t stream);
-__C __export infiniStatus_t infinirtEventQuery(infinirtEvent_t event, infinirtEventStatus *status_ptr);
+__C __export infiniStatus_t infinirtEventQuery(infinirtEvent_t event, infinirtEventStatus_t *status_ptr);
 __C __export infiniStatus_t infinirtEventSynchronize(infinirtEvent_t event);
 __C __export infiniStatus_t infinirtEventDestroy(infinirtEvent_t event);