DumpTensor

函数功能

基于算子工程开发的算子，可以使用该接口Dump指定Tensor的内容。同时支持打印自定义的附加信息（仅支持uint32_t 数据类型的信息），比如打印当前行号等。

具体的使用方法如下：

1. 增加算子工程编译选项-DASCENDC_DUMP：修改算子工程op_kernel目录下的CMakeLists.txt文件，首行增加编译选项，打开DUMP开关，样例如下：

   add_ops_compile_options(ALL OPTIONS -DASCENDC_DUMP)

2. 在算子kernel侧实现代码中需要打印Tensor数据的地方调用DumpTensor接口打印相关内容。

   DumpTensor(srcLocal,5, dataLen);

Dump时，每个block核的dump信息前会增加对应信息头DumpHead（32字节大小），用于记录核号和资源使用信息；每次Dump的Tensor数据前也会添加信息头DumpTensorHead（32字节大小），用于记录Tensor的相关信息。如下图所示，展示了多核打印场景下的打印信息结构。

DumpHead的具体信息如下：

• block_id：当前运行的核号；
• total_block_num：此次dump的核数；
• block_remain_len：当前核剩余可用的dump的空间；
• block_initial_space：当前核初始分配的dump空间；
• magic：内存校验魔术字。

DumpTensorHead的具体信息如下：

• desc：用户自定义附加信息；
• addr：Tensor的地址；
• data_type：Tensor的数据类型;
• position：表示Tensor所在的物理存储位置，当前仅支持Unified Buffer/L1 Buffer/L0C Buffer/Global Memory。

打印结果的样例如下：

DumpHead: block_id=0, total_block_num=16, block_remain_len=1048448, block_initial_space=1048576, magic=5aa5bccd
DumpTensor: desc=5, addr=0, data_type=DT_FLOAT16, position=UB
[40, 82, 60, 11, 24, 55, 52, 60, 31, 86, 53, 61, 47, 54, 34, 62, 84, 29, 48, 95, 16, 0, 20, 77, 3, 55, 69, 73, 75, 40, 35, 13]
DumpHead: block_id=1, total_block_num=16, block_remain_len=1048448, block_initial_space=1048576, magic=5aa5bccd
DumpTensor: desc=5, addr=0, data_type=DT_FLOAT16, position=UB
[58, 84, 22, 54, 41, 93, 1, 45, 50, 9, 72, 81, 23, 96, 86, 45, 36, 9, 36, 34, 78, 7, 2, 29, 47, 26, 13, 24, 27, 55, 90, 5]
...
DumpHead: block_id=7, total_block_num=16, block_remain_len=1048448, block_initial_space=1048576, magic=5aa5bccd
DumpTensor: desc=5, addr=0, data_type=DT_FLOAT16, position=UB
[28, 27, 79, 39, 86, 5, 23, 97, 89, 5, 65, 69, 59, 13, 49, 2, 34, 6, 52, 38, 4, 90, 11, 11, 61, 50, 71, 98, 19, 54, 54, 99]

函数原型

void DumpTensor(LocalTensor<T> &tensor, uint32_t desc, uint32_t dumpSize)

void DumpTensor(GlobalTensor<T>& tensor, uint32_t desc, uint32_t dumpSize)

参数说明

返回值

无

支持的型号

Atlas A2训练系列产品

Atlas推理系列产品AI Core

约束说明

• 该功能仅在如下场景支持：
  • 基于Kernel Launch算子工程，通过基础调用（Kernel Launch）方式调用算子。
  • 通过单算子API执行的方式开发单算子调用应用。
  • 间接调用单算子API(aclnnxxx)接口：Pytorch框架单算子直调的场景。
• 当前仅支持打印存储位置为Unified Buffer/L1 Buffer/L0C Buffer/Global Memory的Tensor信息。
• 操作数地址偏移对齐要求请参见通用约束。
• 待dump的元素总长度需要32Byte对齐。
• 程序中调用printf使用的空间+调用DumpTensor接口使用的空间+框架dump功能所使用的空间，每个核上不可超过1M。请开发者自行控制待打印的内容数据量，超出则不会打印。

调用示例

DumpTensor(srcLocal,5, dataLen);