Skip to content

Commit 5f638ea

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-48
1 parent fe74661 commit 5f638ea

103 files changed

Lines changed: 2112 additions & 2090 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/kthvalue.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ class Kthvalue {
1010
public:
1111
// Schema signature: values(out), indices(out), input, k, dim, keepdim
1212
using schema = void (*)(Tensor, Tensor, Tensor, int64_t, int64_t, bool);
13-
13+
1414
static void execute(Tensor values, Tensor indices, Tensor input, int64_t k, int64_t dim, bool keepdim);
1515
static common::OpDispatcher<schema> &dispatcher();
1616
};
@@ -21,4 +21,4 @@ std::tuple<Tensor, Tensor> kthvalue(Tensor input, int64_t k, int64_t dim = -1, b
2121
// In-place/Output-provided API
2222
void kthvalue_(Tensor values, Tensor indices, Tensor input, int64_t k, int64_t dim, bool keepdim);
2323

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

‎include/infinicore/ops/ldexp.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ class Ldexp {
99
public:
1010
// Schema signature: output(out), input(x), other(exp)
1111
using schema = void (*)(Tensor, Tensor, Tensor);
12-
12+
1313
static void execute(Tensor output, Tensor input, Tensor other);
1414
static common::OpDispatcher<schema> &dispatcher();
1515
};
@@ -21,4 +21,4 @@ Tensor ldexp(Tensor input, Tensor other);
2121
// Writes the result into 'output'
2222
void ldexp_(Tensor output, Tensor input, Tensor other);
2323

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

‎include/infinicore/ops/lerp.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,4 +24,4 @@ Tensor lerp(Tensor start, Tensor end, float weight);
2424
void lerp_(Tensor output, Tensor start, Tensor end, Tensor weight);
2525
void lerp_(Tensor output, Tensor start, Tensor end, float weight);
2626

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

‎include/infinicore/ops/triplet_margin_loss.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ class TripletMarginLoss {
99
public:
1010
// Schema signature: output, anchor, positive, negative, margin, p, eps, swap, reduction
1111
using schema = void (*)(Tensor, Tensor, Tensor, Tensor, float, int64_t, float, bool, int64_t);
12-
12+
1313
static void execute(Tensor output, Tensor anchor, Tensor positive, Tensor negative, float margin, int64_t p, float eps, bool swap, int64_t reduction);
1414
static common::OpDispatcher<schema> &dispatcher();
1515
};
@@ -21,4 +21,4 @@ Tensor triplet_margin_loss(Tensor anchor, Tensor positive, Tensor negative, floa
2121
// In-place / Explicit Output API
2222
void triplet_margin_loss_(Tensor output, Tensor anchor, Tensor positive, Tensor negative, float margin, int64_t p, float eps, bool swap, int64_t reduction);
2323

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

‎include/infinicore/ops/upsample_bilinear.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ class UpsampleBilinear {
1010
public:
1111
// Schema signature: output, input, align_corners
1212
using schema = void (*)(Tensor, Tensor, bool);
13-
13+
1414
static void execute(Tensor output, Tensor input, bool align_corners);
1515
static common::OpDispatcher<schema> &dispatcher();
1616
};
@@ -19,4 +19,4 @@ class UpsampleBilinear {
1919
Tensor upsample_bilinear(Tensor input, std::vector<int64_t> output_size, bool align_corners = false);
2020
void upsample_bilinear_(Tensor output, Tensor input, bool align_corners);
2121

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

‎include/infiniop.h‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,11 @@
4545
#include "infiniop/ops/index_copy.h"
4646
#include "infiniop/ops/inner.h"
4747
#include "infiniop/ops/int8_gemm.h"
48+
#include "infiniop/ops/kthvalue.h"
4849
#include "infiniop/ops/kv_caching.h"
4950
#include "infiniop/ops/layer_norm.h"
51+
#include "infiniop/ops/ldexp.h"
52+
#include "infiniop/ops/lerp.h"
5053
#include "infiniop/ops/logsoftmax.h"
5154
#include "infiniop/ops/lp_norm.h"
5255
#include "infiniop/ops/masked_select.h"
@@ -81,13 +84,10 @@
8184
#include "infiniop/ops/topk.h"
8285
#include "infiniop/ops/topkrouter.h"
8386
#include "infiniop/ops/topksoftmax.h"
84-
#include "infiniop/ops/var.h"
85-
#include "infiniop/ops/var_mean.h"
8687
#include "infiniop/ops/triplet_margin_loss.h"
8788
#include "infiniop/ops/upsample_bilinear.h"
88-
#include "infiniop/ops/kthvalue.h"
89-
#include "infiniop/ops/lerp.h"
90-
#include "infiniop/ops/ldexp.h"
89+
#include "infiniop/ops/var.h"
90+
#include "infiniop/ops/var_mean.h"
9191
#include "infiniop/ops/zeros.h"
9292
#include "infiniop/tensor_descriptor.h"
9393
#endif // __INFINIOP_API_H__

‎include/infiniop/ops/kthvalue.h‎

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

66
typedef struct InfiniopDescriptor *infiniopKthvalueDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateKthvalueDescriptor(infiniopHandle_t handle,
9-
infiniopKthvalueDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t values,
11-
infiniopTensorDescriptor_t indices,
12-
infiniopTensorDescriptor_t input,
13-
int k,
14-
int dim,
15-
int keepdim);
8+
__INFINI_C __export infiniStatus_t infiniopCreateKthvalueDescriptor(infiniopHandle_t handle,
9+
infiniopKthvalueDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t values,
11+
infiniopTensorDescriptor_t indices,
12+
infiniopTensorDescriptor_t input,
13+
int k,
14+
int dim,
15+
int keepdim);
1616

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

19-
__C __export infiniStatus_t infiniopKthvalue(infiniopKthvalueDescriptor_t desc,
20-
void *workspace,
21-
size_t workspace_size,
22-
void *values,
23-
void *indices,
24-
const void *input,
25-
void *stream);
19+
__INFINI_C __export infiniStatus_t infiniopKthvalue(infiniopKthvalueDescriptor_t desc,
20+
void *workspace,
21+
size_t workspace_size,
22+
void *values,
23+
void *indices,
24+
const void *input,
25+
void *stream);
2626

27-
__C __export infiniStatus_t infiniopDestroyKthvalueDescriptor(infiniopKthvalueDescriptor_t desc);
27+
__INFINI_C __export infiniStatus_t infiniopDestroyKthvalueDescriptor(infiniopKthvalueDescriptor_t desc);
2828

29-
#endif // __INFINIOP_KTHVALUE_API_H__
29+
#endif // __INFINIOP_KTHVALUE_API_H__

‎include/infiniop/ops/ldexp.h‎

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

66
typedef struct InfiniopDescriptor *infiniopLdexpDescriptor_t;
7-
__C __export infiniStatus_t infiniopCreateLdexpDescriptor(infiniopHandle_t handle,
8-
infiniopLdexpDescriptor_t *desc_ptr,
9-
infiniopTensorDescriptor_t y,
10-
infiniopTensorDescriptor_t x,
11-
infiniopTensorDescriptor_t exp);
12-
__C __export infiniStatus_t infiniopGetLdexpWorkspaceSize(infiniopLdexpDescriptor_t desc, size_t *size);
7+
__INFINI_C __export infiniStatus_t infiniopCreateLdexpDescriptor(infiniopHandle_t handle,
8+
infiniopLdexpDescriptor_t *desc_ptr,
9+
infiniopTensorDescriptor_t y,
10+
infiniopTensorDescriptor_t x,
11+
infiniopTensorDescriptor_t exp);
12+
__INFINI_C __export infiniStatus_t infiniopGetLdexpWorkspaceSize(infiniopLdexpDescriptor_t desc, size_t *size);
1313

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

22-
__C __export infiniStatus_t infiniopDestroyLdexpDescriptor(infiniopLdexpDescriptor_t desc);
22+
__INFINI_C __export infiniStatus_t infiniopDestroyLdexpDescriptor(infiniopLdexpDescriptor_t desc);
2323

24-
#endif // __INFINIOP_LDEXP_API_H__
24+
#endif // __INFINIOP_LDEXP_API_H__

‎include/infiniop/ops/lerp.h‎

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

66
typedef struct InfiniopDescriptor *infiniopLerpDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateLerpDescriptor(infiniopHandle_t handle,
9-
infiniopLerpDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t output,
11-
infiniopTensorDescriptor_t start,
12-
infiniopTensorDescriptor_t end,
13-
infiniopTensorDescriptor_t weight,
14-
float weight_scalar);
8+
__INFINI_C __export infiniStatus_t infiniopCreateLerpDescriptor(infiniopHandle_t handle,
9+
infiniopLerpDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t output,
11+
infiniopTensorDescriptor_t start,
12+
infiniopTensorDescriptor_t end,
13+
infiniopTensorDescriptor_t weight,
14+
float weight_scalar);
1515

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

18-
__C __export infiniStatus_t infiniopLerp(infiniopLerpDescriptor_t desc,
19-
void *workspace,
20-
size_t workspace_size,
21-
void *output,
22-
const void *start,
23-
const void *end,
24-
const void *weight,
25-
void *stream);
18+
__INFINI_C __export infiniStatus_t infiniopLerp(infiniopLerpDescriptor_t desc,
19+
void *workspace,
20+
size_t workspace_size,
21+
void *output,
22+
const void *start,
23+
const void *end,
24+
const void *weight,
25+
void *stream);
2626

27-
__C __export infiniStatus_t infiniopDestroyLerpDescriptor(infiniopLerpDescriptor_t desc);
27+
__INFINI_C __export infiniStatus_t infiniopDestroyLerpDescriptor(infiniopLerpDescriptor_t desc);
2828

29-
#endif // __INFINIOP_LERP_API_H__
29+
#endif // __INFINIOP_LERP_API_H__

‎include/infiniop/ops/triplet_margin_loss.h‎

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

66
typedef struct InfiniopDescriptor *infiniopTripletMarginLossDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateTripletMarginLossDescriptor(infiniopHandle_t handle,
9-
infiniopTripletMarginLossDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t output,
11-
infiniopTensorDescriptor_t anchor,
12-
infiniopTensorDescriptor_t positive,
13-
infiniopTensorDescriptor_t negative,
14-
float margin,
15-
int p,
16-
float eps,
17-
int swap, // 0: False, 1: True
18-
int reduction); // 0: None, 1: Mean, 2: Sum
8+
__INFINI_C __export infiniStatus_t infiniopCreateTripletMarginLossDescriptor(infiniopHandle_t handle,
9+
infiniopTripletMarginLossDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t output,
11+
infiniopTensorDescriptor_t anchor,
12+
infiniopTensorDescriptor_t positive,
13+
infiniopTensorDescriptor_t negative,
14+
float margin,
15+
int p,
16+
float eps,
17+
int swap, // 0: False, 1: True
18+
int reduction); // 0: None, 1: Mean, 2: Sum
1919

20-
__C __export infiniStatus_t infiniopGetTripletMarginLossWorkspaceSize(infiniopTripletMarginLossDescriptor_t desc, size_t *size);
20+
__INFINI_C __export infiniStatus_t infiniopGetTripletMarginLossWorkspaceSize(infiniopTripletMarginLossDescriptor_t desc, size_t *size);
2121

22-
__C __export infiniStatus_t infiniopTripletMarginLoss(infiniopTripletMarginLossDescriptor_t desc,
23-
void *workspace,
24-
size_t workspace_size,
25-
void *output,
26-
const void *anchor,
27-
const void *positive,
28-
const void *negative,
29-
void *stream);
22+
__INFINI_C __export infiniStatus_t infiniopTripletMarginLoss(infiniopTripletMarginLossDescriptor_t desc,
23+
void *workspace,
24+
size_t workspace_size,
25+
void *output,
26+
const void *anchor,
27+
const void *positive,
28+
const void *negative,
29+
void *stream);
3030

31-
__C __export infiniStatus_t infiniopDestroyTripletMarginLossDescriptor(infiniopTripletMarginLossDescriptor_t desc);
31+
__INFINI_C __export infiniStatus_t infiniopDestroyTripletMarginLossDescriptor(infiniopTripletMarginLossDescriptor_t desc);
3232

33-
#endif // __INFINIOP_TRIPLET_MARGIN_LOSS_API_H__
33+
#endif // __INFINIOP_TRIPLET_MARGIN_LOSS_API_H__

0 commit comments

Comments
 (0)