Skip to content

Commit b9df56c

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-38
1 parent 24927de commit b9df56c

97 files changed

Lines changed: 1786 additions & 1738 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/flipud.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,11 @@ class Flipud {
99
public:
1010
// Schema signature: (Output, Input)
1111
using schema = void (*)(Tensor, Tensor);
12-
12+
1313
static void execute(Tensor output, Tensor input);
1414
static common::OpDispatcher<schema> &dispatcher();
1515
};
1616
Tensor flipud(Tensor input);
1717
void flipud_(Tensor output, Tensor input);
1818

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

‎include/infinicore/ops/float_power.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,8 @@ class FloatPower {
3737
// Dispatchers
3838
// ==========================================================
3939

40-
static common::OpDispatcher<schema_scalar>& dispatcher_scalar();
41-
static common::OpDispatcher<schema_tensor>& dispatcher_tensor();
40+
static common::OpDispatcher<schema_scalar> &dispatcher_scalar();
41+
static common::OpDispatcher<schema_tensor> &dispatcher_tensor();
4242
};
4343

4444
// =======================================================================

‎include/infinicore/ops/floor_divide.hpp‎

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

1414
Tensor floor_divide(Tensor a, Tensor b);
1515
void floor_divide_(Tensor c, Tensor a, Tensor b);
16-
} // namespace infinicore::op
16+
} // namespace infinicore::op

‎include/infinicore/ops/multi_margin_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 MultiMarginLoss {
99
public:
1010
using schema = void (*)(Tensor, Tensor, Tensor, Tensor, int64_t, float, int64_t);
11-
11+
1212
static void execute(Tensor output, Tensor input, Tensor target, Tensor weight, int64_t p, float margin, int64_t reduction);
1313
static common::OpDispatcher<schema> &dispatcher();
1414
};
1515

1616
Tensor multi_margin_loss(Tensor input, Tensor target, Tensor weight = {}, int64_t p = 1, float margin = 1.0f, int64_t reduction = 1);
1717
void multi_margin_loss_(Tensor output, Tensor input, Tensor target, Tensor weight, int64_t p, float margin, int64_t reduction);
1818

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

