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

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

parent b4eadd81
#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 {
......
......@@ -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);
......
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