PyTorch GPU2Ascend提供脚本分析功能,帮助用户在执行迁移操作前,分析PyTorch训练脚本的算子和API支持情况,并输出训练脚本中API精度和性能调优的专家建议。
参数 |
参数说明 |
---|---|
PyTorch Version |
待分析脚本的PyTorch版本。目前支持1.11.0、2.0.1、2.1.0。 必选。默认为1.11.0。 |
Input Path |
待分析脚本文件所在目录。单击文件夹图标选择目录。 必选。 |
Output Path |
分析结果文件输出路径。单击文件夹图标选择目录。 必选。 |
完成后,Output Path输出目录下查看结果文件。
├── xxx_analysis // 分析结果输出目录 │ ├── cuda_op_list.csv //cuda算子列表 │ ├── unknown_api.csv //支持情况存疑的API列表 │ ├── unsupported_api.csv //不支持的API列表 │ ├── api_precision_advice.csv //API精度调优的专家建议 │ ├── api_performance_advice.csv //API性能调优的专家建议 │ ├── pytorch_analysis.txt // 分析过程日志
对于当前框架不支持的API,可以在昇腾开源社区寻求帮助。