Skip to content

Commit 2904ec0

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-9 format
1 parent 6a0503f commit 2904ec0

File tree

8 files changed

+52
-54
lines changed

8 files changed

+52
-54
lines changed

‎include/infiniop/ops/adaptive_max_pool1d.h‎

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

66
typedef struct InfiniopDescriptor *infiniopAdaptiveMaxPool1dDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateAdaptiveMaxPool1dDescriptor(
8+
__INFINI_C __export infiniStatus_t infiniopCreateAdaptiveMaxPool1dDescriptor(
99
infiniopHandle_t handle,
1010
infiniopAdaptiveMaxPool1dDescriptor_t *desc,
1111
infiniopTensorDescriptor_t y_desc,
1212
infiniopTensorDescriptor_t x_desc,
1313
size_t output_size);
1414

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

17-
__C __export infiniStatus_t infiniopAdaptiveMaxPool1d(infiniopAdaptiveMaxPool1dDescriptor_t desc, void *workspace, size_t workspace_size,
18-
void *y, const void *x, void *stream);
17+
__INFINI_C __export infiniStatus_t infiniopAdaptiveMaxPool1d(infiniopAdaptiveMaxPool1dDescriptor_t desc, void *workspace, size_t workspace_size,
18+
void *y, const void *x, void *stream);
1919

20-
__C __export infiniStatus_t infiniopDestroyAdaptiveMaxPool1dDescriptor(infiniopAdaptiveMaxPool1dDescriptor_t desc);
20+
__INFINI_C __export infiniStatus_t infiniopDestroyAdaptiveMaxPool1dDescriptor(infiniopAdaptiveMaxPool1dDescriptor_t desc);
2121

2222
#endif

‎include/infiniop/ops/asinh.h‎

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

66
typedef struct InfiniopDescriptor *infiniopAsinhDescriptor_t;
77

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

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

15-
__C __export infiniStatus_t infiniopAsinh(infiniopAsinhDescriptor_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 infiniopAsinh(infiniopAsinhDescriptor_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 infiniopDestroyAsinhDescriptor(infiniopAsinhDescriptor_t desc);
22+
__INFINI_C __export infiniStatus_t infiniopDestroyAsinhDescriptor(infiniopAsinhDescriptor_t desc);
2323

2424
#endif

‎include/infiniop/ops/fmod.h‎

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

66
typedef struct InfiniopDescriptor *infiniopFmodDescriptor_t;
77

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

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

16-
__C __export infiniStatus_t infiniopFmod(infiniopFmodDescriptor_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 infiniopFmod(infiniopFmodDescriptor_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 infiniopDestroyFmodDescriptor(infiniopFmodDescriptor_t desc);
24+
__INFINI_C __export infiniStatus_t infiniopDestroyFmodDescriptor(infiniopFmodDescriptor_t desc);
2525

2626
#endif

‎python/infinicore/__init__.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,12 @@
4949
)
5050
from infinicore.ops.add import add
5151
from infinicore.ops.add_rms_norm import add_rms_norm
52+
from infinicore.ops.asinh import asinh
5253
from infinicore.ops.attention import attention
54+
from infinicore.ops.baddbmm import baddbmm
55+
from infinicore.ops.bilinear import bilinear
56+
from infinicore.ops.fmod import fmod
5357
from infinicore.ops.kv_caching import kv_caching
54-
from infinicore.ops.asinh import asinh
5558
from infinicore.ops.matmul import matmul
5659
from infinicore.ops.mha_varlen import mha_varlen
5760
from infinicore.ops.mul import mul
@@ -62,9 +65,6 @@
6265
from infinicore.ops.rearrange import rearrange
6366
from infinicore.ops.squeeze import squeeze
6467
from infinicore.ops.unsqueeze import unsqueeze
65-
from infinicore.ops.baddbmm import baddbmm
66-
from infinicore.ops.bilinear import bilinear
67-
from infinicore.ops.fmod import fmod
6868
from infinicore.tensor import (
6969
Tensor,
7070
empty,

‎python/infinicore/nn/functional/adaptive_max_pool1d.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
from typing import List
2-
31
from infinicore.lib import _infinicore
42
from infinicore.tensor import Tensor
53

‎src/infiniop/ops/adaptive_max_pool1d/operator.cc‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
#include "moore/adaptive_max_pool1d_moore.h"
1616
#endif
1717

18-
__C infiniStatus_t infiniopCreateAdaptiveMaxPool1dDescriptor(
18+
__INFINI_C infiniStatus_t infiniopCreateAdaptiveMaxPool1dDescriptor(
1919
infiniopHandle_t handle,
2020
infiniopAdaptiveMaxPool1dDescriptor_t *desc_ptr,
2121
infiniopTensorDescriptor_t y_desc,
@@ -53,7 +53,7 @@ __C infiniStatus_t infiniopCreateAdaptiveMaxPool1dDescriptor(
5353
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
5454
}
5555

56-
__C infiniStatus_t infiniopGetAdaptiveMaxPool1dWorkspaceSize(
56+
__INFINI_C infiniStatus_t infiniopGetAdaptiveMaxPool1dWorkspaceSize(
5757
infiniopAdaptiveMaxPool1dDescriptor_t desc,
5858
size_t *size) {
5959
#define GET(CASE, NAMESPACE) \
@@ -83,7 +83,7 @@ __C infiniStatus_t infiniopGetAdaptiveMaxPool1dWorkspaceSize(
8383
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
8484
}
8585

86-
__C infiniStatus_t infiniopAdaptiveMaxPool1d(
86+
__INFINI_C infiniStatus_t infiniopAdaptiveMaxPool1d(
8787
infiniopAdaptiveMaxPool1dDescriptor_t desc,
8888
void *workspace,
8989
size_t workspace_size,
@@ -117,7 +117,7 @@ __C infiniStatus_t infiniopAdaptiveMaxPool1d(
117117
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
118118
}
119119

120-
__C infiniStatus_t infiniopDestroyAdaptiveMaxPool1dDescriptor(
120+
__INFINI_C infiniStatus_t infiniopDestroyAdaptiveMaxPool1dDescriptor(
121121
infiniopAdaptiveMaxPool1dDescriptor_t desc) {
122122
#define DESTROY(CASE, NAMESPACE) \
123123
case CASE: \

‎src/infiniop/ops/asinh/operator.cc‎

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
#include "moore/asinh_moore.h"
1616
#endif
1717

18-
__C infiniStatus_t infiniopCreateAsinhDescriptor(
18+
__INFINI_C infiniStatus_t infiniopCreateAsinhDescriptor(
1919
infiniopHandle_t handle,
2020
infiniopAsinhDescriptor_t *desc_ptr,
2121
infiniopTensorDescriptor_t y_desc,
@@ -50,7 +50,7 @@ __C infiniStatus_t infiniopCreateAsinhDescriptor(
5050
#undef CREATE
5151
}
5252

53-
__C infiniStatus_t infiniopGetAsinhWorkspaceSize(infiniopAsinhDescriptor_t desc, size_t *size) {
53+
__INFINI_C infiniStatus_t infiniopGetAsinhWorkspaceSize(infiniopAsinhDescriptor_t desc, size_t *size) {
5454

5555
#define GET(CASE, NAMESPACE) \
5656
case CASE: \
@@ -80,12 +80,12 @@ __C infiniStatus_t infiniopGetAsinhWorkspaceSize(infiniopAsinhDescriptor_t desc,
8080
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
8181
}
8282

83-
__C infiniStatus_t infiniopAsinh(infiniopAsinhDescriptor_t desc,
84-
void *workspace,
85-
size_t workspace_size,
86-
void *y,
87-
const void *x,
88-
void *stream) {
83+
__INFINI_C infiniStatus_t infiniopAsinh(infiniopAsinhDescriptor_t desc,
84+
void *workspace,
85+
size_t workspace_size,
86+
void *y,
87+
const void *x,
88+
void *stream) {
8989
#define CALCULATE(CASE, NAMESPACE) \
9090
case CASE: \
9191
return reinterpret_cast<const op::asinh::NAMESPACE::Descriptor *>(desc) \
@@ -112,7 +112,7 @@ __C infiniStatus_t infiniopAsinh(infiniopAsinhDescriptor_t desc,
112112
#undef CALCULATE
113113
}
114114

115-
__C infiniStatus_t infiniopDestroyAsinhDescriptor(infiniopAsinhDescriptor_t desc) {
115+
__INFINI_C infiniStatus_t infiniopDestroyAsinhDescriptor(infiniopAsinhDescriptor_t desc) {
116116
#define DESTROY(CASE, NAMESPACE) \
117117
case CASE: \
118118
delete reinterpret_cast<op::asinh::NAMESPACE::Descriptor *>(desc); \

‎src/infiniop/ops/fmod/operator.cc‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
#include "moore/fmod_moore.h"
1616
#endif
1717

18-
__C infiniStatus_t infiniopCreateFmodDescriptor(
18+
__INFINI_C infiniStatus_t infiniopCreateFmodDescriptor(
1919
infiniopHandle_t handle,
2020
infiniopFmodDescriptor_t *desc_ptr,
2121
infiniopTensorDescriptor_t c_desc,
@@ -54,7 +54,7 @@ __C infiniStatus_t infiniopCreateFmodDescriptor(
5454
#undef CREATE
5555
}
5656

57-
__C infiniStatus_t infiniopGetFmodWorkspaceSize(infiniopFmodDescriptor_t desc, size_t *size) {
57+
__INFINI_C infiniStatus_t infiniopGetFmodWorkspaceSize(infiniopFmodDescriptor_t desc, size_t *size) {
5858

5959
#define GET(CASE, NAMESPACE) \
6060
case CASE: \
@@ -84,7 +84,7 @@ __C infiniStatus_t infiniopGetFmodWorkspaceSize(infiniopFmodDescriptor_t desc, s
8484
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
8585
}
8686

87-
__C infiniStatus_t infiniopFmod(
87+
__INFINI_C infiniStatus_t infiniopFmod(
8888
infiniopFmodDescriptor_t desc,
8989
void *workspace,
9090
size_t workspace_size,
@@ -119,7 +119,7 @@ __C infiniStatus_t infiniopFmod(
119119
#undef CALCULATE
120120
}
121121

122-
__C infiniStatus_t infiniopDestroyFmodDescriptor(infiniopFmodDescriptor_t desc) {
122+
__INFINI_C infiniStatus_t infiniopDestroyFmodDescriptor(infiniopFmodDescriptor_t desc) {
123123

124124
#define GET(CASE, NAMESPACE) \
125125
case CASE: \

0 commit comments

Comments
 (0)