Skip to content

Commit 2d7713b

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-8
1 parent 310fcdb commit 2d7713b

71 files changed

Lines changed: 501 additions & 622 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎include/infinicore/ops.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@
22

33
#include "ops/adaptive_max_pool1d.hpp"
44
#include "ops/add.hpp"
5-
#include "ops/asin.hpp"
65
#include "ops/add_rms_norm.hpp"
76
#include "ops/addcmul.hpp"
7+
#include "ops/asin.hpp"
88
#include "ops/asinh.hpp"
99
#include "ops/atanh.hpp"
1010
#include "ops/attention.hpp"
@@ -17,11 +17,11 @@
1717
#include "ops/cross_entropy.hpp"
1818
#include "ops/embedding.hpp"
1919
#include "ops/flash_attention.hpp"
20+
#include "ops/fmin.hpp"
2021
#include "ops/fmod.hpp"
2122
#include "ops/hardswish.hpp"
2223
#include "ops/hardtanh.hpp"
2324
#include "ops/kv_caching.hpp"
24-
#include "ops/fmin.hpp"
2525
#include "ops/matmul.hpp"
2626
#include "ops/ones.hpp"
2727
#include "ops/paged_attention.hpp"

‎include/infinicore/ops/adaptive_avg_pool3d.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,4 @@ class AdaptiveAvgPool3D {
1313

1414
Tensor adaptive_avg_pool3d(Tensor x, std::vector<size_t> output_size);
1515
void adaptive_avg_pool3d_(Tensor y, Tensor x);
16-
} // namespace infinicore::op
16+
} // namespace infinicore::op

‎include/infinicore/ops/addr.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,4 @@ class Addr {
1313

1414
Tensor addr(Tensor input, Tensor vec1, Tensor vec2, float beta = 1.0f, float alpha = 1.0f);
1515
void addr_(Tensor out, Tensor input, Tensor vec1, Tensor vec2, float beta = 1.0f, float alpha = 1.0f);
16-
} // namespace infinicore::op
16+
} // namespace infinicore::op

‎include/infinicore/ops/argwhere.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,4 +11,4 @@ class Argwhere {
1111
static common::OpDispatcher<schema> &dispatcher();
1212
};
1313
Tensor argwhere(Tensor x);
14-
} // namespace infinicore::op
14+
} // namespace infinicore::op

‎include/infinicore/ops/asin.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,4 @@ class Asin {
1515
Tensor asin(Tensor input);
1616
void asin_(Tensor output, Tensor input);
1717

18-
} // namespace infinicore::op
18+
} // namespace infinicore::op

‎include/infiniop.h‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,16 +2,16 @@
22
#define __INFINIOP_API_H__
33

44
#include "infiniop/handle.h"
5-
#include "infiniop/ops/adaptive_max_pool1d.h"
65
#include "infiniop/ops/adaptive_avg_pool3d.h"
6+
#include "infiniop/ops/adaptive_max_pool1d.h"
77
#include "infiniop/ops/add.h"
8-
#include "infiniop/ops/addr.h"
98
#include "infiniop/ops/add_rms_norm.h"
109
#include "infiniop/ops/addcmul.h"
10+
#include "infiniop/ops/addr.h"
1111
#include "infiniop/ops/all.h"
12+
#include "infiniop/ops/asin.h"
1213
#include "infiniop/ops/asinh.h"
1314
#include "infiniop/ops/atanh.h"
14-
#include "infiniop/ops/asin.h"
1515
#include "infiniop/ops/attention.h"
1616
#include "infiniop/ops/avg_pool1d.h"
1717
#include "infiniop/ops/binary_cross_entropy_with_logits.h"
@@ -24,8 +24,8 @@
2424
#include "infiniop/ops/embedding.h"
2525
#include "infiniop/ops/equal.h"
2626
#include "infiniop/ops/flash_attention.h"
27-
#include "infiniop/ops/fmod.h"
2827
#include "infiniop/ops/fmin.h"
28+
#include "infiniop/ops/fmod.h"
2929
#include "infiniop/ops/gelu.h"
3030
#include "infiniop/ops/gemm.h"
3131
#include "infiniop/ops/hardswish.h"

‎include/infiniop/ops/adaptive_avg_pool3d.h‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,25 +6,25 @@
66

77
typedef struct InfiniopDescriptor *infiniopAdaptiveAvgPool3DDescriptor_t;
88

