Commit b4eadd81 authored by PanZezhong's avatar PanZezhong
Browse files

issue/69 修改event query接口,删除多余编译判断

parent a9db6af1
......@@ -2,8 +2,6 @@
#define __INFINICORE_H__
#include <stdint.h>
#ifndef __INFINI_EXPORT_C__
#define __INFINI_EXPORT_C__
#if defined(_WIN32)
#define __export __declspec(dllexport)
#elif defined(__GNUC__) && ((__GNUC__ >= 4) || (__GNUC__ == 3 && __GNUC_MINOR__ >= 3))
......@@ -19,10 +17,7 @@
#define __C
#include <stddef>
#endif
#endif // __INFINI_EXPORT_C__
#ifndef __INFINI_STATUS__
#define __INFINI_STATUS__
typedef enum {
// Success
INFINI_STATUS_SUCCESS = 0,
......@@ -39,10 +34,7 @@ typedef enum {
INFINI_STATUS_BAD_TENSOR_STRIDES = 12,
INFINI_STATUS_INSUFFICIENT_WORKSPACE = 13,
} infiniStatus_t;
#endif // __INFINI_STATUS__
#ifndef __INFINI_DEVICE__
#define __INFINI_DEVICE__
typedef enum {
INFINI_DEVICE_CPU = 0,
INFINI_DEVICE_NVIDIA = 1,
......@@ -54,10 +46,7 @@ typedef enum {
INFINI_DEVICE_KUNLUN = 7,
INFINI_DEVICE_SUGON = 8,
} infiniDevice_t;
#endif // __INFINI_DEVICE__
#ifndef __INFINI_DTYPE__
#define __INFINI_DTYPE__
typedef enum {
INFINI_DTYPE_INVALID = 0,
INFINI_DTYPE_BYTE = 1,
......@@ -127,6 +116,5 @@ inline size_t infiniSizeof(infiniDtype_t dtype) {
return 0;
}
}
#endif // __INFINI_DTYPE__
#endif // __INFINICORE_H__
......@@ -21,9 +21,14 @@ __C __export infiniStatus_t infinirtStreamSynchronize(infinirtStream_t stream);
__C __export infiniStatus_t infinirtStreamWaitEvent(infinirtStream_t stream, infinirtEvent_t event);
// Event
typedef enum {
INFINIRT_EVENT_COMPLETE = 0,
INFINIRT_EVENT_NOT_READY = 1,
} infinirtEventStatus;
__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);
__C __export infiniStatus_t infinirtEventQuery(infinirtEvent_t event, infinirtEventStatus *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