22 lines
580 B
C
22 lines
580 B
C
#pragma once
|
|
#include <stdint.h>
|
|
#include <stdbool.h>
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
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 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
|
|
}
|
|
#endif
|