master

ascend-robot【重构】融合算子注册改为配置化，支持用户通过 YAML 添加自定义融合算子

1f00e639创建于 1 天前历史提交

文件	最后提交记录	最后更新时间
mindspore_accuracy_checker_instruct.md	docs:资料优化 Co-authored-by: cai-weiwei1989<734267852@qq.com> # message auto-generated for no-merge-commit merge: !856 merge master into master docs:资料优化 Created-by: cai-weiwei1989 Commit-by: cai-weiwei1989 Merged-by: ascend-robot Description: docs:资料优化 1、目录结构根据新要求修改，对应更新全部文档链接 2、拼写错误、用词不当等低错修改以及描述优化等 3、资料中涉及torch_npu的**非代码相关**字段统一改为**TorchNPU**；Ascend Extension for PyTorch统一改为Ascend for PyTorch 4、docs/zh/quick_start/pytorch_quick_start.md补充使用conda安装环境说明 See merge request: Ascend/msprobe!856	3 天前
pytorch_accuracy_checker_instruct.md	【重构】融合算子注册改为配置化，支持用户通过 YAML 添加自定义融合算子 Co-authored-by: mnhdxnh<947098055@qq.com> # message auto-generated for no-merge-commit merge: !859 merge master into master 【重构】融合算子注册改为配置化，支持用户通过 YAML 添加自定义融合算子 Created-by: gcw_De30XO7A Commit-by: mnhdxnh Merged-by: ascend-robot Description: 【重构】融合算子预检改为配置化注册，支持用户通过 YAML 添加自定义融合算子  运行代码得到 dump.json import os os.environ['ASCEND_RT_VISIBLE_DEVICES'] = '7' import json, torch, torch_npu from msprobe.pytorch import PrecisionDebugger dump_dir = '/tmp/msprobe_test/dump_gelu_mul' os.makedirs(dump_dir, exist_ok=True) config = {"task": "statistics", "level": "L1", "dump_path": dump_dir, "list": ["npu_gelu_mul"]} with open('/tmp/msprobe_test/cfg_gelu_mul.json', 'w') as f: json.dump(config, f) debugger = PrecisionDebugger(config_path='/tmp/msprobe_test/cfg_gelu_mul.json') debugger.start() x = torch.randn(2, 4, 16).npu() out = torch_npu.npu_gelu_mul(x) print('NPU output:', out[0, :4]) debugger.stop() print('dump ok') 预检msprobe acc_check -api_info ./dump2.json -o ./ -d 7   配置之前：  会直接跳过 See merge request: Ascend/msprobe!859	1 天前