From 0fbf1b7beb9818421ce1bdcda09c5dd14a552fd8 Mon Sep 17 00:00:00 2001 From: jiangchangting1 Date: Sat, 18 Nov 2023 01:20:01 +0000 Subject: [PATCH 1/2] update debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/hook_module/wrap_torch.py. Signed-off-by: jiangchangting1 --- .../src/python/ptdbg_ascend/hook_module/wrap_torch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/hook_module/wrap_torch.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/hook_module/wrap_torch.py index 903ff92c58..d1ad40fe45 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/hook_module/wrap_torch.py +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/hook_module/wrap_torch.py @@ -48,7 +48,7 @@ class TorchOPTemplate(HOOKModule): super().__init__(hook) def input_param_need_adapt(self): - special_op_list = ["broadcast_tensors"] + special_op_list = ["broadcast_tensors", "block_diag"] for item in special_op_list: if item in self.op_name_: return True -- Gitee From bc5c3464c56745f707348b2f8939c36132dd4cb4 Mon Sep 17 00:00:00 2001 From: jiangchangting1 Date: Sat, 18 Nov 2023 06:29:36 +0000 Subject: [PATCH 2/2] update debug/accuracy_tools/api_accuracy_checker/hook_module/wrap_torch.py. Signed-off-by: jiangchangting1 --- .../api_accuracy_checker/hook_module/wrap_torch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/debug/accuracy_tools/api_accuracy_checker/hook_module/wrap_torch.py b/debug/accuracy_tools/api_accuracy_checker/hook_module/wrap_torch.py index c42ada0ed4..70461b0acc 100644 --- a/debug/accuracy_tools/api_accuracy_checker/hook_module/wrap_torch.py +++ b/debug/accuracy_tools/api_accuracy_checker/hook_module/wrap_torch.py @@ -50,7 +50,7 @@ class TorchOPTemplate(HOOKModule): super().__init__(hook) def input_param_need_adapt(self): - special_op_list = ["broadcast_tensors"] + special_op_list = ["broadcast_tensors", "block_diag"] for item in special_op_list: if item in self.op_name_: return True -- Gitee