Skip to content

Commit 4d70bbf

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-30
1 parent fe748cc commit 4d70bbf

64 files changed

Lines changed: 1236 additions & 994 deletions

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/gaussian_nll_loss.hpp‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -11,18 +11,18 @@ namespace infinicore::op {
1111
INFINICORE_GRAPH_OP_CLASS(GaussianNllLoss, Tensor, const Tensor &, const Tensor &, const Tensor &, bool, double, int);
1212

1313
__export Tensor gaussian_nll_loss(const Tensor &input,
14+
const Tensor &target,
15+
const Tensor &var,
16+
bool full = false,
17+
double eps = 1e-6,
18+
int reduction = 1);
19+
20+
__export void gaussian_nll_loss_(Tensor out,
21+
const Tensor &input,
1422
const Tensor &target,
1523
const Tensor &var,
1624
bool full = false,
1725
double eps = 1e-6,
1826
int reduction = 1);
1927

20-
__export void gaussian_nll_loss_(Tensor out,
21-
const Tensor &input,
22-
const Tensor &target,
23-
const Tensor &var,
24-
bool full = false,
25-
double eps = 1e-6,
26-
int reduction = 1);
27-
2828
} // namespace infinicore::op

‎include/infinicore/ops/interpolate.hpp‎

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,16 +15,16 @@ namespace infinicore::op {
1515
INFINICORE_GRAPH_OP_CLASS(Interpolate, Tensor, const Tensor &, std::string, std::vector<int64_t>, std::vector<double>, int);
1616

1717
__export Tensor interpolate(const Tensor &input,
18+
std::string mode,
19+
std::vector<int64_t> size,
20+
std::vector<double> scale_factor,
21+
int align_corners);
22+
23+
__export void interpolate_(Tensor out,
24+
const Tensor &input,
1825
std::string mode,
1926
std::vector<int64_t> size,
2027
std::vector<double> scale_factor,
2128
int align_corners);
2229

23-
__export void interpolate_(Tensor out,
24-
const Tensor &input,
25-
std::string mode,
26-
std::vector<int64_t> size,
27-
std::vector<double> scale_factor,
28-
int align_corners);
29-
3030
} // namespace infinicore::op

‎include/infiniop/ops/bitwise_right_shift.h‎

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

66
typedef struct InfiniopDescriptor *infiniopBitwiseRightShiftDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateBitwiseRightShiftDescriptor(infiniopHandle_t handle,
9-
infiniopBitwiseRightShiftDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t x1,
12-
infiniopTensorDescriptor_t x2);
8+
__INFINI_C __export infiniStatus_t infiniopCreateBitwiseRightShiftDescriptor(infiniopHandle_t handle,
9+
infiniopBitwiseRightShiftDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x1,
12+
infiniopTensorDescriptor_t x2);
1313

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

16-
__C __export infiniStatus_t infiniopBitwiseRightShift(infiniopBitwiseRightShiftDescriptor_t desc,
17-
void *workspace,
18-
size_t workspace_size,
19-
void *y,
20-
const void *x1,
21-
const void *x2,
22-
void *stream);
16+
__INFINI_C __export infiniStatus_t infiniopBitwiseRightShift(infiniopBitwiseRightShiftDescriptor_t desc,
17+
void *workspace,
18+
size_t workspace_size,
19+
void *y,
20+
const void *x1,
21+
const void *x2,
22+
void *stream);
2323

24-
__C __export infiniStatus_t infiniopDestroyBitwiseRightShiftDescriptor(infiniopBitwiseRightShiftDescriptor_t desc);
24+
__INFINI_C __export infiniStatus_t infiniopDestroyBitwiseRightShiftDescriptor(infiniopBitwiseRightShiftDescriptor_t desc);
2525

2626
#endif

‎include/infiniop/ops/gaussian_nll_loss.h‎

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

66
typedef struct InfiniopDescriptor *infiniopGaussianNllLossDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateGaussianNllLossDescriptor(infiniopHandle_t handle,
9-
infiniopGaussianNllLossDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t input,
12-
infiniopTensorDescriptor_t target,
13-
infiniopTensorDescriptor_t var,
14-
int full,
15-
double eps,
16-
int reduction);
8+
__INFINI_C __export infiniStatus_t infiniopCreateGaussianNllLossDescriptor(infiniopHandle_t handle,
9+
infiniopGaussianNllLossDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t input,
12+
infiniopTensorDescriptor_t target,
13+
infiniopTensorDescriptor_t var,
14+
int full,
15+
double eps,
16+
int reduction);
1717

18-
__C __export infiniStatus_t infiniopGetGaussianNllLossWorkspaceSize(infiniopGaussianNllLossDescriptor_t desc, size_t *size);
18+
__INFINI_C __export infiniStatus_t infiniopGetGaussianNllLossWorkspaceSize(infiniopGaussianNllLossDescriptor_t desc, size_t *size);
1919

20-
__C __export infiniStatus_t infiniopGaussianNllLoss(infiniopGaussianNllLossDescriptor_t desc,
21-
void *workspace,
22-
size_t workspace_size,
23-
void *y,
24-
const void *input,
25-
const void *target,
26-
const void *var,
27-
void *stream);
20+
__INFINI_C __export infiniStatus_t infiniopGaussianNllLoss(infiniopGaussianNllLossDescriptor_t desc,
21+
void *workspace,
22+
size_t workspace_size,
23+
void *y,
24+
const void *input,
25+
const void *target,
26+
const void *var,
27+
void *stream);
2828