‎include/infinicore/ops/scatter.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ namespace infinicore::op {
88
class Scatter {
99
public:
1010
using schema = void (*)(Tensor, Tensor, int64_t, Tensor, Tensor, int64_t);
11-
11+
1212
static void execute(Tensor output, Tensor input, int64_t dim, Tensor index, Tensor src, int64_t reduction);
1313
static common::OpDispatcher<schema> &dispatcher();
1414
};
@@ -18,4 +18,4 @@ Tensor scatter(Tensor input, int64_t dim, Tensor index, Tensor src, int64_t redu
1818
// In-place / 指定 Output 接口
1919
void scatter_(Tensor output, Tensor input, int64_t dim, Tensor index, Tensor src, int64_t reduction);
2020

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

‎include/infiniop.h‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@
2626
#include "infiniop/ops/embedding.h"
2727
#include "infiniop/ops/equal.h"
2828
#include "infiniop/ops/flash_attention.h"
29+
#include "infiniop/ops/flipud.h"
30+
#include "infiniop/ops/float_power.h"
31+
#include "infiniop/ops/floor_divide.h"
2932
#include "infiniop/ops/fmin.h"
3033
#include "infiniop/ops/fmod.h"
3134
#include "infiniop/ops/gelu.h"
@@ -38,6 +41,7 @@
3841
#include "infiniop/ops/logsoftmax.h"
3942
#include "infiniop/ops/lp_norm.h"
4043
#include "infiniop/ops/mul.h"
44+
#include "infiniop/ops/multi_margin_loss.h"
4145
#include "infiniop/ops/ones.h"
4246
#include "infiniop/ops/paged_attention.h"
4347
#include "infiniop/ops/paged_attention_prefill.h"
@@ -50,6 +54,7 @@
5054
#include "infiniop/ops/relu.h"
5155
#include "infiniop/ops/rms_norm.h"
5256
#include "infiniop/ops/rope.h"
57+
#include "infiniop/ops/scatter.h"
5358
#include "infiniop/ops/sigmoid.h"
5459
#include "infiniop/ops/silu.h"
5560
#include "infiniop/ops/silu_and_mul.h"
@@ -60,11 +65,6 @@
6065
#include "infiniop/ops/swiglu.h"
6166
#include "infiniop/ops/tanh.h"
6267
#include "infiniop/ops/topk.h"
63-
#include "infiniop/ops/floor_divide.h"
64-
#include "infiniop/ops/float_power.h"
65-
#include "infiniop/ops/flipud.h"
66-
#include "infiniop/ops/scatter.h"
67-
#include "infiniop/ops/multi_margin_loss.h"
6868
#include "infiniop/ops/topkrouter.h"
6969
#include "infiniop/ops/topksoftmax.h"
7070
#include "infiniop/ops/var.h"

‎include/infiniop/ops/flipud.h‎

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

66
typedef struct InfiniopDescriptor *infiniopFlipudDescriptor_t;
77

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

1313
// 获取工作空间大小
14-
__C __export infiniStatus_t infiniopGetFlipudWorkspaceSize(infiniopFlipudDescriptor_t desc, size_t *size);
14+
__INFINI_C __export infiniStatus_t infiniopGetFlipudWorkspaceSize(infiniopFlipudDescriptor_t desc, size_t *size);
1515

1616
// 执行 Flipud 算子
17-
__C __export infiniStatus_t infiniopFlipud(infiniopFlipudDescriptor_t desc,
18-
void *workspace,
19-
size_t workspace_size,
20-
void *output,
21-
const void *input,
22-
void *stream);
17+
__INFINI_C __export infiniStatus_t infiniopFlipud(infiniopFlipudDescriptor_t desc,
18+
void *workspace,
19+
size_t workspace_size,
20+
void *output,
21+
const void *input,
22+
void *stream);
2323

2424
// 销毁描述符
25-
__C __export infiniStatus_t infiniopDestroyFlipudDescriptor(infiniopFlipudDescriptor_t desc);
25+
__INFINI_C __export infiniStatus_t infiniopDestroyFlipudDescriptor(infiniopFlipudDescriptor_t desc);
2626

27-
#endif // __INFINIOP_FLIPUD_API_H__
27+
#endif // __INFINIOP_FLIPUD_API_H__

‎include/infiniop/ops/float_power.h‎

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

66
typedef struct InfiniopDescriptor *infiniopFloatPowerDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateFloatPowerDescriptor(infiniopHandle_t handle,
9-
infiniopFloatPowerDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t x,
12-
infiniopTensorDescriptor_t exponent,
13-
float scalar_exponent);
8+
__INFINI_C __export infiniStatus_t infiniopCreateFloatPowerDescriptor(infiniopHandle_t handle,
9+
infiniopFloatPowerDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x,
12+
infiniopTensorDescriptor_t exponent,
13+
float scalar_exponent);
1414

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

17-
__C __export infiniStatus_t infiniopFloatPower(infiniopFloatPowerDescriptor_t desc,
18-
void *workspace,
19-
size_t workspace_size,
20-
void *y,
21-
const void *x,
22-
const void *exponent,
23-
void *stream);
17+
__INFINI_C __export infiniStatus_t infiniopFloatPower(infiniopFloatPowerDescriptor_t desc,
18+
void *workspace,
19+
size_t workspace_size,
20+
void *y,
21+
const void *x,
22+
const void *exponent,
23+
void *stream);
2424

25-
__C __export infiniStatus_t infiniopDestroyFloatPowerDescriptor(infiniopFloatPowerDescriptor_t desc);
25+
__INFINI_C __export infiniStatus_t infiniopDestroyFloatPowerDescriptor(infiniopFloatPowerDescriptor_t desc);
2626

27-
#endif
27+
#endif

‎include/infiniop/ops/floor_divide.h‎

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

66
typedef struct InfiniopDescriptor *infiniopFloorDivideDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateFloorDivideDescriptor(infiniopHandle_t handle,
9-
infiniopFloorDivideDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t c,
11-
infiniopTensorDescriptor_t a,
12-
infiniopTensorDescriptor_t b);
8+
__INFINI_C __export infiniStatus_t infiniopCreateFloorDivideDescriptor(infiniopHandle_t handle,
9+
infiniopFloorDivideDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t c,
11+
infiniopTensorDescriptor_t a,
12+
infiniopTensorDescriptor_t b);
1313

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

16-
__C __export infiniStatus_t infiniopFloorDivide(infiniopFloorDivideDescriptor_t desc,
17-
void *workspace,
18-
size_t workspace_size,
19-
void *c,
20-
const void *a,
21-
const void *b,
22-
void *stream);
16+
__INFINI_C __export infiniStatus_t infiniopFloorDivide(infiniopFloorDivideDescriptor_t desc,
17+
void *workspace,
18+
size_t workspace_size,
19+
void *c,
20+
const void *a,
21+
const void *b,
22+
void *stream);
2323

24-
__C __export infiniStatus_t infiniopDestroyFloorDivideDescriptor(infiniopFloorDivideDescriptor_t desc);
24+
__INFINI_C __export infiniStatus_t infiniopDestroyFloorDivideDescriptor(infiniopFloorDivideDescriptor_t desc);
2525

26-
#endif
26+
#endif

‎include/infiniop/ops/multi_margin_loss.h‎

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

66
typedef struct InfiniopDescriptor *infiniopMultiMarginLossDescriptor_t;
7-
__C __export infiniStatus_t infiniopCreateMultiMarginLossDescriptor(infiniopHandle_t handle,
8-
infiniopMultiMarginLossDescriptor_t *desc_ptr,
9-
infiniopTensorDescriptor_t output,
10-
infiniopTensorDescriptor_t input,
11-
infiniopTensorDescriptor_t target,
12-
infiniopTensorDescriptor_t weight,
13-
int p,
14-
float margin,
15-
int reduction);
7+
__INFINI_C __export infiniStatus_t infiniopCreateMultiMarginLossDescriptor(infiniopHandle_t handle,
8+
infiniopMultiMarginLossDescriptor_t *desc_ptr,
9+
infiniopTensorDescriptor_t output,
10+
infiniopTensorDescriptor_t input,
11+
infiniopTensorDescriptor_t target,
12+
infiniopTensorDescriptor_t weight,
13+
int p,
14+
float margin,
15+
int reduction);
1616

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

19-
__C __export infiniStatus_t infiniopMultiMarginLoss(infiniopMultiMarginLossDescriptor_t desc,
20-
void *workspace,
21-
size_t workspace_size,
22-
void *output,
23-
const void *input,
24-
const void *target,
25-
const void *weight,
26-
void *stream);
19+
__INFINI_C __export infiniStatus_t infiniopMultiMarginLoss(infiniopMultiMarginLossDescriptor_t desc,
20+
void *workspace,
21+
size_t workspace_size,
22+
void *output,
23+
const void *input,
24+
const void *target,
25+
const void *weight,
26+
void *stream);
2727

28-
__C __export infiniStatus_t infiniopDestroyMultiMarginLossDescriptor(infiniopMultiMarginLossDescriptor_t desc);
28+
__INFINI_C __export infiniStatus_t infiniopDestroyMultiMarginLossDescriptor(infiniopMultiMarginLossDescriptor_t desc);
2929

30-
#endif // __INFINIOP_MULTI_MARGIN_LOSS_API_H__
30+
#endif // __INFINIOP_MULTI_MARGIN_LOSS_API_H__

0 commit comments

Comments
 (0)