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 c42ada0ed4e0ecdab8d86bc09afbbd933ce74d93..70461b0acc211eca8b4a0cf1832fb304119c7ae2 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
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 903ff92c583db3169cac2b852c5bafebb521fefa..d1ad40fe45c0feec8a604f28bb592b0825ea085a 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