更新时间:2021-03-18 GMT+08:00
分享

构造输入数据文件

运行单算子需要构造算子输入数据,并以二进制格式保存到算子工程下。

  1. 以DDK安装用户进入自定义算子工程的算子代码目录下。

    cd $HOME/tools/projects/customop_te/operator/

  2. 执行样例数据生成脚本,生成reduction算子的样例数据文件。

    python data_gen.py reduction

    在算子工程中会生成如图1所示数据文件。

    图1 样例数据生成

    数据文件说明如表1所示。

    表1 数据文件说明

    数据文件

    说明

    Reduction_input_2_3_4_sum_axis_1.data

    reduction算子的二进制格式输入数据文件。

    Reduction_input_2_3_4_sum_axis_1.txt

    将reduction算子的二进制格式文件以txt文件的方式显示出来,方便用户查看结果。

    Reduction_output_2_3_4_sum_axis_1.data

    reduction算子二进制格式输出数据验证文件,用于验证算子运行后的输出结果是否正确。

    Reduction_output_2_3_4_sum_axis_1.txt

    将reduction算子的二进制格式输出数据文件以txt文件的方式显示出来,方便用户查看结果。

分享:

    相关文档

    相关产品