29-
__C __export infiniStatus_t infiniopDestroyGaussianNllLossDescriptor(infiniopGaussianNllLossDescriptor_t desc);
29+
__INFINI_C __export infiniStatus_t infiniopDestroyGaussianNllLossDescriptor(infiniopGaussianNllLossDescriptor_t desc);
3030

3131
#endif

‎include/infiniop/ops/interpolate.h‎

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

66
typedef struct InfiniopDescriptor *infiniopInterpolateDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateInterpolateDescriptor(infiniopHandle_t handle,
9-
infiniopInterpolateDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t x,
12-
const char *mode,
13-
void *size,
14-
void *scale_factor,
15-
int align_corners);
8+
__INFINI_C __export infiniStatus_t infiniopCreateInterpolateDescriptor(infiniopHandle_t handle,
9+
infiniopInterpolateDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x,
12+
const char *mode,
13+
void *size,
14+
void *scale_factor,
15+
int align_corners);
1616

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

19-
__C __export infiniStatus_t infiniopInterpolate(infiniopInterpolateDescriptor_t desc,
20-
void *workspace,
21-
size_t workspace_size,
22-
void *y,
23-
const void *x,
24-
void *stream);
19+
__INFINI_C __export infiniStatus_t infiniopInterpolate(infiniopInterpolateDescriptor_t desc,
20+
void *workspace,
21+
size_t workspace_size,
22+
void *y,
23+
const void *x,
24+
void *stream);
2525

26-
__C __export infiniStatus_t infiniopDestroyInterpolateDescriptor(infiniopInterpolateDescriptor_t desc);
26+
__INFINI_C __export infiniStatus_t infiniopDestroyInterpolateDescriptor(infiniopInterpolateDescriptor_t desc);
2727

2828
#endif

‎include/infiniop/ops/prelu.h‎

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

66
typedef struct InfiniopDescriptor *infiniopPreluDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreatePreluDescriptor(infiniopHandle_t handle,
9-
infiniopPreluDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t x,
12-
infiniopTensorDescriptor_t weight);
8+
__INFINI_C __export infiniStatus_t infiniopCreatePreluDescriptor(infiniopHandle_t handle,
9+
infiniopPreluDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x,
12+
infiniopTensorDescriptor_t weight);
1313

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

16-
__C __export infiniStatus_t infiniopPrelu(infiniopPreluDescriptor_t desc,
17-
void *workspace,
18-
size_t workspace_size,
19-
void *y,
20-
const void *x,
21-
const void *weight,
22-
void *stream);
16+
__INFINI_C __export infiniStatus_t infiniopPrelu(infiniopPreluDescriptor_t desc,
17+
void *workspace,
18+
size_t workspace_size,
19+
void *y,
20+
const void *x,
21+
const void *weight,
22+
void *stream);
2323

24-
__C __export infiniStatus_t infiniopDestroyPreluDescriptor(infiniopPreluDescriptor_t desc);
24+
__INFINI_C __export infiniStatus_t infiniopDestroyPreluDescriptor(infiniopPreluDescriptor_t desc);
2525

2626
#endif

‎include/infiniop/ops/relu6.h‎

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

66
typedef struct InfiniopDescriptor *infiniopRelu6Descriptor_t;
77

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

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

15-
__C __export infiniStatus_t infiniopRelu6(infiniopRelu6Descriptor_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 infiniopRelu6(infiniopRelu6Descriptor_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 infiniopDestroyRelu6Descriptor(infiniopRelu6Descriptor_t desc);
22+
__INFINI_C __export infiniStatus_t infiniopDestroyRelu6Descriptor(infiniopRelu6Descriptor_t desc);
2323

2424
#endif

‎python/infinicore/__init__.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,7 @@
6464
from infinicore.ops.binary_cross_entropy_with_logits import (
6565
binary_cross_entropy_with_logits,
6666
)
67+
from infinicore.ops.bitwise_right_shift import bitwise_right_shift
6768
from infinicore.ops.broadcast_to import broadcast_to
6869
from infinicore.ops.cat import cat
6970
from infinicore.ops.cdist import cdist
@@ -80,7 +81,6 @@
8081
from infinicore.ops.index_copy import index_copy
8182
from infinicore.ops.inner import inner
8283
from infinicore.ops.kthvalue import kthvalue
83-
from infinicore.ops.bitwise_right_shift import bitwise_right_shift
8484
from infinicore.ops.kv_caching import kv_caching
8585
from infinicore.ops.ldexp import ldexp
8686
from infinicore.ops.lerp import lerp

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from .causal_softmax import causal_softmax
88
from .embedding import embedding
99
from .flash_attention import flash_attention
10+
from .gaussian_nll_loss import gaussian_nll_loss
1011
from .hardswish import hardswish
1112
from .hardtanh import hardtanh
1213
from .huber_loss import huber_loss
@@ -15,7 +16,6 @@
1516
from .linear_w8a8i8 import linear_w8a8i8
1617
from .log_softmax import log_softmax
1718
from .multi_margin_loss import multi_margin_loss
18-
from .gaussian_nll_loss import gaussian_nll_loss
1919
from .prelu import prelu
2020
from .random_sample import random_sample
2121
from .relu6 import relu6

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

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,8 @@
1-
from typing import Optional, Sequence, Union
2-
3-
from infinicore.tensor import Tensor
4-
5-
from .upsample_bilinear import upsample_bilinear
6-
from .upsample_nearest import upsample_nearest
71

82
from collections.abc import Iterable
93

104
from infinicore.lib import _infinicore
5+
from infinicore.tensor import Tensor
116

127

138
def _to_int64_list(value) -> list[int]:

0 commit comments

Comments
 (0)