Skip to content

Commit 774d8f8

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-49
1 parent 9bb2770 commit 774d8f8

97 files changed

Lines changed: 1420 additions & 1462 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/log_softmax.hpp‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ class LogSoftmax {
99
public:
1010
// Schema signature: output(out), input, dim
1111
using schema = void (*)(Tensor, Tensor, int64_t);
12-
12+
1313
static void execute(Tensor output, Tensor input, int64_t dim);
1414
static common::OpDispatcher<schema> &dispatcher();
1515
};
@@ -20,4 +20,4 @@ Tensor log_softmax(Tensor input, int64_t dim);
2020
// In-place/Output-provided API
2121
void log_softmax_(Tensor output, Tensor input, int64_t dim);
2222

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

‎include/infinicore/ops/logaddexp.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,4 @@ class LogAddExp {
1515
Tensor logaddexp(Tensor a, Tensor b);
1616
void logaddexp_(Tensor c, Tensor a, Tensor b);
1717

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

‎include/infinicore/ops/logaddexp2.hpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,4 +15,4 @@ class LogAddExp2 {
1515
Tensor logaddexp2(Tensor a, Tensor b);
1616
void logaddexp2_(Tensor c, Tensor a, Tensor b);
1717

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

‎include/infinicore/ops/triplet_margin_with_distance_loss.hpp‎

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

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

‎include/infinicore/ops/upsample_nearest.hpp‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,16 +11,16 @@ class UpsampleNearest {
1111
// Schema signature: output(out), input
1212
// Note: Scales are inferred from output.shape / input.shape
1313
using schema = void (*)(Tensor, Tensor);
14-
14+
1515
static void execute(Tensor output, Tensor input);
1616
static common::OpDispatcher<schema> &dispatcher();
1717
};
1818

1919
// Functional API: Returns the result tensor
2020
// Requires output_size to calculate the shape of the result tensor
21-
Tensor upsample_nearest(Tensor input, const std::vector<int64_t>& output_size);
21+
Tensor upsample_nearest(Tensor input, const std::vector<int64_t> &output_size);
2222

2323
// In-place/Output-provided API
2424
void upsample_nearest_(Tensor output, Tensor input);
2525

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

‎include/infiniop/ops/log_softmax.h‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,4 +22,4 @@ __INFINI_C __export infiniStatus_t infiniopLogSoftmax(infiniopLogSoftmaxDescript
2222

2323
__INFINI_C __export infiniStatus_t infiniopDestroyLogSoftmaxDescriptor(infiniopLogSoftmaxDescriptor_t desc);
2424

25-
#endif // __INFINIOP_LOG_SOFTMAX_API_H__
25+
#endif // __INFINIOP_LOG_SOFTMAX_API_H__

‎include/infiniop/ops/logaddexp.h‎

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

66
typedef struct InfiniopDescriptor *infiniopLogAddExpDescriptor_t;
77

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

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

16-
__C __export infiniStatus_t infiniopLogAddExp(infiniopLogAddExpDescriptor_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 infiniopLogAddExp(infiniopLogAddExpDescriptor_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 infiniopDestroyLogAddExpDescriptor(infiniopLogAddExpDescriptor_t desc);
24+
__INFINI_C __export infiniStatus_t infiniopDestroyLogAddExpDescriptor(infiniopLogAddExpDescriptor_t desc);
2525

26-
#endif // __INFINIOP_LOGADDEXP_API_H__
26+
#endif // __INFINIOP_LOGADDEXP_API_H__

‎include/infiniop/ops/logaddexp2.h‎

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

66
typedef struct InfiniopDescriptor *infiniopLogAddExp2Descriptor_t;
77

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

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

16-
__C __export infiniStatus_t infiniopLogAddExp2(infiniopLogAddExp2Descriptor_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 infiniopLogAddExp2(infiniopLogAddExp2Descriptor_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 infiniopDestroyLogAddExp2Descriptor(infiniopLogAddExp2Descriptor_t desc);
24+
__INFINI_C __export infiniStatus_t infiniopDestroyLogAddExp2Descriptor(infiniopLogAddExp2Descriptor_t desc);
2525

26-
#endif // __INFINIOP_LOGADDEXP2_API_H__
26+
#endif // __INFINIOP_LOGADDEXP2_API_H__

‎include/infiniop/ops/triplet_margin_with_distance_loss.h‎

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

66
typedef struct InfiniopDescriptor *infiniopTripletMarginWithDistanceLossDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateTripletMarginWithDistanceLossDescriptor(
9-
infiniopHandle_t handle,
10-
infiniopTripletMarginWithDistanceLossDescriptor_t *desc_ptr,
11-
infiniopTensorDescriptor_t output,
12-
infiniopTensorDescriptor_t anchor,
13-
infiniopTensorDescriptor_t positive,
14-
infiniopTensorDescriptor_t negative,
15-
float margin,
16-
int swap,
17-
int reduction);
18-
__C __export infiniStatus_t infiniopGetTripletMarginWithDistanceLossWorkspaceSize(
19-
infiniopTripletMarginWithDistanceLossDescriptor_t desc,
20-
size_t *size);
21-
__C __export infiniStatus_t infiniopTripletMarginWithDistanceLoss(infiniopTripletMarginWithDistanceLossDescriptor_t desc,
22-
void *workspace,
23-
size_t workspace_size,
24-
void *output,
25-
const void *anchor,
26-
const void *positive,
27-
const void *negative,
28-
void *stream);
8+
__INFINI_C __export infiniStatus_t infiniopCreateTripletMarginWithDistanceLossDescriptor(
9+
infiniopHandle_t handle,
10+
infiniopTripletMarginWithDistanceLossDescriptor_t *desc_ptr,
11+
infiniopTensorDescriptor_t output,
12+
infiniopTensorDescriptor_t anchor,
13+
infiniopTensorDescriptor_t positive,
14+
infiniopTensorDescriptor_t negative,
15+
float margin,
16+
int swap,
17+
int reduction);
18+
__INFINI_C __export infiniStatus_t infiniopGetTripletMarginWithDistanceLossWorkspaceSize(
19+
infiniopTripletMarginWithDistanceLossDescriptor_t desc,
20+
size_t *size);
21+
__INFINI_C __export infiniStatus_t infiniopTripletMarginWithDistanceLoss(infiniopTripletMarginWithDistanceLossDescriptor_t desc,
22+
void *workspace,
23+
size_t workspace_size,
24+
void *output,
25+
const void *anchor,
26+
const void *positive,
27+
const void *negative,
28+
void *stream);
2929

30-
__C __export infiniStatus_t infiniopDestroyTripletMarginWithDistanceLossDescriptor(
31-
infiniopTripletMarginWithDistanceLossDescriptor_t desc);
32-
#endif // __INFINIOP_TRIPLET_MARGIN_WITH_DISTANCE_LOSS_API_H__
30+
__INFINI_C __export infiniStatus_t infiniopDestroyTripletMarginWithDistanceLossDescriptor(
31+
infiniopTripletMarginWithDistanceLossDescriptor_t desc);
32+
#endif // __INFINIOP_TRIPLET_MARGIN_WITH_DISTANCE_LOSS_API_H__

‎include/infiniop/ops/upsample_nearest.h‎

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

66
typedef struct InfiniopDescriptor *infiniopUpsampleNearestDescriptor_t;
77

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

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

15-
__C __export infiniStatus_t infiniopUpsampleNearest(infiniopUpsampleNearestDescriptor_t desc,
16-
void *workspace,
17-
size_t workspace_size,
18-
void *output,
19-
const void *input,
20-
void *stream);
15+
__INFINI_C __export infiniStatus_t infiniopUpsampleNearest(infiniopUpsampleNearestDescriptor_t desc,
16+
void *workspace,
17+
size_t workspace_size,
18+
void *output,
19+
const void *input,
20+
void *stream);
2121

22-
__C __export infiniStatus_t infiniopDestroyUpsampleNearestDescriptor(infiniopUpsampleNearestDescriptor_t desc);
22+
__INFINI_C __export infiniStatus_t infiniopDestroyUpsampleNearestDescriptor(infiniopUpsampleNearestDescriptor_t desc);
2323

24-
#endif // __INFINIOP_UPSAMPLE_NEAREST_API_H__
24+
#endif // __INFINIOP_UPSAMPLE_NEAREST_API_H__

0 commit comments

Comments
 (0)