功能说明

设置printf参数。

函数原型

def set_printf_params(print_workspace_size=128, single_print_length=512)

参数说明

参数名称	输入/输出	含义
print_workspace_size	输入	打印总空间大小，如果算子有多个核并行运行，则打印总空间大小包括所有核的空间之和，各个核均匀切分。单位：M Byte。范围：[1, 32768] 默认值：128, 表示128M 空间。支持的数据类型：立即数(int)
single_print_length	输入	单个print 语句占用的固定长度。如果单个print语句的空间少于这个值，也会占用这个值的空间。单位：Byte 范围：[64, print_workspace_size10241024//核数 -32]，其中“print_workspace_size”：参看print_workspace_size参数说明；“核数”和算子相关，由算子开发者指定核的数量，即多核for_range的block_num。默认值：512 Byte 支持的数据类型：立即数(int) 注意：值必须是32 字节的整数倍。

参数名称

输入/输出

含义

print_workspace_size

输入

打印总空间大小，如果算子有多个核并行运行，则打印总空间大小包括所有核的空间之和，各个核均匀切分。

单位：M Byte。

范围：[1, 32768]

默认值：128, 表示128M 空间。

支持的数据类型：立即数(int)

single_print_length

输入

单个print 语句占用的固定长度。如果单个print语句的空间少于这个值，也会占用这个值的空间。

单位：Byte

范围：[64, print_workspace_size*1024*1024//核数 -32]，其中“print_workspace_size”：参看print_workspace_size参数说明；“核数”和算子相关，由算子开发者指定核的数量，即多核for_range的block_num。

默认值：512 Byte

支持的数据类型：立即数(int)

注意：值必须是32 字节的整数倍。

支持的芯片型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

返回值

无

注意事项

set_printf_params必须在所有printf前调用，可选择不调用set_printf_params直接使用printf。
在一个算子内，set_printf_params只能调用1次，多次调用会报错。
打印总空间大小的最大大小与设备DDR内存大小有关，否则可能导致算子运行失败。
print_workspace_size*1024*1024 需要被多核for_range的block_num整除。
print_workspace_size*1024*1024 需要大于等于多核for_range的block_num * (32 + single_print_length )。

调用示例

from tbe import tik
tik_instance = tik.Tik()
# 设置整体workspace空间为1MB， 单条打印最大长度为256B
tik_instance.set_printf_params(1, 256)
scalar = tik_instance.Scalar("int8", init_value=-128)
tik_instance.printf("scalar is: %d\n", scalar)
tik_instance.BuildCCE(inputs=[], outputs=[], kernel_name="print_scalar")
tik_instance.tikdb.start_debug({})
输入
[]
输出
scalar is: -128