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
52 additions
and
16 deletions
+52
-16
include/infiniop/ops/swiglu.h
include/infiniop/ops/swiglu.h
+3
-3
include/infiniop/tensor_descriptor.h
include/infiniop/tensor_descriptor.h
+2
-2
include/infinirt.h
include/infinirt.h
+3
-3
src/infiniop/devices/ascend/ascend_handle.h
src/infiniop/devices/ascend/ascend_handle.h
+1
-1
src/infiniop/devices/bang/bang_handle.h
src/infiniop/devices/bang/bang_handle.h
+1
-1
src/infiniop/devices/cpu/cpu_handle.h
src/infiniop/devices/cpu/cpu_handle.h
+1
-1
src/infiniop/devices/cuda/cuda_handle.h
src/infiniop/devices/cuda/cuda_handle.h
+1
-1
src/infiniop/devices/kunlun/kunlun_handle.h
src/infiniop/devices/kunlun/kunlun_handle.h
+1
-1
src/infiniop/handle.h
src/infiniop/handle.h
+11
-0
src/infiniop/operator.h
src/infiniop/operator.h
+11
-0
src/infiniop/operator_descriptor.cc
src/infiniop/operator_descriptor.cc
+1
-1
src/infiniop/ops/causal_softmax/operator.cc
src/infiniop/ops/causal_softmax/operator.cc
+2
-0
src/infiniop/ops/matmul/blas.h
src/infiniop/ops/matmul/blas.h
+1
-1
src/infiniop/ops/matmul/matmul.h
src/infiniop/ops/matmul/matmul.h
+1
-1
src/infiniop/ops/matmul/operator.cc
src/infiniop/ops/matmul/operator.cc
+2
-0
src/infiniop/ops/random_sample/operator.cc
src/infiniop/ops/random_sample/operator.cc
+2
-0
src/infiniop/ops/rearrange/operator.cc
src/infiniop/ops/rearrange/operator.cc
+2
-0
src/infiniop/ops/rms_norm/operator.cc
src/infiniop/ops/rms_norm/operator.cc
+2
-0
src/infiniop/ops/rotary_embedding/operator.cc
src/infiniop/ops/rotary_embedding/operator.cc
+2
-0
src/infiniop/ops/swiglu/operator.cc
src/infiniop/ops/swiglu/operator.cc
+2
-0
No files found.
include/infiniop/ops/swiglu.h
View file @
9d611676
#ifndef __INFINIOP_SWIGLU_H__
#define __INFINIOP_SWIGLU_H__
#ifndef __INFINIOP_SWIGLU_
API_
H__
#define __INFINIOP_SWIGLU_
API_
H__
#include "../operator.h"
#include "../operator
_descriptor
.h"
typedef
InfiniopDescriptor
*
infiniopSwiGLUDescriptor_t
;
...
...
include/infiniop/tensor_descriptor.h
View file @
9d611676
#ifndef __INFINIOP_TENSOR_DESCRIPTOR__
#define __INFINIOP_TENSOR_DESCRIPTOR__
#ifndef __INFINIOP_TENSOR_DESCRIPTOR_
API_H_
_
#define __INFINIOP_TENSOR_DESCRIPTOR_
API_H_
_
#include "../infinicore.h"
...
...
include/infinirt.h
View file @
9d611676
#ifndef __INFINIRT_H__
#define __INFINIRT_H__
#ifndef __INFINIRT_
API_
H__
#define __INFINIRT_
API_
H__
#include "infinicore.h"
...
...
@@ -53,4 +53,4 @@ __C __export infiniStatus_t infinirtMemcpyAsync(void *dst, const void *src, size
__C
__export
infiniStatus_t
infinirtMallocAsync
(
void
**
p_ptr
,
size_t
size
,
infinirtStream_t
stream
);
__C
__export
infiniStatus_t
infinirtFreeAsync
(
void
*
ptr
,
infinirtStream_t
stream
);
#endif // __INFINIRT_H__
#endif // __INFINIRT_
API_
H__
src/infiniop/devices/ascend/ascend_handle.h
View file @
9d611676
#ifndef __INFINIOP_ASCEND_HANDLE_H__
#define __INFINIOP_ASCEND_HANDLE_H__
#include "../../handle.h"
#include "infinicore.h"
#include "infiniop/handle.h"
struct
InfiniopAscendHandle
;
typedef
struct
InfiniopAscendHandle
*
infiniopAscendHandle_t
;
...
...
src/infiniop/devices/bang/bang_handle.h
View file @
9d611676
#ifndef BANG_HANDLE_H
#define BANG_HANDLE_H
#include "
infiniop
/handle.h"
#include "
../..
/handle.h"
struct
InfiniopBangHandle
;
typedef
struct
InfiniopBangHandle
*
infiniopBangHandle_t
;
...
...
src/infiniop/devices/cpu/cpu_handle.h
View file @
9d611676
#ifndef __INFINIOP_CPU_HANDLE_H__
#define __INFINIOP_CPU_HANDLE_H__
#include "
infiniop
/handle.h"
#include "
../..
/handle.h"
typedef
infiniopHandle_t
infiniopCpuHandle_t
;
...
...
src/infiniop/devices/cuda/cuda_handle.h
View file @
9d611676
#ifndef __INFINIOP_CUDA_HANDLE_H__
#define __INFINIOP_CUDA_HANDLE_H__
#include "
infiniop
/handle.h"
#include "
../..
/handle.h"
struct
InfiniopCudaHandle
;
typedef
struct
InfiniopCudaHandle
*
infiniopCudaHandle_t
;
...
...
src/infiniop/devices/kunlun/kunlun_handle.h
View file @
9d611676
#ifndef __INFINIOP_KUNLUN_HANDLE_H__
#define __INFINIOP_KUNLUN_HANDLE_H__
#include "
infiniop
/handle.h"
#include "
../..
/handle.h"
struct
InfiniopKunlunHandle
;
...
...
src/infiniop/handle.h
0 → 100644
View file @
9d611676
#ifndef __INFINIOP_HANDLE_H__
#define __INFINIOP_HANDLE_H__
#include "infiniop/handle.h"
struct
InfiniopHandle
{
infiniDevice_t
device
;
int
device_id
;
};
#endif //__INFINIOP_HANDLE_H__
src/infiniop/operator.h
0 → 100644
View file @
9d611676
#ifndef __INFINIOP_OPERATOR_H__
#define __INFINIOP_OPERATOR_H__
#include "infiniop/operator_descriptor.h"
struct
InfiniopDescriptor
{
infiniDevice_t
device_type
;
int
device_id
;
};
#endif
src/infiniop/operator.cc
→
src/infiniop/operator
_descriptor
.cc
View file @
9d611676
#include "
infiniop/
operator.h"
#include "operator.h"
infiniStatus_t
infiniopGetDescriptorDeviceType
(
const
InfiniopDescriptor
*
desc_ptr
,
...
...
src/infiniop/ops/causal_softmax/operator.cc
View file @
9d611676
#include "../../operator.h"
#include "../../handle.h"
#include "infiniop/ops/causal_softmax.h"
__C
infiniStatus_t
infiniopCreateCausalSoftmaxDescriptor
(
...
...
src/infiniop/ops/matmul/blas.h
View file @
9d611676
#ifndef __BLAS_H__
#define __BLAS_H__
#include "../../operator.h"
#include "../../tensor.h"
#include "infiniop/operator.h"
#include <algorithm>
namespace
matmul
{
...
...
src/infiniop/ops/matmul/matmul.h
View file @
9d611676
#ifndef __MATMUL_H__
#define __MATMUL_H__
#include "../../operator.h"
#include "blas.h"
#include "infiniop/operator.h"
/**
* # 关于 `DESCRIPTOR(NAMESPACE)` 和 `struct Opaque;` 的说明
...
...
src/infiniop/ops/matmul/operator.cc
View file @
9d611676
#include "../../operator.h"
#include "../../handle.h"
#include "infiniop/ops/matmul.h"
#ifdef ENABLE_CPU_API
...
...
src/infiniop/ops/random_sample/operator.cc
View file @
9d611676
#include "../../operator.h"
#include "../../handle.h"
#include "infiniop/ops/random_sample.h"
__C
infiniStatus_t
infiniopCreateRandomSampleDescriptor
(
infiniopHandle_t
handle
,
infiniopRandomSampleDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
result
,
infiniopTensorDescriptor_t
probs
)
{
...
...
src/infiniop/ops/rearrange/operator.cc
View file @
9d611676
#include "../../operator.h"
#include "../../handle.h"
#include "infiniop/ops/rearrange.h"
__C
infiniStatus_t
infiniopCreateRearrangeDescriptor
(
...
...
src/infiniop/ops/rms_norm/operator.cc
View file @
9d611676
#include "../../operator.h"
#include "../../handle.h"
#include "infiniop/ops/rms_norm.h"
__C
infiniStatus_t
infiniopCreateRMSNormDescriptor
(
...
...
src/infiniop/ops/rotary_embedding/operator.cc
View file @
9d611676
#include "../../operator.h"
#include "../../handle.h"
#include "infiniop/ops/rotary_embedding.h"
__C
infiniStatus_t
infiniopCreateRoPEDescriptor
(
...
...
src/infiniop/ops/swiglu/operator.cc
View file @
9d611676
#include "../../operator.h"
#include "../../handle.h"
#include "infiniop/ops/swiglu.h"
__C
infiniStatus_t
infiniopCreateSwiGLUDescriptor
(
...
...
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