op_ut_run工具使用

功能描述

CANN开发套件包中提供了UT测试工具：op_ut_run，支持算子在仿真环境中进行功能验证并生成dump数据文件。

工具路径

通过可执行文件“op_ut_run”进行UT测试，其功能和安装路径如下。

使用前提

• CANN组合包提供进程级环境变量设置脚本，供用户在进程中引用，以自动完成环境变量设置。执行命令参考如下，以下示例均为root或非root用户默认安装路径，请以实际安装路径为准。

  # 以root用户安装toolkit包
  . /usr/local/Ascend/ascend-toolkit/set_env.sh 
  # 以非root用户安装toolkit包
  . ${HOME}/Ascend/ascend-toolkit/set_env.sh 

• 使用此工具生成算子测试用例前，需要将要测试的算子部署到算子库中。
• 若在实现算子UT功能验证时使用到AI框架，请完成所需AI框架的安装。

  TensorFlow框架的安装请参见安装TensorFlow。

  PyTorch框架的安装请参见安装PyTorch。

使用方法

1. 请用户完成输入文件的准备工作。
   • 获取待测试算子的测试用例定义文件（test_*_impl.py），请参见生成UT测试用例定义文件。
   • 获取算子实现文件（*.cpp文件）请参见kernel侧算子实现。

2. 进入op_ut_run工具所在目录执行如下命令，参数说明请参见表2。
   $HOME/Ascend/ascend-toolkit/latest/python/site-packages/bin/op_ut_run --case_files {test case file} --case_name {case name } --data_path {test case data file} --process_num {process num} --simulator_lib_path {simulator library path} --simulator_data_path {simulator dump data files} --simulator_mode {simulator model} --ascendc_op_path {*.cpp file } --block_dim {block_dim} --soc_version {soc_version}

   表2 参数说明

   参数名称	参数描述	是否必选
   --case_files	测试用例定义文件（test_*_impl.py文件）。	是
   --case_name	测试用例文件中的用例名称。 配置为需要执行的case的名字，若需要同时运行多个case，多个case之间使用逗号分隔。 若配置为“all”，或者不配置此参数，代表执行所有case。	否
   --data_path	保存运行过程中生成的测试用例数据.bin文件所在路径，可配置为绝对路径或者相对路径，并且工具执行用户具有可读写权限。 默认值：“./data”。	否
   --process_num	UT测试的进程数目。 默认值：“1”。	否
   --simulator_lib_path	配置仿真环境运行依赖，指定配置为$HOME/Ascend/ascend-toolkit/latest/toolkit/tools/simulator目录。	是
   --simulator_data_path	仿真环境dump数据存放目录，可配置为绝对路径或者相对路径，并且工具执行用户具有可读写权限。 默认值：“./model”。	否
   --simulator_mode	运行测试用例的仿真环境，可支持配置为ca或pv。 pv：即PVModel，为功能仿真环境（Simulator Function）。 ca：即CAModel，为性能仿真环境（Simulator Performance）。	是
   --ascendc_op_path	算子实现文件（.cpp文件）路径，可配置为绝对路径或者相对路径 说明： 仅支持配置一个需要运行的算子实现文件。	是
   --block_dim	算子核函数运行所需核数。 默认值：“1”。	否
   --soc_version	配置昇腾AI处理器的类型，请根据实际版本进行选择。 说明： 可从CANN软件安装后文件存储路径的"compiler/data/platform_config"路径下查看支持的昇腾AI处理器的类型，对应“*.ini”文件的名字即为soc_version。	是

   示例：

   进入op_ut_run工具所在目录，执行如下命令:

   ./op_ut_run --case_files=xx/test_add_custom_impl.py --data_path=./data --simulator_data_path=./model --simulator_lib_path=/usr/local/Ascend/ascend-toolkit/latest/toolkit/tools/simulator --simulator_mode=ca --soc_version=Ascendxxxyy --case_name=add_custom_1 --ascendc_op_path=xx/add_custom.cpp --block_dim=8

3. 查看生成的dump数据，目录结构如下所示：

   ├──{model}      //默认目录或自定义数据存放目录
   │   └── ca     //simulator_mode配置为ca时生成此目录
   │       └── add_custom     //根据op_type生成     
   │           └── add_custom_pre_static_test_Add_auto_case_name_1     //以{op_type}_pre_static_test_{case_name}命名的目录下存放的dump数据
   │               ├── core0_xx.dump
   │               ├── core0_xx_xx.dump
   │               ├── core1_xx.dump
   │               ├── core1_xx_xx.dump
   │               ├── core10_xx.dump
   │               ├── core10_xx_xx.dump
   │   └── pv      //simulator_mode配置为pv时生成此目录
   │       └── add_custom      ///根据op_type生成 
   │           └──add_custom_pre_static_test_Add_auto_case_name_1     //以{op_type}_pre_static_test_{case_name}命名的目录下存放的dump数据
   │               ├── core0_xx.dump
   │               ├── core0_xx_xx.dump
   │               ├── core1_xx.dump
   │               ├── core1_xx_xx.dump
   │               ├── core10_xx.dump
   │               ├── core10_xx_xx.dump