Skip to content

Commit 332c0a9

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-5
1 parent 23c8fcd commit 332c0a9

93 files changed

Lines changed: 1296 additions & 1308 deletions

File tree

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/acos.hpp‎

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

1414
Tensor acos(Tensor input);
1515
void acos_(Tensor output, Tensor input);
16-
} // namespace infinicore::op
16+
} // namespace infinicore::op

‎include/infinicore/ops/adaptive_avg_pool1d.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,4 @@ class AdaptiveAvgPool1d {
1515

1616
Tensor adaptive_avg_pool1d(Tensor input, int64_t output_size);
1717

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

‎include/infinicore/ops/addbmm.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,11 @@ class Addbmm {
99
public:
1010
using schema = void (*)(Tensor, Tensor, Tensor, Tensor, float, float);
1111
static void execute(Tensor output, Tensor input, Tensor batch1, Tensor batch2, float beta, float alpha);
12-
12+
1313
static common::OpDispatcher<schema> &dispatcher();
1414
};
1515
Tensor addbmm(Tensor input, Tensor batch1, Tensor batch2, float beta = 1.0f, float alpha = 1.0f);
1616

1717
void addbmm_(Tensor output, Tensor input, Tensor batch1, Tensor batch2, float beta, float alpha);
1818

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

‎include/infinicore/ops/affine_grid.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,6 @@ class AffineGrid {
1212
static void execute(Tensor output, Tensor theta, bool align_corners);
1313
static common::OpDispatcher<schema> &dispatcher();
1414
};
15-
Tensor affine_grid(Tensor theta, const std::vector<int64_t>& size, bool align_corners = false);
15+
Tensor affine_grid(Tensor theta, const std::vector<int64_t> &size, bool align_corners = false);
1616

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

‎include/infinicore/ops/floor.hpp‎

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

1414
Tensor floor(Tensor input);
1515
void floor_(Tensor output, Tensor input);
16-
} // namespace infinicore::op
16+
} // namespace infinicore::op

‎include/infiniop.h‎

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

44
#include "infiniop/handle.h"
5+
#include "infiniop/ops/acos.h"
6+
#include "infiniop/ops/adaptive_avg_pool1d.h"
57
#include "infiniop/ops/adaptive_avg_pool3d.h"
68
#include "infiniop/ops/adaptive_max_pool1d.h"
79
#include "infiniop/ops/add.h"
810
#include "infiniop/ops/add_rms_norm.h"
11+
#include "infiniop/ops/addbmm.h"
912
#include "infiniop/ops/addcmul.h"
1013
#include "infiniop/ops/addr.h"
14+
#include "infiniop/ops/affine_grid.h"
1115
#include "infiniop/ops/all.h"
1216
#include "infiniop/ops/asin.h"
1317
#include "infiniop/ops/asinh.h"
1418
#include "infiniop/ops/atanh.h"
15-
#include "infiniop/ops/acos.h"
16-
#include "infiniop/ops/add.h"
17-
#include "infiniop/ops/adaptive_avg_pool1d.h"
18-
#include "infiniop/ops/addbmm.h"
19-
#include "infiniop/ops/affine_grid.h"
20-
#include "infiniop/ops/floor.h"
2119
#include "infiniop/ops/attention.h"
2220
#include "infiniop/ops/avg_pool1d.h"
2321
#include "infiniop/ops/binary_cross_entropy_with_logits.h"
@@ -32,6 +30,7 @@
3230
#include "infiniop/ops/embedding.h"
3331
#include "infiniop/ops/equal.h"
3432
#include "infiniop/ops/flash_attention.h"
33+
#include "infiniop/ops/floor.h"
3534
#include "infiniop/ops/fmin.h"
3635
#include "infiniop/ops/fmod.h"
3736
#include "infiniop/ops/gelu.h"

‎include/infiniop/ops/acos.h‎

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

66
typedef struct InfiniopDescriptor *infiniopAcosDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateAcosDescriptor(infiniopHandle_t handle,
9-
infiniopAcosDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t x );
8+
__INFINI_C __export infiniStatus_t infiniopCreateAcosDescriptor(infiniopHandle_t handle,
9+
infiniopAcosDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x);
1212

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

15-
__C __export infiniStatus_t infiniopAcos(infiniopAcosDescriptor_t desc,
16-
void *workspace,
17-
size_t workspace_size,
18-
void *y,
19-
const void *x,
20-
void *stream);
15+
__INFINI_C __export infiniStatus_t infiniopAcos(infiniopAcosDescriptor_t desc,
16+
void *workspace,
17+
size_t workspace_size,
18+
void *y,
19+
const void *x,
20+
void *stream);
2121

22-
__C __export infiniStatus_t infiniopDestroyAcosDescriptor(infiniopAcosDescriptor_t desc);
22+
__INFINI_C __export infiniStatus_t infiniopDestroyAcosDescriptor(infiniopAcosDescriptor_t desc);
2323

24-
#endif
24+
#endif

‎include/infiniop/ops/adaptive_avg_pool1d.h‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,19 +7,19 @@
77
typedef struct InfiniopDescriptor *infiniopAdaptiveAvgPool1dDescriptor_t;
88

