master

文件	最后提交记录	最后更新时间
__init__.py	【重构】融合算子注册改为配置化，支持用户通过 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	3 个月前
apply_adam.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
apply_adam_w.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
confusion_transpose.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
fast_gelu.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
fusion_operator_config.yaml	【重构】融合算子注册改为配置化，支持用户通过 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	3 个月前
group_norm_silu.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
layer_norm_eval.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
linear.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
matmul_backward.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
mish.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
moe_gating_top_k_softmax.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
npu_fusion_attention.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
rms_norm.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
rotary_mul.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
scaled_mask_softmax.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
sort_v2.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
swiglu.py	更新为木兰版权头 Co-authored-by: wugengjun<wugengjun1@huawei.com>	8 个月前
template.py	【重构】融合算子注册改为配置化，支持用户通过 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	3 个月前