Commit b4eadd81 authored by PanZezhong's avatar PanZezhong
Browse files

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

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