99
// 1. 创建算子描述符
10-
__C __export infiniStatus_t infiniopCreateAdaptiveAvgPool1dDescriptor(
10+
__INFINI_C __export infiniStatus_t infiniopCreateAdaptiveAvgPool1dDescriptor(
1111
infiniopHandle_t handle,
1212
infiniopAdaptiveAvgPool1dDescriptor_t *desc_ptr,
1313
infiniopTensorDescriptor_t out_desc,
1414
infiniopTensorDescriptor_t input_desc);
1515

1616
// 2. 获取 Workspace 大小
17-
__C __export infiniStatus_t infiniopGetAdaptiveAvgPool1dWorkspaceSize(
18-
infiniopAdaptiveAvgPool1dDescriptor_t desc,
17+
__INFINI_C __export infiniStatus_t infiniopGetAdaptiveAvgPool1dWorkspaceSize(
18+
infiniopAdaptiveAvgPool1dDescriptor_t desc,
1919
size_t *size);
2020

2121
// 3. 执行计算
22-
__C __export infiniStatus_t infiniopAdaptiveAvgPool1d(
22+
__INFINI_C __export infiniStatus_t infiniopAdaptiveAvgPool1d(
2323
infiniopAdaptiveAvgPool1dDescriptor_t desc,
2424
void *workspace,
2525
size_t workspace_size,
@@ -28,7 +28,7 @@ __C __export infiniStatus_t infiniopAdaptiveAvgPool1d(
2828
void *stream);
2929

3030
// 4. 销毁描述符
31-
__C __export infiniStatus_t infiniopDestroyAdaptiveAvgPool1dDescriptor(
31+
__INFINI_C __export infiniStatus_t infiniopDestroyAdaptiveAvgPool1dDescriptor(
3232
infiniopAdaptiveAvgPool1dDescriptor_t desc);
3333

34-
#endif // __INFINIOP_ADAPTIVE_AVG_POOL1D_API_H__
34+
#endif // __INFINIOP_ADAPTIVE_AVG_POOL1D_API_H__

‎include/infiniop/ops/addbmm.h‎

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

66
typedef struct InfiniopDescriptor *infiniopAddbmmDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateAddbmmDescriptor(infiniopHandle_t handle,
9-
infiniopAddbmmDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t out_desc,
11-
infiniopTensorDescriptor_t input_desc,
12-
infiniopTensorDescriptor_t batch1_desc,
13-
infiniopTensorDescriptor_t batch2_desc,
14-
float alpha,
15-
float beta);
8+
__INFINI_C __export infiniStatus_t infiniopCreateAddbmmDescriptor(infiniopHandle_t handle,
9+
infiniopAddbmmDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t out_desc,
11+
infiniopTensorDescriptor_t input_desc,
12+
infiniopTensorDescriptor_t batch1_desc,
13+
infiniopTensorDescriptor_t batch2_desc,
14+
float alpha,
15+
float beta);
1616

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

19-
__C __export infiniStatus_t infiniopAddbmm(infiniopAddbmmDescriptor_t desc,
20-
void *workspace,
21-
size_t workspace_size,
22-
void *output,
23-
const void *input,
24-
const void *batch1,
25-
const void *batch2,
26-
void *stream);
19+
__INFINI_C __export infiniStatus_t infiniopAddbmm(infiniopAddbmmDescriptor_t desc,
20+
void *workspace,
21+
size_t workspace_size,
22+
void *output,
23+
const void *input,
24+
const void *batch1,
25+
const void *batch2,
26+
void *stream);
2727

28-
__C __export infiniStatus_t infiniopDestroyAddbmmDescriptor(infiniopAddbmmDescriptor_t desc);
28+
__INFINI_C __export infiniStatus_t infiniopDestroyAddbmmDescriptor(infiniopAddbmmDescriptor_t desc);
2929

30-
#endif // __INFINIOP_ADDBMM_API_H__
30+
#endif // __INFINIOP_ADDBMM_API_H__

‎include/infiniop/ops/affine_grid.h‎

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,25 +1,25 @@
11
#ifndef __INFINIOP_AFFINE_GRID_API_H__
22
#define __INFINIOP_AFFINE_GRID_API_H__
3-
#include <stdint.h>
43
#include "../operator_descriptor.h"
4+
#include <stdint.h>
55

66
typedef struct InfiniopDescriptor *infiniopAffineGridDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateAffineGridDescriptor(infiniopHandle_t handle,
9-
infiniopAffineGridDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t output_desc,
11-
infiniopTensorDescriptor_t input_desc,
12-
uint8_t align_corners);
8+
__INFINI_C __export infiniStatus_t infiniopCreateAffineGridDescriptor(infiniopHandle_t handle,
9+
infiniopAffineGridDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t output_desc,
11+
infiniopTensorDescriptor_t input_desc,
12+
uint8_t align_corners);
1313

14-
__C __export infiniStatus_t infiniopGetAffineGridWorkspaceSize(infiniopAffineGridDescriptor_t desc, size_t *size);
14+
__INFINI_C __export infiniStatus_t infiniopGetAffineGridWorkspaceSize(infiniopAffineGridDescriptor_t desc, size_t *size);
1515

16-
__C __export infiniStatus_t infiniopAffineGrid(infiniopAffineGridDescriptor_t desc,
17-
void *workspace,
18-
size_t workspace_size,
19-
void *output,
20-
const void *input,
21-
void *stream);
16+
__INFINI_C __export infiniStatus_t infiniopAffineGrid(infiniopAffineGridDescriptor_t desc,
17+
void *workspace,
18+
size_t workspace_size,
19+
void *output,
20+
const void *input,
21+
void *stream);
2222

23-
__C __export infiniStatus_t infiniopDestroyAffineGridDescriptor(infiniopAffineGridDescriptor_t desc);
23+
__INFINI_C __export infiniStatus_t infiniopDestroyAffineGridDescriptor(infiniopAffineGridDescriptor_t desc);
2424

25-
#endif // __INFINIOP_AFFINE_GRID_API_H__
25+
#endif // __INFINIOP_AFFINE_GRID_API_H__

0 commit comments

Comments
 (0)