鲲鹏社区首页
中文
注册
开发者
我要评分
获取效率
正确性
完整性
易理解
在线提单
论坛求助

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矩阵。

  • 如果identifier= 'A',返回pack 矩阵A所需字节数。
  • 如果identifier= 'B',返回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
     */