Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
jerrrrry
infinicore
Commits
b4eadd81
Commit
b4eadd81
authored
Feb 24, 2025
by
PanZezhong
Browse files
issue/69 修改event query接口,删除多余编译判断
parent
a9db6af1
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
6 additions
and
13 deletions
+6
-13
include/infinicore.h
include/infinicore.h
+0
-12
include/infinirt.h
include/infinirt.h
+6
-1
No files found.
include/infinicore.h
View file @
b4eadd81
...
...
@@ -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__
include/infinirt.h
View file @
b4eadd81
...
...
@@ -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
);
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment