Commit 2904ec00 authored by PanZezhong's avatar PanZezhong
Browse files

issue/1031 fix T1-1-9 format

parent 6a0503fc
...@@ -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
...@@ -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
...@@ -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
...@@ -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,
......
from typing import List
from infinicore.lib import _infinicore from infinicore.lib import _infinicore
from infinicore.tensor import Tensor from infinicore.tensor import Tensor
......
...@@ -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: \
......
...@@ -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); \
......
...@@ -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: \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment