Commit 4ee4c4f4 authored by PanZezhong's avatar PanZezhong
Browse files

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

parent b4eadd81
#ifndef __INFINICORE_H__ #ifndef __INFINICORE_H__
#define __INFINICORE_H__ #define __INFINICORE_H__
#include <stdint.h>
#if defined(_WIN32) #if defined(_WIN32)
#define __export __declspec(dllexport) #define __export __declspec(dllexport)
...@@ -15,7 +14,7 @@ ...@@ -15,7 +14,7 @@
#include <cstddef> #include <cstddef>
#else #else
#define __C #define __C
#include <stddef> #include <stddef.h>
#endif #endif
typedef enum { typedef enum {
...@@ -45,6 +44,7 @@ typedef enum { ...@@ -45,6 +44,7 @@ typedef enum {
INFINI_DEVICE_ILUVATAR = 6, INFINI_DEVICE_ILUVATAR = 6,
INFINI_DEVICE_KUNLUN = 7, INFINI_DEVICE_KUNLUN = 7,
INFINI_DEVICE_SUGON = 8, INFINI_DEVICE_SUGON = 8,
INFINI_DEVICE_TYPE_COUNT
} infiniDevice_t; } infiniDevice_t;
typedef enum { typedef enum {
......
...@@ -9,9 +9,10 @@ typedef void *infinirtEvent_t; ...@@ -9,9 +9,10 @@ typedef void *infinirtEvent_t;
__C __export infiniStatus_t infinirtInit(infiniDevice_t device); __C __export infiniStatus_t infinirtInit(infiniDevice_t device);
// 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 infinirtSetDevice(infiniDevice_t device, int device_id);
__C __export infiniStatus_t infinirtGetDevice(infiniDevice_t *device_ptr, int *device_id_ptr); __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(); __C __export infiniStatus_t infinirtDeviceSynchronize();
// Stream // Stream
...@@ -24,11 +25,11 @@ __C __export infiniStatus_t infinirtStreamWaitEvent(infinirtStream_t stream, inf ...@@ -24,11 +25,11 @@ __C __export infiniStatus_t infinirtStreamWaitEvent(infinirtStream_t stream, inf
typedef enum { typedef enum {
INFINIRT_EVENT_COMPLETE = 0, INFINIRT_EVENT_COMPLETE = 0,
INFINIRT_EVENT_NOT_READY = 1, INFINIRT_EVENT_NOT_READY = 1,
} infinirtEventStatus; } infinirtEventStatus_t;
__C __export infiniStatus_t infinirtEventCreate(infinirtEvent_t *event_ptr); __C __export infiniStatus_t infinirtEventCreate(infinirtEvent_t *event_ptr);
__C __export infiniStatus_t infinirtEventRecord(infinirtEvent_t event, infinirtStream_t stream); __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 infinirtEventSynchronize(infinirtEvent_t event);
__C __export infiniStatus_t infinirtEventDestroy(infinirtEvent_t event); __C __export infiniStatus_t infinirtEventDestroy(infinirtEvent_t event);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment