Skip to content

Commit de55c1e

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-26
1 parent 2c3ccd3 commit de55c1e

File tree

75 files changed

+1158
-1098
lines changed

Some content is hidden

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

75 files changed

+1158
-1098
lines changed

‎include/infinicore/ops/broadcast_to.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ class BroadcastTo {
1212
static void execute(Tensor y, Tensor x);
1313
static common::OpDispatcher<schema> &dispatcher();
1414
};
15-
Tensor broadcast_to(Tensor x, const std::vector<int64_t>& shape);
15+
Tensor broadcast_to(Tensor x, const std::vector<int64_t> &shape);
1616
void broadcast_to_(Tensor y, Tensor x);
1717

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

‎include/infinicore/ops/huber_loss.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ class HuberLoss {
99
public:
1010
// Schema: output, input, target, delta, reduction
1111
using schema = void (*)(Tensor, Tensor, Tensor, float, int64_t);
12-
12+
1313
static void execute(Tensor output, Tensor input, Tensor target, float delta, int64_t reduction);
1414
static common::OpDispatcher<schema> &dispatcher();
1515
};
@@ -18,4 +18,4 @@ class HuberLoss {
1818
Tensor huber_loss(Tensor input, Tensor target, float delta = 1.0f, int64_t reduction = 1);
1919
void huber_loss_(Tensor output, Tensor input, Tensor target, float delta, int64_t reduction);
2020

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

‎include/infinicore/ops/linear.hpp‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
#include "common/op.hpp"
44
#include <optional>
55

6-
76
namespace infinicore::op {
87

98
Tensor linear(Tensor input, Tensor weight, std::optional<Tensor> bias);

‎include/infinicore/ops/softplus.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,4 +17,4 @@ Tensor softplus(Tensor x, float beta = 1.0f, float threshold = 20.0f);
1717

1818
void softplus_(Tensor y, Tensor x, float beta = 1.0f, float threshold = 20.0f);
1919

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

‎include/infinicore/ops/softsign.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,4 @@ class Softsign {
1313
// 返回新 Tensor 的函数接口
1414
Tensor softsign(Tensor x);
1515
void softsign_(Tensor y, Tensor x);
16-
} // namespace infinicore::op
16+
} // namespace infinicore::op

‎include/infiniop.h‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
#include "infiniop/ops/attention.h"
2020
#include "infiniop/ops/avg_pool1d.h"
2121
#include "infiniop/ops/binary_cross_entropy_with_logits.h"
22+
#include "infiniop/ops/broadcast_to.h"
2223
#include "infiniop/ops/causal_softmax.h"
2324
#include "infiniop/ops/cdist.h"
2425
#include "infiniop/ops/clip.h"
@@ -40,6 +41,7 @@
4041
#include "infiniop/ops/gemm.h"
4142
#include "infiniop/ops/hardswish.h"
4243
#include "infiniop/ops/hardtanh.h"
44+
#include "infiniop/ops/huber_loss.h"
4345
#include "infiniop/ops/hypot.h"
4446
#include "infiniop/ops/index_add.h"
4547
#include "infiniop/ops/index_copy.h"
@@ -73,6 +75,8 @@
7375
#include "infiniop/ops/silu_and_mul.h"
7476
#include "infiniop/ops/smooth_l1_loss.h"
7577
#include "infiniop/ops/softmax.h"
78+
#include "infiniop/ops/softplus.h"
79+
#include "infiniop/ops/softsign.h"
7680
#include "infiniop/ops/sub.h"
7781
#include "infiniop/ops/sum.h"
7882
#include "infiniop/ops/swiglu.h"
@@ -88,9 +92,5 @@
8892
#include "infiniop/ops/var.h"
8993
#include "infiniop/ops/var_mean.h"
9094
#include "infiniop/ops/zeros.h"
91-
#include "infiniop/ops/broadcast_to.h"
92-
#include "infiniop/ops/softplus.h"
93-
#include "infiniop/ops/softsign.h"
94-
#include "infiniop/ops/huber_loss.h"
9595
#include "infiniop/tensor_descriptor.h"
9696
#endif // __INFINIOP_API_H__

‎include/infiniop/ops/broadcast_to.h‎

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

66
typedef struct InfiniopDescriptor *infiniopBroadcastToDescriptor_t;
77

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

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

15-
__C __export infiniStatus_t infiniopBroadcastTo(infiniopBroadcastToDescriptor_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 infiniopBroadcastTo(infiniopBroadcastToDescriptor_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 infiniopDestroyBroadcastToDescriptor(infiniopBroadcastToDescriptor_t desc);
22+
__INFINI_C __export infiniStatus_t infiniopDestroyBroadcastToDescriptor(infiniopBroadcastToDescriptor_t desc);
2323

24-
#endif
24+
#endif

‎include/infiniop/ops/huber_loss.h‎

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

66
typedef struct InfiniopDescriptor *infiniopHuberLossDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateHuberLossDescriptor(infiniopHandle_t handle,
9-
infiniopHuberLossDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t output,
11-
infiniopTensorDescriptor_t input,
12-
infiniopTensorDescriptor_t target,
13-
float delta,
14-
int reduction);
8+
__INFINI_C __export infiniStatus_t infiniopCreateHuberLossDescriptor(infiniopHandle_t handle,
9+
infiniopHuberLossDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t output,
11+
infiniopTensorDescriptor_t input,
12+
infiniopTensorDescriptor_t target,
13+
float delta,
14+
int reduction);
1515

16-
__C __export infiniStatus_t infiniopGetHuberLossWorkspaceSize(infiniopHuberLossDescriptor_t desc, size_t *size);
16+
__INFINI_C __export infiniStatus_t infiniopGetHuberLossWorkspaceSize(infiniopHuberLossDescriptor_t desc, size_t *size);
1717

18-
__C __export infiniStatus_t infiniopHuberLoss(infiniopHuberLossDescriptor_t desc,
19-
void *workspace,
20-
size_t workspace_size,
21-
void *output,
22-
const void *input,
23-
const void *target,
24-
void *stream);
18+
__INFINI_C __export infiniStatus_t infiniopHuberLoss(infiniopHuberLossDescriptor_t desc,
19+
void *workspace,
20+
size_t workspace_size,
21+
void *output,
22+
const void *input,
23+
const void *target,
24+
void *stream);
2525

26-
__C __export infiniStatus_t infiniopDestroyHuberLossDescriptor(infiniopHuberLossDescriptor_t desc);
26+
__INFINI_C __export infiniStatus_t infiniopDestroyHuberLossDescriptor(infiniopHuberLossDescriptor_t desc);
2727

28-
#endif // __INFINIOP_HUBER_LOSS_API_H__
28+
#endif // __INFINIOP_HUBER_LOSS_API_H__

‎include/infiniop/ops/softplus.h‎

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -2,31 +2,28 @@
22
#define __INFINIOP_OPS_SOFTPLUS_API_H__
33
#include "../tensor_descriptor.h"
44

5-
6-
typedef struct InfiniopSoftplusDescriptor *infiniopSoftplusDescriptor_t;
7-
__C __export infiniStatus_t infiniopCreateSoftplusDescriptor(
5+
typedef struct InfiniopDescriptor *infiniopSoftplusDescriptor_t;
6+
__INFINI_C __export infiniStatus_t infiniopCreateSoftplusDescriptor(
87
infiniopHandle_t handle,
98
infiniopSoftplusDescriptor_t *desc_ptr,
109
infiniopTensorDescriptor_t y_desc,
1110
infiniopTensorDescriptor_t x_desc,
12-
float beta,
13-
float threshold
14-
);
11+
float beta,
12+
float threshold);
1513

16-
__C __export infiniStatus_t infiniopGetSoftplusWorkspaceSize(
17-
infiniopSoftplusDescriptor_t desc,
14+
__INFINI_C __export infiniStatus_t infiniopGetSoftplusWorkspaceSize(
15+
infiniopSoftplusDescriptor_t desc,
1816
size_t *size);
1917

20-
__C __export infiniStatus_t infiniopSoftplus(
18+
__INFINI_C __export infiniStatus_t infiniopSoftplus(
2119
infiniopSoftplusDescriptor_t desc,
2220
void *workspace,
2321
size_t workspace_size,
2422
void *y,
2523
const void *x,
2624
void *stream);
2725

28-
__C __export infiniStatus_t infiniopDestroySoftplusDescriptor(
26+
__INFINI_C __export infiniStatus_t infiniopDestroySoftplusDescriptor(
2927
infiniopSoftplusDescriptor_t desc);
3028

31-
32-
#endif // __INFINIOP_OPS_SOFTPLUS_API_H__
29+
#endif // __INFINIOP_OPS_SOFTPLUS_API_H__

‎include/infiniop/ops/softsign.h‎

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

77
typedef struct InfiniopDescriptor *infiniopSoftsignDescriptor_t;
88

9-
__C __export infiniStatus_t infiniopCreateSoftsignDescriptor(infiniopHandle_t handle,
10-
infiniopSoftsignDescriptor_t *desc_ptr,
11-
infiniopTensorDescriptor_t y,
12-
infiniopTensorDescriptor_t x);
9+
__INFINI_C __export infiniStatus_t infiniopCreateSoftsignDescriptor(infiniopHandle_t handle,
10+
infiniopSoftsignDescriptor_t *desc_ptr,
11+
infiniopTensorDescriptor_t y,
12+
infiniopTensorDescriptor_t x);
1313

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

16-
__C __export infiniStatus_t infiniopSoftsign(infiniopSoftsignDescriptor_t desc,
17-
void *workspace,
18-
size_t workspace_size,
19-
void *y,
20-
const void *x,
21-
void *stream);
16+
__INFINI_C __export infiniStatus_t infiniopSoftsign(infiniopSoftsignDescriptor_t desc,
17+
void *workspace,
18+
size_t workspace_size,
19+
void *y,
20+
const void *x,
21+
void *stream);
2222

23-
__C __export infiniStatus_t infiniopDestroySoftsignDescriptor(infiniopSoftsignDescriptor_t desc);
23+
__INFINI_C __export infiniStatus_t infiniopDestroySoftsignDescriptor(infiniopSoftsignDescriptor_t desc);
2424

25-
#endif
25+
#endif

0 commit comments

Comments
 (0)