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
9d611676
Commit
9d611676
authored
Mar 05, 2025
by
PanZezhong
Browse files
issue/85 重命名头文件
parent
3cbe5e02
Changes
43
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
61 additions
and
68 deletions
+61
-68
include/infinicore.h
include/infinicore.h
+3
-4
include/infiniop.h
include/infiniop.h
+3
-3
include/infiniop/handle.h
include/infiniop/handle.h
+3
-6
include/infiniop/operator_descriptor.h
include/infiniop/operator_descriptor.h
+4
-7
include/infiniop/ops/add.h
include/infiniop/ops/add.h
+3
-3
include/infiniop/ops/attention.h
include/infiniop/ops/attention.h
+3
-3
include/infiniop/ops/avg_pool.h
include/infiniop/ops/avg_pool.h
+3
-3
include/infiniop/ops/causal_softmax.h
include/infiniop/ops/causal_softmax.h
+3
-3
include/infiniop/ops/conv.h
include/infiniop/ops/conv.h
+3
-3
include/infiniop/ops/expand.h
include/infiniop/ops/expand.h
+3
-3
include/infiniop/ops/gemm.h
include/infiniop/ops/gemm.h
+3
-3
include/infiniop/ops/global_avg_pool.h
include/infiniop/ops/global_avg_pool.h
+3
-3
include/infiniop/ops/matmul.h
include/infiniop/ops/matmul.h
+3
-3
include/infiniop/ops/max_pool.h
include/infiniop/ops/max_pool.h
+3
-3
include/infiniop/ops/mlp.h
include/infiniop/ops/mlp.h
+3
-3
include/infiniop/ops/random_sample.h
include/infiniop/ops/random_sample.h
+3
-3
include/infiniop/ops/rearrange.h
include/infiniop/ops/rearrange.h
+3
-3
include/infiniop/ops/relu.h
include/infiniop/ops/relu.h
+3
-3
include/infiniop/ops/rms_norm.h
include/infiniop/ops/rms_norm.h
+3
-3
include/infiniop/ops/rotary_embedding.h
include/infiniop/ops/rotary_embedding.h
+3
-3
No files found.
include/infinicore.h
View file @
9d611676
#ifndef __INFINICORE_H__
#define __INFINICORE_H__
#ifndef __INFINICORE_
API_
H__
#define __INFINICORE_
API_
H__
#if defined(_WIN32)
#define __export __declspec(dllexport)
...
...
@@ -71,5 +71,4 @@ typedef enum {
INFINI_DTYPE_BF16
=
19
,
}
infiniDtype_t
;
#endif // __INFINICORE_H__
#endif // __INFINICORE_API_H__
include/infiniop.h
View file @
9d611676
#ifndef __INFINIOP_H__
#define __INFINIOP_H__
#ifndef __INFINIOP_
API_
H__
#define __INFINIOP_
API_
H__
#include "infiniop/tensor_descriptor.h"
#include "infiniop/handle.h"
#include "infiniop/ops/add.h"
...
...
@@ -20,4 +20,4 @@
#include "infiniop/ops/rotary_embedding.h"
#include "infiniop/ops/swiglu.h"
#endif // __INFINIOP_H__
#endif // __INFINIOP_
API_
H__
include/infiniop/handle.h
View file @
9d611676
#ifndef __INFINIOP_HANDLE__
#define __INFINIOP_HANDLE__
#ifndef __INFINIOP_HANDLE_
API_H_
_
#define __INFINIOP_HANDLE_
API_H_
_
#include "../infinicore.h"
typedef
struct
InfiniopHandle
{
infiniDevice_t
device
;
int
device_id
;
}
InfiniopHandle
;
struct
InfiniopHandle
;
typedef
InfiniopHandle
*
infiniopHandle_t
;
...
...
include/infiniop/operator.h
→
include/infiniop/operator
_descriptor
.h
View file @
9d611676
#ifndef __INFINIOP_OPERATOR___
#define __INFINIOP_OPERATOR___
#ifndef __INFINIOP_OPERATOR_
DESCRIPTOR_API_H
__
#define __INFINIOP_OPERATOR_
DESCRIPTOR_API_H
__
#include "handle.h"
#include "tensor_descriptor.h"
// Base descriptor for all operators
typedef
struct
InfiniopDescriptor
{
infiniDevice_t
device_type
;
int
device_id
;
}
InfiniopDescriptor
;
struct
InfiniopDescriptor
;
__C
__export
infiniStatus_t
infiniopGetDescriptorDeviceType
(
const
InfiniopDescriptor
*
desc_ptr
,
infiniDevice_t
*
device_type
);
__C
__export
infiniStatus_t
infiniopGetDescriptorDeviceId
(
const
InfiniopDescriptor
*
desc_ptr
,
int
*
device_id
);
#endif //__INFINIOP_OPERATOR___
#endif //__INFINIOP_OPERATOR_
DESCRIPTOR_API_H
__
include/infiniop/ops/add.h
View file @
9d611676
#ifndef __INFINIOP_ADD_H__
#define __INFINIOP_ADD_H__
#ifndef __INFINIOP_ADD_
API_
H__
#define __INFINIOP_ADD_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopAddDescriptor_t
;
...
...
include/infiniop/ops/attention.h
View file @
9d611676
#ifndef __INFINIOP_ATTENTION_H__
#define __INFINIOP_ATTENTION_H__
#ifndef __INFINIOP_ATTENTION_
API_
H__
#define __INFINIOP_ATTENTION_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
#include "matmul.h"
#include "swiglu.h"
...
...
include/infiniop/ops/avg_pool.h
View file @
9d611676
#ifndef __INFINIOP_AVG_POOL_H__
#define __INFINIOP_AVG_POOL_H__
#ifndef __INFINIOP_AVG_POOL_
API_
H__
#define __INFINIOP_AVG_POOL_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopAvgPoolDescriptor_t
;
...
...
include/infiniop/ops/causal_softmax.h
View file @
9d611676
#ifndef __INFINIOP_CAUSAL_SOFTMAX_H__
#define __INFINIOP_CAUSAL_SOFTMAX_H__
#ifndef __INFINIOP_CAUSAL_SOFTMAX_
API_
H__
#define __INFINIOP_CAUSAL_SOFTMAX_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopCausalSoftmaxDescriptor_t
;
...
...
include/infiniop/ops/conv.h
View file @
9d611676
#ifndef __INFINIOP_CONV_H__
#define __INFINIOP_CONV_H__
#ifndef __INFINIOP_CONV_
API_
H__
#define __INFINIOP_CONV_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopConvDescriptor_t
;
...
...
include/infiniop/ops/expand.h
View file @
9d611676
#ifndef __INFINIOP_EXPAND_H__
#define __INFINIOP_EXPAND_H__
#ifndef __INFINIOP_EXPAND_
API_
H__
#define __INFINIOP_EXPAND_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopExpandDescriptor_t
;
...
...
include/infiniop/ops/gemm.h
View file @
9d611676
#ifndef __INFINIOP_GEMM_H__
#define __INFINIOP_GEMM_H__
#ifndef __INFINIOP_GEMM_
API_
H__
#define __INFINIOP_GEMM_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopGEMMDescriptor_t
;
...
...
include/infiniop/ops/global_avg_pool.h
View file @
9d611676
#ifndef __INFINIOP_GLOBAL_AVG_POOL_H__
#define __INFINIOP_GLOBAL_AVG_POOL_H__
#ifndef __INFINIOP_GLOBAL_AVG_POOL_
API_
H__
#define __INFINIOP_GLOBAL_AVG_POOL_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopGlobalAvgPoolDescriptor_t
;
...
...
include/infiniop/ops/matmul.h
View file @
9d611676
#ifndef __INFINIOP_MATMUL_H__
#define __INFINIOP_MATMUL_H__
#ifndef __INFINIOP_MATMUL_
API_
H__
#define __INFINIOP_MATMUL_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopMatmulDescriptor_t
;
...
...
include/infiniop/ops/max_pool.h
View file @
9d611676
#ifndef __INFINIOP_MAX_POOL_H__
#define __INFINIOP_MAX_POOL_H__
#ifndef __INFINIOP_MAX_POOL_
API_
H__
#define __INFINIOP_MAX_POOL_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopMaxPoolDescriptor_t
;
...
...
include/infiniop/ops/mlp.h
View file @
9d611676
#ifndef __INFINIOP_MLP_H__
#define __INFINIOP_MLP_H__
#ifndef __INFINIOP_MLP_
API_
H__
#define __INFINIOP_MLP_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
#include "matmul.h"
#include "swiglu.h"
...
...
include/infiniop/ops/random_sample.h
View file @
9d611676
#ifndef __INFINIOP_RANDOM_SAMPLE_H__
#define __INFINIOP_RANDOM_SAMPLE_H__
#ifndef __INFINIOP_RANDOM_SAMPLE_
API_
H__
#define __INFINIOP_RANDOM_SAMPLE_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopRandomSampleDescriptor_t
;
...
...
include/infiniop/ops/rearrange.h
View file @
9d611676
#ifndef __INFINIOP_REARRANGE_H__
#define __INFINIOP_REARRANGE_H__
#ifndef __INFINIOP_REARRANGE_
API_
H__
#define __INFINIOP_REARRANGE_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopRearrangeDescriptor_t
;
...
...
include/infiniop/ops/relu.h
View file @
9d611676
#ifndef __INFINIOP_RELU_H__
#define __INFINIOP_RELU_H__
#ifndef __INFINIOP_RELU_
API_
H__
#define __INFINIOP_RELU_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopReluDescriptor_t
;
...
...
include/infiniop/ops/rms_norm.h
View file @
9d611676
#ifndef __INFINIOP_RMS_NORM_H__
#define __INFINIOP_RMS_NORM_H__
#ifndef __INFINIOP_RMS_NORM_
API_
H__
#define __INFINIOP_RMS_NORM_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopRMSNormDescriptor_t
;
...
...
include/infiniop/ops/rotary_embedding.h
View file @
9d611676
#ifndef __INFINIOP_ROTARY_EMBEDDING_H__
#define __INFINIOP_ROTARY_EMBEDDING_H__
#ifndef __INFINIOP_ROTARY_EMBEDDING_
API_
H__
#define __INFINIOP_ROTARY_EMBEDDING_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopRoPEDescriptor_t
;
...
...
Prev
1
2
3
Next
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