Skip to content

Commit 4621503

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-40
1 parent 51f1c4c commit 4621503

File tree

97 files changed

+1887
-1836
lines changed

Some content is hidden

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

97 files changed

+1887
-1836
lines changed

‎include/infinicore/ops/hypot.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,12 @@ namespace infinicore::op {
77
class Hypot {
88
public:
99
using schema = void (*)(Tensor, Tensor, Tensor);
10-
10+
1111
static void execute(Tensor output, Tensor input_a, Tensor input_b);
1212
static common::OpDispatcher<schema> &dispatcher();
1313
};
1414

1515
Tensor hypot(Tensor input_a, Tensor input_b);
1616

1717
void hypot_(Tensor output, Tensor input_a, Tensor input_b);
18-
} // namespace infinicore::op
18+
} // namespace infinicore::op

‎include/infinicore/ops/index_add.hpp‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,7 @@ class IndexAdd {
1313
static common::OpDispatcher<schema> &dispatcher();
1414
};
1515

16-
1716
Tensor index_add(Tensor input, int64_t dim, Tensor index, Tensor source, float alpha = 1.0f);
1817
void index_add_(Tensor output, Tensor input, int64_t dim, Tensor index, Tensor source, float alpha);
1918

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

‎include/infinicore/ops/index_copy.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,4 @@ class IndexCopy {
1515
Tensor index_copy(Tensor input, int64_t dim, Tensor index, Tensor source);
1616
void index_copy_(Tensor output, Tensor input, int64_t dim, Tensor index, Tensor source);
1717

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

‎include/infinicore/ops/smooth_l1_loss.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,12 @@ namespace infinicore::op {
88
class SmoothL1Loss {
99
public:
1010
using schema = void (*)(Tensor, Tensor, Tensor, float, int64_t);
11-
11+
1212
static void execute(Tensor output, Tensor input, Tensor target, float beta, int64_t reduction);
1313
static common::OpDispatcher<schema> &dispatcher();
1414
};
1515

1616
Tensor smooth_l1_loss(Tensor input, Tensor target, float beta = 1.0f, int64_t reduction = 1);
1717
void smooth_l1_loss_(Tensor output, Tensor input, Tensor target, float beta, int64_t reduction);
1818

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

‎include/infinicore/ops/take.hpp‎

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,14 +8,13 @@ namespace infinicore::op {
88
class Take {
99
public:
1010
using schema = void (*)(Tensor, Tensor, Tensor);
11-
11+
1212
static void execute(Tensor output, Tensor input, Tensor indices);
1313
static common::OpDispatcher<schema> &dispatcher();
1414
};
1515

16-
1716
Tensor take(Tensor input, Tensor indices);
1817

1918
void take_(Tensor output, Tensor input, Tensor indices);
2019

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

‎include/infiniop.h‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,20 +19,20 @@
1919
#include "infiniop/ops/cross_entropy.h"
2020
#include "infiniop/ops/dequant/per_tensor_dequant_int8.h"
2121
#include "infiniop/ops/dequantize_awq.h"
22+
#include "infiniop/ops/dequantize_gptq.h"
2223
#include "infiniop/ops/embedding.h"
2324
#include "infiniop/ops/equal.h"
2425
#include "infiniop/ops/flash_attention.h"
2526
#include "infiniop/ops/fmod.h"
26-
#include "infiniop/ops/dequantize_gptq.h"
2727
#include "infiniop/ops/gelu.h"
2828
#include "infiniop/ops/gemm.h"
2929
#include "infiniop/ops/hardswish.h"
3030
#include "infiniop/ops/hardtanh.h"
31-
#include "infiniop/ops/int8_gemm.h"
32-
#include "infiniop/ops/kv_caching.h"
3331
#include "infiniop/ops/hypot.h"
3432
#include "infiniop/ops/index_add.h"
3533
#include "infiniop/ops/index_copy.h"
34+
#include "infiniop/ops/int8_gemm.h"
35+
#include "infiniop/ops/kv_caching.h"
3636
#include "infiniop/ops/layer_norm.h"
3737
#include "infiniop/ops/logsoftmax.h"
3838
#include "infiniop/ops/lp_norm.h"
@@ -58,9 +58,9 @@
5858
#include "infiniop/ops/sub.h"
5959
#include "infiniop/ops/sum.h"
6060
#include "infiniop/ops/swiglu.h"
61+
#include "infiniop/ops/take.h"
6162
#include "infiniop/ops/tanh.h"
6263
#include "infiniop/ops/topk.h"
63-
#include "infiniop/ops/take.h"
6464
#include "infiniop/ops/topkrouter.h"
6565
#include "infiniop/ops/topksoftmax.h"
6666
#include "infiniop/ops/var.h"

‎include/infiniop/ops/hypot.h‎

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

66
typedef struct InfiniopDescriptor *infiniopHypotDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateHypotDescriptor(infiniopHandle_t handle,
9-
infiniopHypotDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t output,
11-
infiniopTensorDescriptor_t input_a,
12-
infiniopTensorDescriptor_t input_b);
8+
__INFINI_C __export infiniStatus_t infiniopCreateHypotDescriptor(infiniopHandle_t handle,
9+
infiniopHypotDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t output,
11+
infiniopTensorDescriptor_t input_a,
12+
infiniopTensorDescriptor_t input_b);
1313

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

16-
__C __export infiniStatus_t infiniopHypot(infiniopHypotDescriptor_t desc,
17-
void *workspace,
18-
size_t workspace_size,
19-
void *output,
20-
const void *input_a,
21-
const void *input_b,
22-
void *stream);
16+
__INFINI_C __export infiniStatus_t infiniopHypot(infiniopHypotDescriptor_t desc,
17+
void *workspace,
18+
size_t workspace_size,
19+
void *output,
20+
const void *input_a,
21+
const void *input_b,
22+
void *stream);
2323

24-
__C __export infiniStatus_t infiniopDestroyHypotDescriptor(infiniopHypotDescriptor_t desc);
24+
__INFINI_C __export infiniStatus_t infiniopDestroyHypotDescriptor(infiniopHypotDescriptor_t desc);
2525

26-
#endif
26+
#endif

‎include/infiniop/ops/index_add.h‎

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

66
typedef struct InfiniopDescriptor *infiniopIndexAddDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateIndexAddDescriptor(infiniopHandle_t handle,
9-
infiniopIndexAddDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t output,
11-
infiniopTensorDescriptor_t input,
12-
int64_t dim,
13-
infiniopTensorDescriptor_t index,
14-
infiniopTensorDescriptor_t source,
15-
float alpha);
8+
__INFINI_C __export infiniStatus_t infiniopCreateIndexAddDescriptor(infiniopHandle_t handle,
9+
infiniopIndexAddDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t output,
11+
infiniopTensorDescriptor_t input,
12+
int64_t dim,
13+
infiniopTensorDescriptor_t index,
14+
infiniopTensorDescriptor_t source,
15+
float alpha);
1616

17-
__C __export infiniStatus_t infiniopGetIndexAddWorkspaceSize(infiniopIndexAddDescriptor_t desc, size_t *size);
18-
__C __export infiniStatus_t infiniopIndexAdd(infiniopIndexAddDescriptor_t desc,
19-
void *workspace,
20-
size_t workspace_size,
21-
void *output,
22-
const void *input,
23-
const void *index,
24-
const void *source,
25-
void *stream);
17+
__INFINI_C __export infiniStatus_t infiniopGetIndexAddWorkspaceSize(infiniopIndexAddDescriptor_t desc, size_t *size);
18+
__INFINI_C __export infiniStatus_t infiniopIndexAdd(infiniopIndexAddDescriptor_t desc,
19+
void *workspace,
20+
size_t workspace_size,
21+
void *output,
22+
const void *input,
23+
const void *index,
24+
const void *source,
25+
void *stream);
2626

27-
__C __export infiniStatus_t infiniopDestroyIndexAddDescriptor(infiniopIndexAddDescriptor_t desc);
27+
__INFINI_C __export infiniStatus_t infiniopDestroyIndexAddDescriptor(infiniopIndexAddDescriptor_t desc);
2828

29-
#endif
29+
#endif

‎include/infiniop/ops/index_copy.h‎

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

66
typedef struct InfiniopDescriptor *infiniopIndexCopyDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateIndexCopyDescriptor(infiniopHandle_t handle,
9-
infiniopIndexCopyDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t output,
11-
infiniopTensorDescriptor_t input,
12-
int64_t dim,
13-
infiniopTensorDescriptor_t index,
14-
infiniopTensorDescriptor_t source);
8+
__INFINI_C __export infiniStatus_t infiniopCreateIndexCopyDescriptor(infiniopHandle_t handle,
9+
infiniopIndexCopyDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t output,
11+
infiniopTensorDescriptor_t input,
12+
int64_t dim,
13+
infiniopTensorDescriptor_t index,
14+
infiniopTensorDescriptor_t source);
1515

16-
__C __export infiniStatus_t infiniopGetIndexCopyWorkspaceSize(infiniopIndexCopyDescriptor_t desc, size_t *size);
17-
__C __export infiniStatus_t infiniopIndexCopy(infiniopIndexCopyDescriptor_t desc,
18-
void *workspace,
19-
size_t workspace_size,
20-
void *output,
21-
const void *input,
22-
const void *index,
23-
const void *source,
24-
void *stream);
16+
__INFINI_C __export infiniStatus_t infiniopGetIndexCopyWorkspaceSize(infiniopIndexCopyDescriptor_t desc, size_t *size);
17+
__INFINI_C __export infiniStatus_t infiniopIndexCopy(infiniopIndexCopyDescriptor_t desc,
18+
void *workspace,
19+
size_t workspace_size,
20+
void *output,
21+
const void *input,
22+
const void *index,
23+
const void *source,
24+
void *stream);
2525

26-
__C __export infiniStatus_t infiniopDestroyIndexCopyDescriptor(infiniopIndexCopyDescriptor_t desc);
26+
__INFINI_C __export infiniStatus_t infiniopDestroyIndexCopyDescriptor(infiniopIndexCopyDescriptor_t desc);
2727

28-
#endif
28+
#endif

‎include/infiniop/ops/smooth_l1_loss.h‎

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -4,24 +4,24 @@
44
#include "../operator_descriptor.h"
55

66
typedef struct InfiniopDescriptor *infiniopSmoothL1LossDescriptor_t;
7-
__C __export infiniStatus_t infiniopCreateSmoothL1LossDescriptor(infiniopHandle_t handle,
8-
infiniopSmoothL1LossDescriptor_t *desc_ptr,
9-
infiniopTensorDescriptor_t output,
10-
infiniopTensorDescriptor_t input,
11-
infiniopTensorDescriptor_t target,
12-
float beta,
13-
int reduction);
7+
__INFINI_C __export infiniStatus_t infiniopCreateSmoothL1LossDescriptor(infiniopHandle_t handle,
8+
infiniopSmoothL1LossDescriptor_t *desc_ptr,
9+
infiniopTensorDescriptor_t output,
10+
infiniopTensorDescriptor_t input,
11+
infiniopTensorDescriptor_t target,
12+
float beta,
13+
int reduction);
1414

15-
__C __export infiniStatus_t infiniopGetSmoothL1LossWorkspaceSize(infiniopSmoothL1LossDescriptor_t desc, size_t *size);
15+
__INFINI_C __export infiniStatus_t infiniopGetSmoothL1LossWorkspaceSize(infiniopSmoothL1LossDescriptor_t desc, size_t *size);
1616

17-
__C __export infiniStatus_t infiniopSmoothL1Loss(infiniopSmoothL1LossDescriptor_t desc,
18-
void *workspace,
19-
size_t workspace_size,
20-
void *output,
21-
const void *input,
22-
const void *target,
23-
void *stream);
17+
__INFINI_C __export infiniStatus_t infiniopSmoothL1Loss(infiniopSmoothL1LossDescriptor_t desc,
18+
void *workspace,
19+
size_t workspace_size,
20+
void *output,
21+
const void *input,
22+
const void *target,
23+
void *stream);
2424

25-
__C __export infiniStatus_t infiniopDestroySmoothL1LossDescriptor(infiniopSmoothL1LossDescriptor_t desc);
25+
__INFINI_C __export infiniStatus_t infiniopDestroySmoothL1LossDescriptor(infiniopSmoothL1LossDescriptor_t desc);
2626

27-
#endif // __INFINIOP_SMOOTH_L1_LOSS_API_H__
27+
#endif // __INFINIOP_SMOOTH_L1_LOSS_API_H__

0 commit comments

Comments
 (0)