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
b1ee0a8a
Commit
b1ee0a8a
authored
Mar 05, 2026
by
wooway777
Browse files
issue/1033 - replace __C with __INFINI_C
parent
c17e3b83
Changes
87
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
79 additions
and
79 deletions
+79
-79
src/infiniop/ops/logsoftmax/operator.cc
src/infiniop/ops/logsoftmax/operator.cc
+4
-4
src/infiniop/ops/lp_norm/operator.cc
src/infiniop/ops/lp_norm/operator.cc
+4
-4
src/infiniop/ops/mul/operator.cc
src/infiniop/ops/mul/operator.cc
+4
-4
src/infiniop/ops/ones/operator.cc
src/infiniop/ops/ones/operator.cc
+4
-4
src/infiniop/ops/paged_attention/operator.cc
src/infiniop/ops/paged_attention/operator.cc
+4
-4
src/infiniop/ops/paged_attention_prefill/operator.cc
src/infiniop/ops/paged_attention_prefill/operator.cc
+4
-4
src/infiniop/ops/paged_caching/operator.cc
src/infiniop/ops/paged_caching/operator.cc
+4
-4
src/infiniop/ops/quant/per_channel_quant_int8/operator.cc
src/infiniop/ops/quant/per_channel_quant_int8/operator.cc
+4
-4
src/infiniop/ops/random_sample/operator.cc
src/infiniop/ops/random_sample/operator.cc
+4
-4
src/infiniop/ops/rearrange/operator.cc
src/infiniop/ops/rearrange/operator.cc
+3
-3
src/infiniop/ops/relu/operator.cc
src/infiniop/ops/relu/operator.cc
+4
-4
src/infiniop/ops/rms_norm/operator.cc
src/infiniop/ops/rms_norm/operator.cc
+4
-4
src/infiniop/ops/rope/operator.cc
src/infiniop/ops/rope/operator.cc
+4
-4
src/infiniop/ops/scaled_mm/operator.cc
src/infiniop/ops/scaled_mm/operator.cc
+4
-4
src/infiniop/ops/sigmoid/operator.cc
src/infiniop/ops/sigmoid/operator.cc
+4
-4
src/infiniop/ops/silu/operator.cc
src/infiniop/ops/silu/operator.cc
+4
-4
src/infiniop/ops/silu_and_mul/operator.cc
src/infiniop/ops/silu_and_mul/operator.cc
+4
-4
src/infiniop/ops/softmax/operator.cc
src/infiniop/ops/softmax/operator.cc
+4
-4
src/infiniop/ops/softplus/operator.cc
src/infiniop/ops/softplus/operator.cc
+4
-4
src/infiniop/ops/sub/operator.cc
src/infiniop/ops/sub/operator.cc
+4
-4
No files found.
src/infiniop/ops/logsoftmax/operator.cc
View file @
b1ee0a8a
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
// #include "ascend/logsoftmax_ascend.h"
// #include "ascend/logsoftmax_ascend.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateLogSoftmaxDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateLogSoftmaxDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopLogSoftmaxDescriptor_t
*
desc_ptr
,
infiniopLogSoftmaxDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
...
@@ -56,7 +56,7 @@ __C infiniStatus_t infiniopCreateLogSoftmaxDescriptor(
...
@@ -56,7 +56,7 @@ __C infiniStatus_t infiniopCreateLogSoftmaxDescriptor(
}
}
}
}
__C
infiniStatus_t
infiniopGetLogSoftmaxWorkspaceSize
(
infiniopLogSoftmaxDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetLogSoftmaxWorkspaceSize
(
infiniopLogSoftmaxDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -90,7 +90,7 @@ __C infiniStatus_t infiniopGetLogSoftmaxWorkspaceSize(infiniopLogSoftmaxDescript
...
@@ -90,7 +90,7 @@ __C infiniStatus_t infiniopGetLogSoftmaxWorkspaceSize(infiniopLogSoftmaxDescript
}
}
}
}
__C
infiniStatus_t
infiniopLogSoftmax
(
__
INFINI_
C
infiniStatus_t
infiniopLogSoftmax
(
infiniopLogSoftmaxDescriptor_t
desc
,
infiniopLogSoftmaxDescriptor_t
desc
,
void
*
workspace
,
size_t
workspace_size
,
void
*
workspace
,
size_t
workspace_size
,
void
*
y
,
void
*
y
,
...
@@ -129,7 +129,7 @@ __C infiniStatus_t infiniopLogSoftmax(
...
@@ -129,7 +129,7 @@ __C infiniStatus_t infiniopLogSoftmax(
}
}
}
}
__C
infiniStatus_t
infiniopDestroyLogSoftmaxDescriptor
(
infiniopLogSoftmaxDescriptor_t
desc
)
{
__
INFINI_
C
infiniStatus_t
infiniopDestroyLogSoftmaxDescriptor
(
infiniopLogSoftmaxDescriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
...
src/infiniop/ops/lp_norm/operator.cc
View file @
b1ee0a8a
...
@@ -6,7 +6,7 @@
...
@@ -6,7 +6,7 @@
#include "nvidia/lp_norm_nvidia.cuh"
#include "nvidia/lp_norm_nvidia.cuh"
#endif
#endif
__C
infiniStatus_t
infiniopCreateLPNormDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateLPNormDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopLPNormDescriptor_t
*
desc_ptr
,
infiniopLPNormDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
output_desc
,
infiniopTensorDescriptor_t
output_desc
,
...
@@ -47,7 +47,7 @@ __C infiniStatus_t infiniopCreateLPNormDescriptor(
...
@@ -47,7 +47,7 @@ __C infiniStatus_t infiniopCreateLPNormDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetLPNormWorkspaceSize
(
infiniopLPNormDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetLPNormWorkspaceSize
(
infiniopLPNormDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
*size = reinterpret_cast<op::lp_norm::NAMESPACE::Descriptor *>(desc)->workspaceSize(); \
*size = reinterpret_cast<op::lp_norm::NAMESPACE::Descriptor *>(desc)->workspaceSize(); \
...
@@ -75,7 +75,7 @@ __C infiniStatus_t infiniopGetLPNormWorkspaceSize(infiniopLPNormDescriptor_t des
...
@@ -75,7 +75,7 @@ __C infiniStatus_t infiniopGetLPNormWorkspaceSize(infiniopLPNormDescriptor_t des
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopLPNorm
(
__
INFINI_
C
infiniStatus_t
infiniopLPNorm
(
infiniopLPNormDescriptor_t
desc
,
infiniopLPNormDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -114,7 +114,7 @@ __C infiniStatus_t infiniopLPNorm(
...
@@ -114,7 +114,7 @@ __C infiniStatus_t infiniopLPNorm(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
__
INFINI_
C
infiniStatus_t
infiniopDestroyLPNormDescriptor
(
infiniopLPNormDescriptor_t
desc
)
{
infiniopDestroyLPNormDescriptor
(
infiniopLPNormDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
src/infiniop/ops/mul/operator.cc
View file @
b1ee0a8a
...
@@ -18,7 +18,7 @@
...
@@ -18,7 +18,7 @@
#include "moore/mul_moore.h"
#include "moore/mul_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateMulDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateMulDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopMulDescriptor_t
*
desc_ptr
,
infiniopMulDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
c_desc
,
infiniopTensorDescriptor_t
c_desc
,
...
@@ -68,7 +68,7 @@ __C infiniStatus_t infiniopCreateMulDescriptor(
...
@@ -68,7 +68,7 @@ __C infiniStatus_t infiniopCreateMulDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetMulWorkspaceSize
(
infiniopMulDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetMulWorkspaceSize
(
infiniopMulDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -109,7 +109,7 @@ __C infiniStatus_t infiniopGetMulWorkspaceSize(infiniopMulDescriptor_t desc, siz
...
@@ -109,7 +109,7 @@ __C infiniStatus_t infiniopGetMulWorkspaceSize(infiniopMulDescriptor_t desc, siz
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopMul
(
__
INFINI_
C
infiniStatus_t
infiniopMul
(
infiniopMulDescriptor_t
desc
,
infiniopMulDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -157,7 +157,7 @@ __C infiniStatus_t infiniopMul(
...
@@ -157,7 +157,7 @@ __C infiniStatus_t infiniopMul(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
__
INFINI_
C
infiniStatus_t
infiniopDestroyMulDescriptor
(
infiniopMulDescriptor_t
desc
)
{
infiniopDestroyMulDescriptor
(
infiniopMulDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
src/infiniop/ops/ones/operator.cc
View file @
b1ee0a8a
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
#include "moore/ones_moore.h"
#include "moore/ones_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateOnesDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateOnesDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopOnesDescriptor_t
*
desc_ptr
,
infiniopOnesDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
...
@@ -60,7 +60,7 @@ __C infiniStatus_t infiniopCreateOnesDescriptor(
...
@@ -60,7 +60,7 @@ __C infiniStatus_t infiniopCreateOnesDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetOnesWorkspaceSize
(
infiniopOnesDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetOnesWorkspaceSize
(
infiniopOnesDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -98,7 +98,7 @@ __C infiniStatus_t infiniopGetOnesWorkspaceSize(infiniopOnesDescriptor_t desc, s
...
@@ -98,7 +98,7 @@ __C infiniStatus_t infiniopGetOnesWorkspaceSize(infiniopOnesDescriptor_t desc, s
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopOnes
(
__
INFINI_
C
infiniStatus_t
infiniopOnes
(
infiniopOnesDescriptor_t
desc
,
infiniopOnesDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -142,7 +142,7 @@ __C infiniStatus_t infiniopOnes(
...
@@ -142,7 +142,7 @@ __C infiniStatus_t infiniopOnes(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
__
INFINI_
C
infiniStatus_t
infiniopDestroyOnesDescriptor
(
infiniopOnesDescriptor_t
desc
)
{
infiniopDestroyOnesDescriptor
(
infiniopOnesDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
src/infiniop/ops/paged_attention/operator.cc
View file @
b1ee0a8a
...
@@ -12,7 +12,7 @@
...
@@ -12,7 +12,7 @@
#include "metax/paged_attention_metax.h"
#include "metax/paged_attention_metax.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreatePagedAttentionDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreatePagedAttentionDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopPagedAttentionDescriptor_t
*
desc_ptr
,
infiniopPagedAttentionDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
out_desc
,
infiniopTensorDescriptor_t
out_desc
,
...
@@ -54,7 +54,7 @@ __C infiniStatus_t infiniopCreatePagedAttentionDescriptor(
...
@@ -54,7 +54,7 @@ __C infiniStatus_t infiniopCreatePagedAttentionDescriptor(
}
}
}
}
__C
infiniStatus_t
infiniopGetPagedAttentionWorkspaceSize
(
__
INFINI_
C
infiniStatus_t
infiniopGetPagedAttentionWorkspaceSize
(
infiniopPagedAttentionDescriptor_t
desc
,
infiniopPagedAttentionDescriptor_t
desc
,
size_t
*
size
)
{
size_t
*
size
)
{
...
@@ -84,7 +84,7 @@ __C infiniStatus_t infiniopGetPagedAttentionWorkspaceSize(
...
@@ -84,7 +84,7 @@ __C infiniStatus_t infiniopGetPagedAttentionWorkspaceSize(
}
}
}
}
__C
infiniStatus_t
infiniopPagedAttention
(
__
INFINI_
C
infiniStatus_t
infiniopPagedAttention
(
infiniopPagedAttentionDescriptor_t
desc
,
infiniopPagedAttentionDescriptor_t
desc
,
void
*
workspace
,
size_t
workspace_size
,
void
*
workspace
,
size_t
workspace_size
,
void
*
out
,
const
void
*
q
,
const
void
*
k_cache
,
const
void
*
v_cache
,
void
*
out
,
const
void
*
q
,
const
void
*
k_cache
,
const
void
*
v_cache
,
...
@@ -118,7 +118,7 @@ __C infiniStatus_t infiniopPagedAttention(
...
@@ -118,7 +118,7 @@ __C infiniStatus_t infiniopPagedAttention(
}
}
}
}
__C
infiniStatus_t
infiniopDestroyPagedAttentionDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopDestroyPagedAttentionDescriptor
(
infiniopPagedAttentionDescriptor_t
desc
)
{
infiniopPagedAttentionDescriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
...
...
src/infiniop/ops/paged_attention_prefill/operator.cc
View file @
b1ee0a8a
...
@@ -12,7 +12,7 @@
...
@@ -12,7 +12,7 @@
#include "moore/paged_attention_prefill_moore.h"
#include "moore/paged_attention_prefill_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreatePagedAttentionPrefillDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreatePagedAttentionPrefillDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopPagedAttentionPrefillDescriptor_t
*
desc_ptr
,
infiniopPagedAttentionPrefillDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
out_desc
,
infiniopTensorDescriptor_t
out_desc
,
...
@@ -56,7 +56,7 @@ __C infiniStatus_t infiniopCreatePagedAttentionPrefillDescriptor(
...
@@ -56,7 +56,7 @@ __C infiniStatus_t infiniopCreatePagedAttentionPrefillDescriptor(
}
}
}
}
__C
infiniStatus_t
infiniopGetPagedAttentionPrefillWorkspaceSize
(
__
INFINI_
C
infiniStatus_t
infiniopGetPagedAttentionPrefillWorkspaceSize
(
infiniopPagedAttentionPrefillDescriptor_t
desc
,
infiniopPagedAttentionPrefillDescriptor_t
desc
,
size_t
*
size
)
{
size_t
*
size
)
{
...
@@ -86,7 +86,7 @@ __C infiniStatus_t infiniopGetPagedAttentionPrefillWorkspaceSize(
...
@@ -86,7 +86,7 @@ __C infiniStatus_t infiniopGetPagedAttentionPrefillWorkspaceSize(
}
}
}
}
__C
infiniStatus_t
infiniopPagedAttentionPrefill
(
__
INFINI_
C
infiniStatus_t
infiniopPagedAttentionPrefill
(
infiniopPagedAttentionPrefillDescriptor_t
desc
,
infiniopPagedAttentionPrefillDescriptor_t
desc
,
void
*
workspace
,
size_t
workspace_size
,
void
*
workspace
,
size_t
workspace_size
,
void
*
out
,
const
void
*
q
,
const
void
*
k_cache
,
const
void
*
v_cache
,
void
*
out
,
const
void
*
q
,
const
void
*
k_cache
,
const
void
*
v_cache
,
...
@@ -123,7 +123,7 @@ __C infiniStatus_t infiniopPagedAttentionPrefill(
...
@@ -123,7 +123,7 @@ __C infiniStatus_t infiniopPagedAttentionPrefill(
}
}
}
}
__C
infiniStatus_t
infiniopDestroyPagedAttentionPrefillDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopDestroyPagedAttentionPrefillDescriptor
(
infiniopPagedAttentionPrefillDescriptor_t
desc
)
{
infiniopPagedAttentionPrefillDescriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
...
...
src/infiniop/ops/paged_caching/operator.cc
View file @
b1ee0a8a
...
@@ -12,7 +12,7 @@
...
@@ -12,7 +12,7 @@
#include "moore/paged_caching_moore.h"
#include "moore/paged_caching_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreatePagedCachingDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreatePagedCachingDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopPagedCachingDescriptor_t
*
desc_ptr
,
infiniopPagedCachingDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
k_cache_desc
,
infiniopTensorDescriptor_t
k_cache_desc
,
...
@@ -49,7 +49,7 @@ __C infiniStatus_t infiniopCreatePagedCachingDescriptor(
...
@@ -49,7 +49,7 @@ __C infiniStatus_t infiniopCreatePagedCachingDescriptor(
}
}
}
}
__C
infiniStatus_t
infiniopGetPagedCachingWorkspaceSize
(
__
INFINI_
C
infiniStatus_t
infiniopGetPagedCachingWorkspaceSize
(
infiniopPagedCachingDescriptor_t
desc
,
infiniopPagedCachingDescriptor_t
desc
,
size_t
*
size
)
{
size_t
*
size
)
{
...
@@ -79,7 +79,7 @@ __C infiniStatus_t infiniopGetPagedCachingWorkspaceSize(
...
@@ -79,7 +79,7 @@ __C infiniStatus_t infiniopGetPagedCachingWorkspaceSize(
}
}
}
}
__C
infiniStatus_t
infiniopPagedCaching
(
__
INFINI_
C
infiniStatus_t
infiniopPagedCaching
(
infiniopPagedCachingDescriptor_t
desc
,
infiniopPagedCachingDescriptor_t
desc
,
void
*
workspace
,
size_t
workspace_size
,
void
*
workspace
,
size_t
workspace_size
,
void
*
k_cache
,
void
*
v_cache
,
void
*
k_cache
,
void
*
v_cache
,
...
@@ -113,7 +113,7 @@ __C infiniStatus_t infiniopPagedCaching(
...
@@ -113,7 +113,7 @@ __C infiniStatus_t infiniopPagedCaching(
}
}
}
}
__C
infiniStatus_t
infiniopDestroyPagedCachingDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopDestroyPagedCachingDescriptor
(
infiniopPagedCachingDescriptor_t
desc
)
{
infiniopPagedCachingDescriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
...
...
src/infiniop/ops/quant/per_channel_quant_int8/operator.cc
View file @
b1ee0a8a
...
@@ -9,7 +9,7 @@
...
@@ -9,7 +9,7 @@
#include "moore/per_channel_quant_int8_moore.h"
#include "moore/per_channel_quant_int8_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreatePerChannelQuantI8Descriptor
(
infiniopHandle_t
handle
,
__
INFINI_
C
infiniStatus_t
infiniopCreatePerChannelQuantI8Descriptor
(
infiniopHandle_t
handle
,
infiniopPerChannelQuantI8Descriptor_t
*
desc_ptr
,
infiniopPerChannelQuantI8Descriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
x_packed_desc
,
infiniopTensorDescriptor_t
x_packed_desc
,
infiniopTensorDescriptor_t
x_scale_desc
,
infiniopTensorDescriptor_t
x_scale_desc
,
...
@@ -40,7 +40,7 @@ __C infiniStatus_t infiniopCreatePerChannelQuantI8Descriptor(infiniopHandle_t ha
...
@@ -40,7 +40,7 @@ __C infiniStatus_t infiniopCreatePerChannelQuantI8Descriptor(infiniopHandle_t ha
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetPerChannelQuantI8WorkspaceSize
(
infiniopPerChannelQuantI8Descriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetPerChannelQuantI8WorkspaceSize
(
infiniopPerChannelQuantI8Descriptor_t
desc
,
size_t
*
size
)
{
switch
(
desc
->
device_type
)
{
switch
(
desc
->
device_type
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -61,7 +61,7 @@ __C infiniStatus_t infiniopGetPerChannelQuantI8WorkspaceSize(infiniopPerChannelQ
...
@@ -61,7 +61,7 @@ __C infiniStatus_t infiniopGetPerChannelQuantI8WorkspaceSize(infiniopPerChannelQ
#undef GET
#undef GET
}
}
__C
infiniStatus_t
infiniopPerChannelQuantI8
(
infiniopPerChannelQuantI8Descriptor_t
desc
,
__
INFINI_
C
infiniStatus_t
infiniopPerChannelQuantI8
(
infiniopPerChannelQuantI8Descriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
void
*
x_packed
,
void
*
x_packed
,
...
@@ -90,7 +90,7 @@ __C infiniStatus_t infiniopPerChannelQuantI8(infiniopPerChannelQuantI8Descriptor
...
@@ -90,7 +90,7 @@ __C infiniStatus_t infiniopPerChannelQuantI8(infiniopPerChannelQuantI8Descriptor
#undef QUANT
#undef QUANT
}
}
__C
infiniStatus_t
infiniopDestroyPerChannelQuantI8Descriptor
(
infiniopPerChannelQuantI8Descriptor_t
desc
)
{
__
INFINI_
C
infiniStatus_t
infiniopDestroyPerChannelQuantI8Descriptor
(
infiniopPerChannelQuantI8Descriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
case CASE: \
case CASE: \
delete reinterpret_cast<op::per_channel_quant_int8::NAMESPACE::Descriptor *>(desc); \
delete reinterpret_cast<op::per_channel_quant_int8::NAMESPACE::Descriptor *>(desc); \
...
...
src/infiniop/ops/random_sample/operator.cc
View file @
b1ee0a8a
...
@@ -24,7 +24,7 @@
...
@@ -24,7 +24,7 @@
#include "kunlun/random_sample_kunlun.h"
#include "kunlun/random_sample_kunlun.h"
#endif
#endif
__C
infiniStatus_t
__
INFINI_
C
infiniStatus_t
infiniopCreateRandomSampleDescriptor
(
infiniopCreateRandomSampleDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopRandomSampleDescriptor_t
*
desc_ptr
,
infiniopRandomSampleDescriptor_t
*
desc_ptr
,
...
@@ -82,7 +82,7 @@ infiniopCreateRandomSampleDescriptor(
...
@@ -82,7 +82,7 @@ infiniopCreateRandomSampleDescriptor(
#undef CREATE
#undef CREATE
};
};
__C
infiniStatus_t
infiniopGetRandomSampleWorkspaceSize
(
__
INFINI_
C
infiniStatus_t
infiniopGetRandomSampleWorkspaceSize
(
infiniopRandomSampleDescriptor_t
desc
,
infiniopRandomSampleDescriptor_t
desc
,
size_t
*
size
)
{
size_t
*
size
)
{
...
@@ -136,7 +136,7 @@ __C infiniStatus_t infiniopGetRandomSampleWorkspaceSize(
...
@@ -136,7 +136,7 @@ __C infiniStatus_t infiniopGetRandomSampleWorkspaceSize(
#undef GET
#undef GET
}
}
__C
infiniStatus_t
infiniopRandomSample
(
__
INFINI_
C
infiniStatus_t
infiniopRandomSample
(
infiniopRandomSampleDescriptor_t
desc
,
infiniopRandomSampleDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -200,7 +200,7 @@ __C infiniStatus_t infiniopRandomSample(
...
@@ -200,7 +200,7 @@ __C infiniStatus_t infiniopRandomSample(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
infiniopDestroyRandomSampleDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopDestroyRandomSampleDescriptor
(
infiniopRandomSampleDescriptor_t
desc
)
{
infiniopRandomSampleDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
src/infiniop/ops/rearrange/operator.cc
View file @
b1ee0a8a
...
@@ -24,7 +24,7 @@
...
@@ -24,7 +24,7 @@
#include "kunlun/rearrange_kunlun.h"
#include "kunlun/rearrange_kunlun.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateRearrangeDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateRearrangeDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopRearrangeDescriptor_t
*
desc_ptr
,
infiniopRearrangeDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
dst
,
infiniopTensorDescriptor_t
dst
,
...
@@ -80,7 +80,7 @@ __C infiniStatus_t infiniopCreateRearrangeDescriptor(
...
@@ -80,7 +80,7 @@ __C infiniStatus_t infiniopCreateRearrangeDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopRearrange
(
__
INFINI_
C
infiniStatus_t
infiniopRearrange
(
infiniopRearrangeDescriptor_t
desc
,
infiniopRearrangeDescriptor_t
desc
,
void
*
dst
,
void
*
dst
,
const
void
*
src
,
const
void
*
src
,
...
@@ -134,7 +134,7 @@ __C infiniStatus_t infiniopRearrange(
...
@@ -134,7 +134,7 @@ __C infiniStatus_t infiniopRearrange(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
infiniopDestroyRearrangeDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopDestroyRearrangeDescriptor
(
infiniopRearrangeDescriptor_t
desc
)
{
infiniopRearrangeDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
src/infiniop/ops/relu/operator.cc
View file @
b1ee0a8a
...
@@ -14,7 +14,7 @@
...
@@ -14,7 +14,7 @@
#endif
#endif
#endif
#endif
__C
infiniStatus_t
infiniopCreateReluDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateReluDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopReluDescriptor_t
*
desc_ptr
,
infiniopReluDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
...
@@ -58,7 +58,7 @@ __C infiniStatus_t infiniopCreateReluDescriptor(
...
@@ -58,7 +58,7 @@ __C infiniStatus_t infiniopCreateReluDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetReluWorkspaceSize
(
infiniopReluDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetReluWorkspaceSize
(
infiniopReluDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -95,7 +95,7 @@ __C infiniStatus_t infiniopGetReluWorkspaceSize(infiniopReluDescriptor_t desc, s
...
@@ -95,7 +95,7 @@ __C infiniStatus_t infiniopGetReluWorkspaceSize(infiniopReluDescriptor_t desc, s
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopRelu
(
__
INFINI_
C
infiniStatus_t
infiniopRelu
(
infiniopReluDescriptor_t
desc
,
infiniopReluDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -138,7 +138,7 @@ __C infiniStatus_t infiniopRelu(
...
@@ -138,7 +138,7 @@ __C infiniStatus_t infiniopRelu(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
__
INFINI_
C
infiniStatus_t
infiniopDestroyReluDescriptor
(
infiniopReluDescriptor_t
desc
)
{
infiniopDestroyReluDescriptor
(
infiniopReluDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
src/infiniop/ops/rms_norm/operator.cc
View file @
b1ee0a8a
...
@@ -24,7 +24,7 @@
...
@@ -24,7 +24,7 @@
#include "kunlun/rms_norm_kunlun.h"
#include "kunlun/rms_norm_kunlun.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateRMSNormDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateRMSNormDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopRMSNormDescriptor_t
*
desc_ptr
,
infiniopRMSNormDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
...
@@ -83,7 +83,7 @@ __C infiniStatus_t infiniopCreateRMSNormDescriptor(
...
@@ -83,7 +83,7 @@ __C infiniStatus_t infiniopCreateRMSNormDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetRMSNormWorkspaceSize
(
infiniopRMSNormDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetRMSNormWorkspaceSize
(
infiniopRMSNormDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -131,7 +131,7 @@ __C infiniStatus_t infiniopGetRMSNormWorkspaceSize(infiniopRMSNormDescriptor_t d
...
@@ -131,7 +131,7 @@ __C infiniStatus_t infiniopGetRMSNormWorkspaceSize(infiniopRMSNormDescriptor_t d
#undef GET
#undef GET
}
}
__C
infiniStatus_t
infiniopRMSNorm
(
infiniopRMSNormDescriptor_t
desc
,
void
*
workspace
,
size_t
workspace_size
,
__
INFINI_
C
infiniStatus_t
infiniopRMSNorm
(
infiniopRMSNormDescriptor_t
desc
,
void
*
workspace
,
size_t
workspace_size
,
void
*
y
,
const
void
*
x
,
const
void
*
w
,
void
*
stream
)
{
void
*
y
,
const
void
*
x
,
const
void
*
w
,
void
*
stream
)
{
#define CALCULATE(CASE, NAMESPACE) \
#define CALCULATE(CASE, NAMESPACE) \
...
@@ -180,7 +180,7 @@ __C infiniStatus_t infiniopRMSNorm(infiniopRMSNormDescriptor_t desc, void *works
...
@@ -180,7 +180,7 @@ __C infiniStatus_t infiniopRMSNorm(infiniopRMSNormDescriptor_t desc, void *works
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
infiniopDestroyRMSNormDescriptor
(
infiniopRMSNormDescriptor_t
desc
)
{
__
INFINI_
C
infiniStatus_t
infiniopDestroyRMSNormDescriptor
(
infiniopRMSNormDescriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
...
src/infiniop/ops/rope/operator.cc
View file @
b1ee0a8a
...
@@ -24,7 +24,7 @@
...
@@ -24,7 +24,7 @@
#include "moore/rope_moore.h"
#include "moore/rope_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateRoPEDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateRoPEDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopRoPEDescriptor_t
*
desc_ptr
,
infiniopRoPEDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y
,
infiniopTensorDescriptor_t
y
,
...
@@ -87,7 +87,7 @@ __C infiniStatus_t infiniopCreateRoPEDescriptor(
...
@@ -87,7 +87,7 @@ __C infiniStatus_t infiniopCreateRoPEDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetRoPEWorkspaceSize
(
infiniopRoPEDescriptor_t
desc
,
__
INFINI_
C
infiniStatus_t
infiniopGetRoPEWorkspaceSize
(
infiniopRoPEDescriptor_t
desc
,
size_t
*
size
)
{
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -135,7 +135,7 @@ __C infiniStatus_t infiniopGetRoPEWorkspaceSize(infiniopRoPEDescriptor_t desc,
...
@@ -135,7 +135,7 @@ __C infiniStatus_t infiniopGetRoPEWorkspaceSize(infiniopRoPEDescriptor_t desc,
#undef GET
#undef GET
}
}
__C
infiniStatus_t
infiniopRoPE
(
__
INFINI_
C
infiniStatus_t
infiniopRoPE
(
infiniopRoPEDescriptor_t
desc
,
infiniopRoPEDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -192,7 +192,7 @@ __C infiniStatus_t infiniopRoPE(
...
@@ -192,7 +192,7 @@ __C infiniStatus_t infiniopRoPE(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
__
INFINI_
C
infiniStatus_t
infiniopDestroyRoPEDescriptor
(
infiniopRoPEDescriptor_t
desc
)
{
infiniopDestroyRoPEDescriptor
(
infiniopRoPEDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
src/infiniop/ops/scaled_mm/operator.cc
View file @
b1ee0a8a
...
@@ -10,7 +10,7 @@
...
@@ -10,7 +10,7 @@
#include "moore/int8_gemm_moore.h"
#include "moore/int8_gemm_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateI8GemmDescriptor
(
infiniopHandle_t
handle
,
__
INFINI_
C
infiniStatus_t
infiniopCreateI8GemmDescriptor
(
infiniopHandle_t
handle
,
infiniopI8GemmDescriptor_t
*
desc_ptr
,
infiniopI8GemmDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
out_desc
,
infiniopTensorDescriptor_t
out_desc
,
infiniopTensorDescriptor_t
bias_desc
,
infiniopTensorDescriptor_t
bias_desc
,
...
@@ -45,7 +45,7 @@ __C infiniStatus_t infiniopCreateI8GemmDescriptor(infiniopHandle_t handle,
...
@@ -45,7 +45,7 @@ __C infiniStatus_t infiniopCreateI8GemmDescriptor(infiniopHandle_t handle,
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetI8GemmWorkspaceSize
(
infiniopI8GemmDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetI8GemmWorkspaceSize
(
infiniopI8GemmDescriptor_t
desc
,
size_t
*
size
)
{
switch
(
desc
->
device_type
)
{
switch
(
desc
->
device_type
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -66,7 +66,7 @@ __C infiniStatus_t infiniopGetI8GemmWorkspaceSize(infiniopI8GemmDescriptor_t des
...
@@ -66,7 +66,7 @@ __C infiniStatus_t infiniopGetI8GemmWorkspaceSize(infiniopI8GemmDescriptor_t des
#undef GET
#undef GET
}
}
__C
infiniStatus_t
infiniopI8Gemm
(
infiniopI8GemmDescriptor_t
desc
,
__
INFINI_
C
infiniStatus_t
infiniopI8Gemm
(
infiniopI8GemmDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
void
*
out
,
void
*
out
,
...
@@ -96,7 +96,7 @@ __C infiniStatus_t infiniopI8Gemm(infiniopI8GemmDescriptor_t desc,
...
@@ -96,7 +96,7 @@ __C infiniStatus_t infiniopI8Gemm(infiniopI8GemmDescriptor_t desc,
#undef CACULATE
#undef CACULATE
}
}
__C
infiniStatus_t
infiniopDestroyI8GemmDescriptor
(
infiniopI8GemmDescriptor_t
desc
)
{
__
INFINI_
C
infiniStatus_t
infiniopDestroyI8GemmDescriptor
(
infiniopI8GemmDescriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
case CASE: \
case CASE: \
delete reinterpret_cast<op::i8gemm::NAMESPACE::Descriptor *>(desc); \
delete reinterpret_cast<op::i8gemm::NAMESPACE::Descriptor *>(desc); \
...
...
src/infiniop/ops/sigmoid/operator.cc
View file @
b1ee0a8a
...
@@ -9,7 +9,7 @@
...
@@ -9,7 +9,7 @@
#include "nvidia/sigmoid_nvidia.cuh"
#include "nvidia/sigmoid_nvidia.cuh"
#endif
#endif
__C
infiniStatus_t
infiniopCreateSigmoidDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateSigmoidDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopSigmoidDescriptor_t
*
desc_ptr
,
infiniopSigmoidDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
...
@@ -48,7 +48,7 @@ __C infiniStatus_t infiniopCreateSigmoidDescriptor(
...
@@ -48,7 +48,7 @@ __C infiniStatus_t infiniopCreateSigmoidDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetSigmoidWorkspaceSize
(
infiniopSigmoidDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetSigmoidWorkspaceSize
(
infiniopSigmoidDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -79,7 +79,7 @@ __C infiniStatus_t infiniopGetSigmoidWorkspaceSize(infiniopSigmoidDescriptor_t d
...
@@ -79,7 +79,7 @@ __C infiniStatus_t infiniopGetSigmoidWorkspaceSize(infiniopSigmoidDescriptor_t d
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopSigmoid
(
__
INFINI_
C
infiniStatus_t
infiniopSigmoid
(
infiniopSigmoidDescriptor_t
desc
,
infiniopSigmoidDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -116,7 +116,7 @@ __C infiniStatus_t infiniopSigmoid(
...
@@ -116,7 +116,7 @@ __C infiniStatus_t infiniopSigmoid(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
__
INFINI_
C
infiniStatus_t
infiniopDestroySigmoidDescriptor
(
infiniopSigmoidDescriptor_t
desc
)
{
infiniopDestroySigmoidDescriptor
(
infiniopSigmoidDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
src/infiniop/ops/silu/operator.cc
View file @
b1ee0a8a
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
#include "moore/silu_moore.h"
#include "moore/silu_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateSiluDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateSiluDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopSiluDescriptor_t
*
desc_ptr
,
infiniopSiluDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
output_desc
,
infiniopTensorDescriptor_t
output_desc
,
...
@@ -57,7 +57,7 @@ __C infiniStatus_t infiniopCreateSiluDescriptor(
...
@@ -57,7 +57,7 @@ __C infiniStatus_t infiniopCreateSiluDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetSiluWorkspaceSize
(
infiniopSiluDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetSiluWorkspaceSize
(
infiniopSiluDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -92,7 +92,7 @@ __C infiniStatus_t infiniopGetSiluWorkspaceSize(infiniopSiluDescriptor_t desc, s
...
@@ -92,7 +92,7 @@ __C infiniStatus_t infiniopGetSiluWorkspaceSize(infiniopSiluDescriptor_t desc, s
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopSilu
(
__
INFINI_
C
infiniStatus_t
infiniopSilu
(
infiniopSiluDescriptor_t
desc
,
infiniopSiluDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -133,7 +133,7 @@ __C infiniStatus_t infiniopSilu(
...
@@ -133,7 +133,7 @@ __C infiniStatus_t infiniopSilu(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
__
INFINI_
C
infiniStatus_t
infiniopDestroySiluDescriptor
(
infiniopSiluDescriptor_t
desc
)
{
infiniopDestroySiluDescriptor
(
infiniopSiluDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
src/infiniop/ops/silu_and_mul/operator.cc
View file @
b1ee0a8a
...
@@ -6,7 +6,7 @@
...
@@ -6,7 +6,7 @@
#include "moore/silu_and_mul_moore.h"
#include "moore/silu_and_mul_moore.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateSiluAndMulDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateSiluAndMulDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopSiluAndMulDescriptor_t
*
desc_ptr
,
infiniopSiluAndMulDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
...
@@ -31,7 +31,7 @@ __C infiniStatus_t infiniopCreateSiluAndMulDescriptor(
...
@@ -31,7 +31,7 @@ __C infiniStatus_t infiniopCreateSiluAndMulDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetSiluAndMulWorkspaceSize
(
infiniopSiluAndMulDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetSiluAndMulWorkspaceSize
(
infiniopSiluAndMulDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -49,7 +49,7 @@ __C infiniStatus_t infiniopGetSiluAndMulWorkspaceSize(infiniopSiluAndMulDescript
...
@@ -49,7 +49,7 @@ __C infiniStatus_t infiniopGetSiluAndMulWorkspaceSize(infiniopSiluAndMulDescript
#undef GET
#undef GET
}
}
__C
infiniStatus_t
infiniopSiluAndMul
(
__
INFINI_
C
infiniStatus_t
infiniopSiluAndMul
(
infiniopSiluAndMulDescriptor_t
desc
,
infiniopSiluAndMulDescriptor_t
desc
,
void
*
workspace
,
size_t
workspace_size
,
void
*
workspace
,
size_t
workspace_size
,
void
*
y
,
void
*
y
,
...
@@ -72,7 +72,7 @@ __C infiniStatus_t infiniopSiluAndMul(
...
@@ -72,7 +72,7 @@ __C infiniStatus_t infiniopSiluAndMul(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
infiniopDestroySiluAndMulDescriptor
(
infiniopSiluAndMulDescriptor_t
desc
)
{
__
INFINI_
C
infiniStatus_t
infiniopDestroySiluAndMulDescriptor
(
infiniopSiluAndMulDescriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
...
src/infiniop/ops/softmax/operator.cc
View file @
b1ee0a8a
...
@@ -6,7 +6,7 @@
...
@@ -6,7 +6,7 @@
#include "nvidia/softmax_nvidia.cuh"
#include "nvidia/softmax_nvidia.cuh"
#endif
#endif
__C
infiniStatus_t
infiniopCreateSoftmaxDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateSoftmaxDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopSoftmaxDescriptor_t
*
desc_ptr
,
infiniopSoftmaxDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
...
@@ -42,7 +42,7 @@ __C infiniStatus_t infiniopCreateSoftmaxDescriptor(
...
@@ -42,7 +42,7 @@ __C infiniStatus_t infiniopCreateSoftmaxDescriptor(
}
}
}
}
__C
infiniStatus_t
infiniopGetSoftmaxWorkspaceSize
(
infiniopSoftmaxDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetSoftmaxWorkspaceSize
(
infiniopSoftmaxDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -70,7 +70,7 @@ __C infiniStatus_t infiniopGetSoftmaxWorkspaceSize(infiniopSoftmaxDescriptor_t d
...
@@ -70,7 +70,7 @@ __C infiniStatus_t infiniopGetSoftmaxWorkspaceSize(infiniopSoftmaxDescriptor_t d
}
}
}
}
__C
infiniStatus_t
infiniopSoftmax
(
__
INFINI_
C
infiniStatus_t
infiniopSoftmax
(
infiniopSoftmaxDescriptor_t
desc
,
infiniopSoftmaxDescriptor_t
desc
,
void
*
workspace
,
size_t
workspace_size
,
void
*
workspace
,
size_t
workspace_size
,
void
*
y
,
void
*
y
,
...
@@ -103,7 +103,7 @@ __C infiniStatus_t infiniopSoftmax(
...
@@ -103,7 +103,7 @@ __C infiniStatus_t infiniopSoftmax(
}
}
}
}
__C
infiniStatus_t
infiniopDestroySoftmaxDescriptor
(
infiniopSoftmaxDescriptor_t
desc
)
{
__
INFINI_
C
infiniStatus_t
infiniopDestroySoftmaxDescriptor
(
infiniopSoftmaxDescriptor_t
desc
)
{
#define DESTROY(CASE, NAMESPACE) \
#define DESTROY(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
...
src/infiniop/ops/softplus/operator.cc
View file @
b1ee0a8a
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
#include "kunlun/softplus_kunlun.h"
#include "kunlun/softplus_kunlun.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateSoftplusDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateSoftplusDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopSoftplusDescriptor_t
*
desc_ptr
,
infiniopSoftplusDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
y_desc
,
infiniopTensorDescriptor_t
y_desc
,
...
@@ -60,7 +60,7 @@ __C infiniStatus_t infiniopCreateSoftplusDescriptor(
...
@@ -60,7 +60,7 @@ __C infiniStatus_t infiniopCreateSoftplusDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetSoftplusWorkspaceSize
(
infiniopSoftplusDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetSoftplusWorkspaceSize
(
infiniopSoftplusDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -98,7 +98,7 @@ __C infiniStatus_t infiniopGetSoftplusWorkspaceSize(infiniopSoftplusDescriptor_t
...
@@ -98,7 +98,7 @@ __C infiniStatus_t infiniopGetSoftplusWorkspaceSize(infiniopSoftplusDescriptor_t
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopSoftplus
(
__
INFINI_
C
infiniStatus_t
infiniopSoftplus
(
infiniopSoftplusDescriptor_t
desc
,
infiniopSoftplusDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -142,7 +142,7 @@ __C infiniStatus_t infiniopSoftplus(
...
@@ -142,7 +142,7 @@ __C infiniStatus_t infiniopSoftplus(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
__
INFINI_
C
infiniStatus_t
infiniopDestroySoftplusDescriptor
(
infiniopSoftplusDescriptor_t
desc
)
{
infiniopDestroySoftplusDescriptor
(
infiniopSoftplusDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
src/infiniop/ops/sub/operator.cc
View file @
b1ee0a8a
...
@@ -15,7 +15,7 @@
...
@@ -15,7 +15,7 @@
#include "kunlun/sub_kunlun.h"
#include "kunlun/sub_kunlun.h"
#endif
#endif
__C
infiniStatus_t
infiniopCreateSubDescriptor
(
__
INFINI_
C
infiniStatus_t
infiniopCreateSubDescriptor
(
infiniopHandle_t
handle
,
infiniopHandle_t
handle
,
infiniopSubDescriptor_t
*
desc_ptr
,
infiniopSubDescriptor_t
*
desc_ptr
,
infiniopTensorDescriptor_t
c_desc
,
infiniopTensorDescriptor_t
c_desc
,
...
@@ -62,7 +62,7 @@ __C infiniStatus_t infiniopCreateSubDescriptor(
...
@@ -62,7 +62,7 @@ __C infiniStatus_t infiniopCreateSubDescriptor(
#undef CREATE
#undef CREATE
}
}
__C
infiniStatus_t
infiniopGetSubWorkspaceSize
(
infiniopSubDescriptor_t
desc
,
size_t
*
size
)
{
__
INFINI_
C
infiniStatus_t
infiniopGetSubWorkspaceSize
(
infiniopSubDescriptor_t
desc
,
size_t
*
size
)
{
#define GET(CASE, NAMESPACE) \
#define GET(CASE, NAMESPACE) \
case CASE: \
case CASE: \
...
@@ -100,7 +100,7 @@ __C infiniStatus_t infiniopGetSubWorkspaceSize(infiniopSubDescriptor_t desc, siz
...
@@ -100,7 +100,7 @@ __C infiniStatus_t infiniopGetSubWorkspaceSize(infiniopSubDescriptor_t desc, siz
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
return
INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED
;
}
}
__C
infiniStatus_t
infiniopSub
(
__
INFINI_
C
infiniStatus_t
infiniopSub
(
infiniopSubDescriptor_t
desc
,
infiniopSubDescriptor_t
desc
,
void
*
workspace
,
void
*
workspace
,
size_t
workspace_size
,
size_t
workspace_size
,
...
@@ -145,7 +145,7 @@ __C infiniStatus_t infiniopSub(
...
@@ -145,7 +145,7 @@ __C infiniStatus_t infiniopSub(
#undef CALCULATE
#undef CALCULATE
}
}
__C
infiniStatus_t
__
INFINI_
C
infiniStatus_t
infiniopDestroySubDescriptor
(
infiniopSubDescriptor_t
desc
)
{
infiniopDestroySubDescriptor
(
infiniopSubDescriptor_t
desc
)
{
#define DELETE(CASE, NAMESPACE) \
#define DELETE(CASE, NAMESPACE) \
...
...
Prev
1
2
3
4
5
Next
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