项目简介
MLU-OPS测试用例生成器是一款基于Python编程语言的工具,主要用于自动生成深度学习模型的测试案例。它能生成带有随机参数的测试用例,可对深度学习模型中的各类操作(如卷积、池化、归一化等)进行测试。该工具通过解析JSON或Protocol Buffers格式的配置文件,覆盖不同的输入参数组合,助力开发者开展大规模自动化测试,进而提升开发效率与软件质量。
项目的主要特性和功能
- 灵活性:支持多种数据类型、数据布局和随机分布,可生成多样化的测试用例。
- 自动化:自动解析配置文件并生成测试用例,减少人工编写测试案例的工作量。
- 可扩展性:借助注册机制,能轻松添加新操作或修改现有操作。
- 可配置性:支持通过配置文件指定测试用例的生成方式、输出格式等。
- 跨平台:支持CUDA和PyTorch等深度学习框架,适配不同计算平台。
安装使用步骤
1. 安装依赖库
安装Python及其相关的库,如numpy、tensorflow等。
2. 获取源代码
(用户已下载本项目的源码文件)
3. 配置文件
创建或获取包含随机参数配置的JSON或Protocol Buffers文件。
4. 运行生成器
运行MLU-OPS生成器脚本,指定配置文件和输出路径。例如:
bash
python run_manual.py div --json_path=./manual_config --prototxt_path=./generated_testcases
5. 测试用例输出
生成器将生成包含随机参数的测试用例,并保存到指定的文件中。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】