diff options
Diffstat (limited to 'iqk_mul_mat.h')
| -rw-r--r-- | iqk_mul_mat.h | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/iqk_mul_mat.h b/iqk_mul_mat.h index 4706714b..c1db5eee 100644 --- a/iqk_mul_mat.h +++ b/iqk_mul_mat.h @@ -5,11 +5,15 @@ extern "C" { #endif -bool iqk_mul_mat(long Nx, long Ny, long ne00, int typeA, const void * A, const void * B, +bool iqk_mul_mat(int task_type, long Nx, long Ny, long ne00, + int typeA, const void * A, long strideA, + int typeB, const void * B, long strideB, float * C, long stride_C, int ith, int nth); -bool iqk_mul_mat_moe(long, long, long, int, int, const void *, const void *, - float *, long, long, const void *, int, int); +bool iqk_mul_mat_moe(long Nx, long Ny, long ne00, int ne11, + int typeA, const void * A, long strideA, + int typeB, const void * B, long strideB, + float * C, long nb1, long nb2, const void * vrow_mapping, int ith, int nth); #ifdef __cplusplus |