9-
__C __export infiniStatus_t infiniopCreateAdaptiveAvgPool3DDescriptor(
9+
__INFINI_C __export infiniStatus_t infiniopCreateAdaptiveAvgPool3DDescriptor(
1010
infiniopHandle_t handle,
1111
infiniopAdaptiveAvgPool3DDescriptor_t *desc_ptr,
1212
infiniopTensorDescriptor_t y,
1313
infiniopTensorDescriptor_t x,
1414
size_t *output_size);
1515

16-
__C __export infiniStatus_t infiniopGetAdaptiveAvgPool3DWorkspaceSize(
16+
__INFINI_C __export infiniStatus_t infiniopGetAdaptiveAvgPool3DWorkspaceSize(
1717
infiniopAdaptiveAvgPool3DDescriptor_t desc,
1818
size_t *size);
1919

20-
__C __export infiniStatus_t infiniopAdaptiveAvgPool3D(
20+
__INFINI_C __export infiniStatus_t infiniopAdaptiveAvgPool3D(
2121
infiniopAdaptiveAvgPool3DDescriptor_t desc,
2222
void *workspace,
2323
size_t workspace_size,
2424
void *y,
2525
const void *x,
2626
void *stream);
2727

28-
__C __export infiniStatus_t infiniopDestroyAdaptiveAvgPool3DDescriptor(infiniopAdaptiveAvgPool3DDescriptor_t desc);
28+
__INFINI_C __export infiniStatus_t infiniopDestroyAdaptiveAvgPool3DDescriptor(infiniopAdaptiveAvgPool3DDescriptor_t desc);
2929

30-
#endif
30+
#endif

‎include/infiniop/ops/addr.h‎

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -5,26 +5,26 @@
55

66
typedef struct InfiniopDescriptor *infiniopAddrDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateAddrDescriptor(infiniopHandle_t handle,
9-
infiniopAddrDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t out,
11-
infiniopTensorDescriptor_t input,
12-
infiniopTensorDescriptor_t vec1,
13-
infiniopTensorDescriptor_t vec2,
14-
float beta,
15-
float alpha);
8+
__INFINI_C __export infiniStatus_t infiniopCreateAddrDescriptor(infiniopHandle_t handle,
9+
infiniopAddrDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t out,
11+
infiniopTensorDescriptor_t input,
12+
infiniopTensorDescriptor_t vec1,
13+
infiniopTensorDescriptor_t vec2,
14+
float beta,
15+
float alpha);
1616

17-
__C __export infiniStatus_t infiniopGetAddrWorkspaceSize(infiniopAddrDescriptor_t desc, size_t *size);
17+
__INFINI_C __export infiniStatus_t infiniopGetAddrWorkspaceSize(infiniopAddrDescriptor_t desc, size_t *size);
1818

19-
__C __export infiniStatus_t infiniopAddr(infiniopAddrDescriptor_t desc,
20-
void *workspace,
21-
size_t workspace_size,
22-
void *out,
23-
const void *input,
24-
const void *vec1,
25-
const void *vec2,
26-
void *stream);
19+
__INFINI_C __export infiniStatus_t infiniopAddr(infiniopAddrDescriptor_t desc,
20+
void *workspace,
21+
size_t workspace_size,
22+
void *out,
23+
const void *input,
24+
const void *vec1,
25+
const void *vec2,
26+
void *stream);
2727

28-
__C __export infiniStatus_t infiniopDestroyAddrDescriptor(infiniopAddrDescriptor_t desc);
28+
__INFINI_C __export infiniStatus_t infiniopDestroyAddrDescriptor(infiniopAddrDescriptor_t desc);
2929

30-
#endif
30+
#endif

‎include/infiniop/ops/argwhere.h‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,16 +5,16 @@
55

66
typedef struct InfiniopDescriptor *infiniopArgwhereDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateArgwhereDescriptor(
8+
__INFINI_C __export infiniStatus_t infiniopCreateArgwhereDescriptor(
99
infiniopHandle_t handle,
1010
infiniopArgwhereDescriptor_t *desc_ptr,
1111
infiniopTensorDescriptor_t input_desc);
1212

13-
__C __export infiniStatus_t infiniopGetArgwhereWorkspaceSize(
13+
__INFINI_C __export infiniStatus_t infiniopGetArgwhereWorkspaceSize(
1414
infiniopArgwhereDescriptor_t desc,
1515
size_t *size);
1616

17-
__C __export infiniStatus_t infiniopArgwhere(
17+
__INFINI_C __export infiniStatus_t infiniopArgwhere(
1818
infiniopArgwhereDescriptor_t desc,
1919
void *workspace,
2020
size_t workspace_size,
@@ -23,7 +23,7 @@ __C __export infiniStatus_t infiniopArgwhere(
2323
const void *input,
2424
void *stream);
2525

26-
__C __export infiniStatus_t infiniopDestroyArgwhereDescriptor(
26+
__INFINI_C __export infiniStatus_t infiniopDestroyArgwhereDescriptor(
2727
infiniopArgwhereDescriptor_t desc);
2828

29-
#endif // __INFINIOP_ARGWHERE_API_H__
29+
#endif // __INFINIOP_ARGWHERE_API_H__

‎include/infiniop/ops/asin.h‎

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -5,20 +5,20 @@
55

66
typedef struct InfiniopDescriptor *infiniopAsinDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateAsinDescriptor(infiniopHandle_t handle,
9-
infiniopAsinDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t output,
11-
infiniopTensorDescriptor_t input);
8+
__INFINI_C __export infiniStatus_t infiniopCreateAsinDescriptor(infiniopHandle_t handle,
9+
infiniopAsinDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t output,
11+
infiniopTensorDescriptor_t input);
1212

13-
__C __export infiniStatus_t infiniopGetAsinWorkspaceSize(infiniopAsinDescriptor_t desc, size_t *size);
13+
__INFINI_C __export infiniStatus_t infiniopGetAsinWorkspaceSize(infiniopAsinDescriptor_t desc, size_t *size);
1414

15-
__C __export infiniStatus_t infiniopAsin(infiniopAsinDescriptor_t desc,
16-
void *workspace,
17-
size_t workspace_size,
18-
void *output,
19-
const void *input,
20-
void *stream);
15+
__INFINI_C __export infiniStatus_t infiniopAsin(infiniopAsinDescriptor_t desc,
16+
void *workspace,
17+
size_t workspace_size,
18+
void *output,
19+
const void *input,
20+
void *stream);
2121

22-
__C __export infiniStatus_t infiniopDestroyAsinDescriptor(infiniopAsinDescriptor_t desc);
22+
__INFINI_C __export infiniStatus_t infiniopDestroyAsinDescriptor(infiniopAsinDescriptor_t desc);
2323

24-
#endif
24+
#endif

0 commit comments

Comments
 (0)