Skip to content

Commit 9b2992e

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-31
1 parent e1aafb1 commit 9b2992e

File tree

85 files changed

+650
-558
lines changed

Some content is hidden

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

85 files changed

+650
-558
lines changed

‎include/infinicore/ops/block_diag.hpp‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,4 +18,3 @@ Tensor block_diag(const std::vector<Tensor> &inputs);
1818
void block_diag_(Tensor output, const std::vector<Tensor> &inputs);
1919

2020
} // namespace infinicore::op
21-

‎include/infinicore/ops/hinge_embedding_loss.hpp‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,4 +16,3 @@ Tensor hinge_embedding_loss(Tensor input, Tensor target, double margin = 1.0, in
1616
void hinge_embedding_loss_(Tensor output, Tensor input, Tensor target, double margin, int reduction);
1717

1818
} // namespace infinicore::op
19-

‎include/infinicore/ops/kron.hpp‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,4 +16,3 @@ Tensor kron(Tensor a, Tensor b);
1616
void kron_(Tensor output, Tensor a, Tensor b);
1717

1818
} // namespace infinicore::op
19-

‎include/infinicore/ops/selu.hpp‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,4 +16,3 @@ Tensor selu(Tensor input);
1616
void selu_(Tensor output, Tensor input);
1717

1818
} // namespace infinicore::op
19-

‎include/infinicore/ops/sinh.hpp‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,4 +16,3 @@ Tensor sinh(Tensor input);
1616
void sinh_(Tensor output, Tensor input);
1717

1818
} // namespace infinicore::op
19-

‎include/infiniop.h‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@
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"
2322
#include "infiniop/ops/block_diag.h"
23+
#include "infiniop/ops/broadcast_to.h"
2424
#include "infiniop/ops/causal_softmax.h"
2525
#include "infiniop/ops/cdist.h"
2626
#include "infiniop/ops/clip.h"
@@ -42,15 +42,15 @@
4242
#include "infiniop/ops/gemm.h"
4343
#include "infiniop/ops/hardswish.h"
4444
#include "infiniop/ops/hardtanh.h"
45+
#include "infiniop/ops/hinge_embedding_loss.h"
4546
#include "infiniop/ops/huber_loss.h"
4647
#include "infiniop/ops/hypot.h"
4748
#include "infiniop/ops/index_add.h"
4849
#include "infiniop/ops/index_copy.h"
4950
#include "infiniop/ops/inner.h"
5051
#include "infiniop/ops/int8_gemm.h"
51-
#include "infiniop/ops/kthvalue.h"
52-
#include "infiniop/ops/hinge_embedding_loss.h"
5352
#include "infiniop/ops/kron.h"
53+
#include "infiniop/ops/kthvalue.h"
5454
#include "infiniop/ops/kv_caching.h"
5555
#include "infiniop/ops/layer_norm.h"
5656
#include "infiniop/ops/ldexp.h"
@@ -75,9 +75,9 @@
7575
#include "infiniop/ops/scatter.h"
7676
#include "infiniop/ops/selu.h"
7777
#include "infiniop/ops/sigmoid.h"
78-
#include "infiniop/ops/sinh.h"
7978
#include "infiniop/ops/silu.h"
8079
#include "infiniop/ops/silu_and_mul.h"
80+
#include "infiniop/ops/sinh.h"
8181
#include "infiniop/ops/smooth_l1_loss.h"
8282
#include "infiniop/ops/softmax.h"
8383
#include "infiniop/ops/softplus.h"

‎include/infiniop/ops/block_diag.h‎

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

66
typedef struct InfiniopDescriptor *infiniopBlockDiagDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateBlockDiagDescriptor(infiniopHandle_t handle,
9-
infiniopBlockDiagDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t *x,
12-
size_t num_inputs);
8+
__INFINI_C __export infiniStatus_t infiniopCreateBlockDiagDescriptor(infiniopHandle_t handle,
9+
infiniopBlockDiagDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t *x,
12+
size_t num_inputs);
1313

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

16-
__C __export infiniStatus_t infiniopBlockDiag(infiniopBlockDiagDescriptor_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 infiniopBlockDiag(infiniopBlockDiagDescriptor_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 infiniopDestroyBlockDiagDescriptor(infiniopBlockDiagDescriptor_t desc);
23+
__INFINI_C __export infiniStatus_t infiniopDestroyBlockDiagDescriptor(infiniopBlockDiagDescriptor_t desc);
2424

2525
#endif

‎include/infiniop/ops/hinge_embedding_loss.h‎

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

66
typedef struct InfiniopDescriptor *infiniopHingeEmbeddingLossDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateHingeEmbeddingLossDescriptor(infiniopHandle_t handle,
9-
infiniopHingeEmbeddingLossDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t input,
12-
infiniopTensorDescriptor_t target,
13-
double margin,
14-
int reduction);
8+
__INFINI_C __export infiniStatus_t infiniopCreateHingeEmbeddingLossDescriptor(infiniopHandle_t handle,
9+
infiniopHingeEmbeddingLossDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t input,
12+
infiniopTensorDescriptor_t target,
13+
double margin,
14+
int reduction);
1515

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

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

26-
__C __export infiniStatus_t infiniopDestroyHingeEmbeddingLossDescriptor(infiniopHingeEmbeddingLossDescriptor_t desc);
26+
__INFINI_C __export infiniStatus_t infiniopDestroyHingeEmbeddingLossDescriptor(infiniopHingeEmbeddingLossDescriptor_t desc);
2727

2828
#endif

‎include/infiniop/ops/kron.h‎

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

66
typedef struct InfiniopDescriptor *infiniopKronDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateKronDescriptor(infiniopHandle_t handle,
9-
infiniopKronDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t x1,
12-
infiniopTensorDescriptor_t x2);
8+
__INFINI_C __export infiniStatus_t infiniopCreateKronDescriptor(infiniopHandle_t handle,
9+
infiniopKronDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x1,
12+
infiniopTensorDescriptor_t x2);
1313

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

16-
__C __export infiniStatus_t infiniopKron(infiniopKronDescriptor_t desc,
17-
void *workspace,
18-
size_t workspace_size,
19-
void *y,
20-
const void *x1,
21-
const void *x2,
22-
void *stream);
16+
__INFINI_C __export infiniStatus_t infiniopKron(infiniopKronDescriptor_t desc,
17+
void *workspace,
18+
size_t workspace_size,
19+
void *y,
20+
const void *x1,
21+
const void *x2,
22+
void *stream);
2323

24-
__C __export infiniStatus_t infiniopDestroyKronDescriptor(infiniopKronDescriptor_t desc);
24+
__INFINI_C __export infiniStatus_t infiniopDestroyKronDescriptor(infiniopKronDescriptor_t desc);
2525

2626
#endif

‎include/infiniop/ops/selu.h‎

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

66
typedef struct InfiniopDescriptor *infiniopSeluDescriptor_t;
77

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

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

15-
__C __export infiniStatus_t infiniopSelu(infiniopSeluDescriptor_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 infiniopSelu(infiniopSeluDescriptor_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 infiniopDestroySeluDescriptor(infiniopSeluDescriptor_t desc);
22+
__INFINI_C __export infiniStatus_t infiniopDestroySeluDescriptor(infiniopSeluDescriptor_t desc);
2323

2424
#endif

0 commit comments

Comments
 (0)