Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
jerrrrry
infinicore
Commits
2904ec00
Commit
2904ec00
authored
Mar 06, 2026
by
PanZezhong
Browse files
issue/1031 fix T1-1-9 format
parent
6a0503fc
Changes
8
Show whitespace changes
Inline
Side-by-side
Showing
8 changed files
with
52 additions
and
54 deletions
+52
-54
include/infiniop/ops/adaptive_max_pool1d.h
include/infiniop/ops/adaptive_max_pool1d.h
+5
-5
include/infiniop/ops/asinh.h
include/infiniop/ops/asinh.h
+12
-12
include/infiniop/ops/fmod.h
include/infiniop/ops/fmod.h
+14
-14
python/infinicore/__init__.py
python/infinicore/__init__.py
+4
-4
python/infinicore/nn/functional/adaptive_max_pool1d.py
python/infinicore/nn/functional/adaptive_max_pool1d.py
+0
-2
src/infiniop/ops/adaptive_max_pool1d/operator.cc
src/infiniop/ops/adaptive_max_pool1d/operator.cc
+4
-4
src/infiniop/ops/asinh/operator.cc
src/infiniop/ops/asinh/operator.cc
+9
-9
src/infiniop/ops/fmod/operator.cc
src/infiniop/ops/fmod/operator.cc
+4
-4
No files found.
include/infiniop/ops/adaptive_max_pool1d.h
View file @
2904ec00
...
@@ -5,18 +5,18 @@
...
@@ -5,18 +5,18 @@
typedef
struct
InfiniopDescriptor
*
infiniopAdaptiveMaxPool1dDescriptor_t
;
typedef
struct
InfiniopDescriptor
*
infiniopAdaptiveMaxPool1dDescriptor_t
;
__C
__export
infiniStatus_t
infiniopCreateAdaptiveMaxPool1dDescriptor
(
__
INFINI_
C
__export
infiniStatus_t
infiniopCreateAdaptiveMaxPool1dDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopAdaptiveMaxPool1dDescriptor_t
*
desc
,
infiniopAdaptiveMaxPool1dDescriptor_t
*
desc
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
x_desc
,
infiniopTensorDescriptor_t
x_desc
,
size_t
output_size
);
size_t
output_size
);
__C
__export
infiniStatus_t
infiniopGetAdaptiveMaxPool1dWorkspaceSize
(
infiniopAdaptiveMaxPool1dDescriptor_t
desc
,
size_t
*
size
);
__
INFINI_
C
__export
infiniStatus_t
infiniopGetAdaptiveMaxPool1dWorkspaceSize
(
infiniopAdaptiveMaxPool1dDescriptor_t
desc
,
size_t
*
size
);
__C
__export
infiniStatus_t
infiniopAdaptiveMaxPool1d
(
infiniopAdaptiveMaxPool1dDescriptor_t
desc
,
void
*
workspace
,
size_t
workspace_size
,
__
INFINI_
C
__export
infiniStatus_t
infiniopAdaptiveMaxPool1d
(
infiniopAdaptiveMaxPool1dDescriptor_t
desc
,
void
*
workspace
,
size_t
workspace_size
,
void
*
y
,
const
void
*
x
,
void
*
stream
);
void
*
y
,
const
void
*
x
,
void
*
stream
);
__C
__export
infiniStatus_t
infiniopDestroyAdaptiveMaxPool1dDescriptor
(
infiniopAdaptiveMaxPool1dDescriptor_t
desc
);
__
INFINI_
C
__export
infiniStatus_t
infiniopDestroyAdaptiveMaxPool1dDescriptor
(
infiniopAdaptiveMaxPool1dDescriptor_t
desc
);
#endif
#endif
\ No newline at end of file
include/infiniop/ops/asinh.h
View file @
2904ec00
...
@@ -5,20 +5,20 @@
...
@@ -5,20 +5,20 @@
typedef
struct
InfiniopDescriptor
*
infiniopAsinhDescriptor_t
;
typedef
struct
InfiniopDescriptor
*
infiniopAsinhDescriptor_t
;
__C
__export
infiniStatus_t
infiniopCreateAsinhDescriptor
(
infiniopHandle_t
handle
,
__
INFINI_
C
__export
infiniStatus_t
infiniopCreateAsinhDescriptor
(
infiniopHandle_t
handle
,
infiniopAsinhDescriptor_t
*
desc_ptr
,
infiniopAsinhDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y
,
infiniopTensorDescriptor_t
y
,
infiniopTensorDescriptor_t
x
);
infiniopTensorDescriptor_t
x
);
__C
__export
infiniStatus_t
infiniopGetAsinhWorkspaceSize
(
infiniopAsinhDescriptor_t
desc
,
size_t
*
size
);
__
INFINI_
C
__export
infiniStatus_t
infiniopGetAsinhWorkspaceSize
(
infiniopAsinhDescriptor_t
desc
,
size_t
*
size
);
__C
__export
infiniStatus_t
infiniopAsinh
(
infiniopAsinhDescriptor_t
desc
,
__
INFINI_
C
__export
infiniStatus_t
infiniopAsinh
(
infiniopAsinhDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
void
*
y
,
void
*
y
,
const
void
*
x
,
const
void
*
x
,
void
*
stream
);
void
*
stream
);
__C
__export
infiniStatus_t
infiniopDestroyAsinhDescriptor
(
infiniopAsinhDescriptor_t
desc
);
__
INFINI_
C
__export
infiniStatus_t
infiniopDestroyAsinhDescriptor
(
infiniopAsinhDescriptor_t
desc
);
#endif
#endif
\ No newline at end of file
include/infiniop/ops/fmod.h
View file @
2904ec00
...
@@ -5,15 +5,15 @@
...
@@ -5,15 +5,15 @@
typedef
struct
InfiniopDescriptor
*
infiniopFmodDescriptor_t
;
typedef
struct
InfiniopDescriptor
*
infiniopFmodDescriptor_t
;
__C
__export
infiniStatus_t
infiniopCreateFmodDescriptor
(
infiniopHandle_t
handle
,
__
INFINI_
C
__export
infiniStatus_t
infiniopCreateFmodDescriptor
(
infiniopHandle_t
handle
,
infiniopFmodDescriptor_t
*
desc_ptr
,
infiniopFmodDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
c
,
infiniopTensorDescriptor_t
c
,
infiniopTensorDescriptor_t
a
,
infiniopTensorDescriptor_t
a
,
infiniopTensorDescriptor_t
b
);
infiniopTensorDescriptor_t
b
);
__C
__export
infiniStatus_t
infiniopGetFmodWorkspaceSize
(
infiniopFmodDescriptor_t
desc
,
size_t
*
size
);
__
INFINI_
C
__export
infiniStatus_t
infiniopGetFmodWorkspaceSize
(
infiniopFmodDescriptor_t
desc
,
size_t
*
size
);
__C
__export
infiniStatus_t
infiniopFmod
(
infiniopFmodDescriptor_t
desc
,
__
INFINI_
C
__export
infiniStatus_t
infiniopFmod
(
infiniopFmodDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
void
*
c
,
void
*
c
,
...
@@ -21,6 +21,6 @@ __C __export infiniStatus_t infiniopFmod(infiniopFmodDescriptor_t desc,
...
@@ -21,6 +21,6 @@ __C __export infiniStatus_t infiniopFmod(infiniopFmodDescriptor_t desc,
const
void
*
b
,
const
void
*
b
,
void
*
stream
);
void
*
stream
);
__C
__export
infiniStatus_t
infiniopDestroyFmodDescriptor
(
infiniopFmodDescriptor_t
desc
);
__
INFINI_
C
__export
infiniStatus_t
infiniopDestroyFmodDescriptor
(
infiniopFmodDescriptor_t
desc
);
#endif
#endif
\ No newline at end of file
python/infinicore/__init__.py
View file @
2904ec00
...
@@ -49,9 +49,12 @@ from infinicore.dtype import (
...
@@ -49,9 +49,12 @@ from infinicore.dtype import (
)
)
from
infinicore.ops.add
import
add
from
infinicore.ops.add
import
add
from
infinicore.ops.add_rms_norm
import
add_rms_norm
from
infinicore.ops.add_rms_norm
import
add_rms_norm
from
infinicore.ops.asinh
import
asinh
from
infinicore.ops.attention
import
attention
from
infinicore.ops.attention
import
attention
from
infinicore.ops.baddbmm
import
baddbmm
from
infinicore.ops.bilinear
import
bilinear
from
infinicore.ops.fmod
import
fmod
from
infinicore.ops.kv_caching
import
kv_caching
from
infinicore.ops.kv_caching
import
kv_caching
from
infinicore.ops.asinh
import
asinh
from
infinicore.ops.matmul
import
matmul
from
infinicore.ops.matmul
import
matmul
from
infinicore.ops.mha_varlen
import
mha_varlen
from
infinicore.ops.mha_varlen
import
mha_varlen
from
infinicore.ops.mul
import
mul
from
infinicore.ops.mul
import
mul
...
@@ -62,9 +65,6 @@ from infinicore.ops.paged_caching import paged_caching
...
@@ -62,9 +65,6 @@ from infinicore.ops.paged_caching import paged_caching
from
infinicore.ops.rearrange
import
rearrange
from
infinicore.ops.rearrange
import
rearrange
from
infinicore.ops.squeeze
import
squeeze
from
infinicore.ops.squeeze
import
squeeze
from
infinicore.ops.unsqueeze
import
unsqueeze
from
infinicore.ops.unsqueeze
import
unsqueeze
from
infinicore.ops.baddbmm
import
baddbmm
from
infinicore.ops.bilinear
import
bilinear
from
infinicore.ops.fmod
import
fmod
from
infinicore.tensor
import
(
from
infinicore.tensor
import
(
Tensor
,
Tensor
,
empty
,
empty
,
...
...
python/infinicore/nn/functional/adaptive_max_pool1d.py
View file @
2904ec00
from
typing
import
List
from
infinicore.lib
import
_infinicore
from
infinicore.lib
import
_infinicore
from
infinicore.tensor
import
Tensor
from
infinicore.tensor
import
Tensor
...
...
src/infiniop/ops/adaptive_max_pool1d/operator.cc
View file @
2904ec00
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
#include "moore/adaptive_max_pool1d_moore.h"
#include "moore/adaptive_max_pool1d_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateAdaptiveMaxPool1dDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateAdaptiveMaxPool1dDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopAdaptiveMaxPool1dDescriptor_t
*
desc_ptr
,
infiniopAdaptiveMaxPool1dDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
...
@@ -53,7 +53,7 @@ __C infiniStatus_t infiniopCreateAdaptiveMaxPool1dDescriptor(
...
@@ -53,7 +53,7 @@ __C infiniStatus_t infiniopCreateAdaptiveMaxPool1dDescriptor(
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopGetAdaptiveMaxPool1dWorkspaceSize
(
__
INFINI_
C
infiniStatus_t
infiniopGetAdaptiveMaxPool1dWorkspaceSize
(
infiniopAdaptiveMaxPool1dDescriptor_t
desc
,
infiniopAdaptiveMaxPool1dDescriptor_t
desc
,
size_t
*
size
)
{
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
...
@@ -83,7 +83,7 @@ __C infiniStatus_t infiniopGetAdaptiveMaxPool1dWorkspaceSize(
...
@@ -83,7 +83,7 @@ __C infiniStatus_t infiniopGetAdaptiveMaxPool1dWorkspaceSize(
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopAdaptiveMaxPool1d
(
__
INFINI_
C
infiniStatus_t
infiniopAdaptiveMaxPool1d
(
infiniopAdaptiveMaxPool1dDescriptor_t
desc
,
infiniopAdaptiveMaxPool1dDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -117,7 +117,7 @@ __C infiniStatus_t infiniopAdaptiveMaxPool1d(
...
@@ -117,7 +117,7 @@ __C infiniStatus_t infiniopAdaptiveMaxPool1d(
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopDestroyAdaptiveMaxPool1dDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopDestroyAdaptiveMaxPool1dDescriptor
(
infiniopAdaptiveMaxPool1dDescriptor_t
desc
)
{
infiniopAdaptiveMaxPool1dDescriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
...
src/infiniop/ops/asinh/operator.cc
View file @
2904ec00
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
#include "moore/asinh_moore.h"
#include "moore/asinh_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateAsinhDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateAsinhDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopAsinhDescriptor_t
*
desc_ptr
,
infiniopAsinhDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
...
@@ -50,7 +50,7 @@ __C infiniStatus_t infiniopCreateAsinhDescriptor(
...
@@ -50,7 +50,7 @@ __C infiniStatus_t infiniopCreateAsinhDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetAsinhWorkspaceSize
(
infiniopAsinhDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetAsinhWorkspaceSize
(
infiniopAsinhDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -80,7 +80,7 @@ __C infiniStatus_t infiniopGetAsinhWorkspaceSize(infiniopAsinhDescriptor_t desc,
...
@@ -80,7 +80,7 @@ __C infiniStatus_t infiniopGetAsinhWorkspaceSize(infiniopAsinhDescriptor_t desc,
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopAsinh
(
infiniopAsinhDescriptor_t
desc
,
__
INFINI_
C
infiniStatus_t
infiniopAsinh
(
infiniopAsinhDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
void
*
y
,
void
*
y
,
...
@@ -112,7 +112,7 @@ __C infiniStatus_t infiniopAsinh(infiniopAsinhDescriptor_t desc,
...
@@ -112,7 +112,7 @@ __C infiniStatus_t infiniopAsinh(infiniopAsinhDescriptor_t desc,
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
infiniopDestroyAsinhDescriptor
(
infiniopAsinhDescriptor_t
desc
)
{
__
INFINI_
C
infiniStatus_t
infiniopDestroyAsinhDescriptor
(
infiniopAsinhDescriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
case CASE: \
case CASE: \
delete reinterpret_cast<op::asinh::NAMESPACE::Descriptor *>(desc); \
delete reinterpret_cast<op::asinh::NAMESPACE::Descriptor *>(desc); \
...
...
src/infiniop/ops/fmod/operator.cc
View file @
2904ec00
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
#include "moore/fmod_moore.h"
#include "moore/fmod_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateFmodDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateFmodDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopFmodDescriptor_t
*
desc_ptr
,
infiniopFmodDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
c_desc
,
infiniopTensorDescriptor_t
c_desc
,
...
@@ -54,7 +54,7 @@ __C infiniStatus_t infiniopCreateFmodDescriptor(
...
@@ -54,7 +54,7 @@ __C infiniStatus_t infiniopCreateFmodDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetFmodWorkspaceSize
(
infiniopFmodDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetFmodWorkspaceSize
(
infiniopFmodDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -84,7 +84,7 @@ __C infiniStatus_t infiniopGetFmodWorkspaceSize(infiniopFmodDescriptor_t desc, s
...
@@ -84,7 +84,7 @@ __C infiniStatus_t infiniopGetFmodWorkspaceSize(infiniopFmodDescriptor_t desc, s
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopFmod
(
__
INFINI_
C
infiniStatus_t
infiniopFmod
(
infiniopFmodDescriptor_t
desc
,
infiniopFmodDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -119,7 +119,7 @@ __C infiniStatus_t infiniopFmod(
...
@@ -119,7 +119,7 @@ __C infiniStatus_t infiniopFmod(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
infiniopDestroyFmodDescriptor
(
infiniopFmodDescriptor_t
desc
)
{
__
INFINI_
C
infiniStatus_t
infiniopDestroyFmodDescriptor
(
infiniopFmodDescriptor_t
desc
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment