kutacc_core_bgemm_pack_get_size
获取存储pack后的矩阵所需的字节数。
接口定义
size_t kutacc_core_bgemm_pack_get_size(char matrixIdentifier, char transA, char transB, const BLASINT m, const BLASINT n, const BLASINT k);
参数
参数名 |
类型 |
描述 |
输入/输出 |
|---|---|---|---|
matrixIdentifier |
字符型 |
指定要pack的矩阵,默认为B矩阵。
|
输入 |
m |
整数型 |
需要pack的矩阵A的行。 |
输入 |
n |
整数型 |
需要pack的矩阵B的列。 |
输入 |
k |
整数型 |
需要pack的矩阵A的列和B的行。 |
输入 |
依赖
#include "kutacc_core.h"
示例
C interface:
1 2 3 4 5 6 | int m = 4, k = 3, n = 4; size_t size = kutacc_core_bgemm_pack_get_size('A', m, n, k); /* * Output size: 12 */ |
父主题: GEMM算子