?gemm_pack_get_size
返回存储pack后的矩阵所需的字节数。
接口定义
C interface：
size_t cblas_sgemm_pack_get_size(const enum CBLAS_IDENTIFIER identifier, const BLASINT m, const BLASINT n, const BLASINT k);
size_t cblas_dgemm_pack_get_size(const enum CBLAS_IDENTIFIER identifier, const BLASINT m, const BLASINT n, const BLASINT k);
size_t cblas_cgemm_pack_get_size(const enum CBLAS_IDENTIFIER identifier, const BLASINT m, const BLASINT n, const BLASINT k);
size_t cblas_zgemm_pack_get_size(const enum CBLAS_IDENTIFIER identifier, const BLASINT m, const BLASINT n, const BLASINT k);
size_t cblas_bgemm_pack_get_size(const enum CBLAS_IDENTIFIER identifier, const BLASINT m, const BLASINT n, const BLASINT k);
参数
参数名 |
类型 |
描述 |
输入/输出 |
|---|---|---|---|
identifier |
枚举类型CBLAS_IDENTIFIER |
指定要pack的矩阵。
|
输入 |
m |
整型数 |
需要pack的矩阵A的行。 |
输入 |
n |
整型数 |
需要pack的矩阵B的列。 |
输入 |
k |
整型数 |
需要pack的矩阵A的列和B的行。 |
输入 |
依赖
#include "kblas.h"
示例
C interface：
1 2 3 4 5 6 | int m = 4, k = 3, n = 4; size_t size = cblas_sgemm_pack_get_size(CblasA, m, n, k); /* * Output size: 64 */ |