增加返回码

Author: hd9568

ci_scripts/hooks/pre-doc-compile.sh

@@ -121,38 +121,26 @@ else
     handle_failure
 fi
 
-echo "INFO: Validating API difference format "
-VALIDATION_OUTPUT=$(python "${APIMAPPING_ROOT}/tools/validate_api_difference_format.py" 2>&1)
-VALIDATION_EXIT_CODE=$?
+python "${APIMAPPING_ROOT}/tools/validate_api_difference_format.py"
 
-# Check if validation output contains "100.00%"
-if [[ "$VALIDATION_EXIT_CODE" -eq 0 && "$VALIDATION_OUTPUT" == *"100.00%"* ]]; then
-    echo "INFO: Validation passed successfully (100.00%)"
+# 获取上一条命令的退出状态码
+exit_code=$?
+
+# 根据退出状态码决定后续操作
+if [ $exit_code -eq 0 ]; then
+    echo "API DIFFERENCE FORMAT VALIDATE SUCCESS!"
+    # 在这里继续添加您需要执行的命令
 else
-    echo "ERROR: API difference validation failed. Expected 100.00%, got: '$VALIDATION_OUTPUT'"
+    echo "ERROR: API DIFFERENCE FORMAT VALIDATE FAILURE! error code: $exit_code" >&2
     exit 1
 fi
 
-echo "INFO: Validating PyTorch API mapping with --skip-url-check"
-VALIDATE_CMD="python ${APIMAPPING_ROOT}/tools/validate_pytorch_api_mapping.py --skip-url-check"
-VALIDATION_OUTPUT=$(eval "$VALIDATE_CMD" 2>&1)
-VALIDATION_EXIT_CODE=$?
+python "${APIMAPPING_ROOT}/tools/validate_pytorch_api_mapping.py" --skip-url-check
+exit_code=$?
 
-# Check if script execution failed
-if [ $VALIDATION_EXIT_CODE -ne 0 ]; then
-    echo "ERROR: validate_pytorch_api_mapping.py execution failed with exit code $VALIDATION_EXIT_CODE"
-    echo "ERROR: Script output:"
-    echo "$VALIDATION_OUTPUT"
-    exit 1
+if [ $exit_code -eq 0 ]; then
+    echo "PYTORCH API MAPPING VALIDATE SUCCESS!"
 else
-    # Get last line of output
-    LAST_LINE=$(echo "$VALIDATION_OUTPUT" | tail -n 1)
-
-    # Check if last line matches error string
-    if [ "$LAST_LINE" = "VALIDATE PYTORCH_API_MAPPING ERROR!" ]; then
-        echo "ERROR: Validation detected error: $LAST_LINE"
-        exit 1
-    else
-        echo "INFO: API mapping validation passed successfully"
-    fi
+    echo "ERROR: PYTORCH API MAPPING VALIDATE FAILURE! error code: $exit_code" >&2
+    exit 1
 fi

docs/guides/model_convert/convert_from_pytorch/tools/validate_api_difference_format.py

@@ -2,6 +2,7 @@
 
 import os
 import re
+import sys
 import traceback
 
 
@@ -798,12 +799,14 @@ def main():
                     print(error)
                     f.write(f"{error}\n")
         print(f"error log saved to: {error_file}")
+        validator.print_results()
+        return 1
     else:
         if os.path.exists(error_file):
             os.remove(error_file)
-
-    validator.print_results()
+        validator.print_results()
+        return 0
 
 
 if __name__ == "__main__":
-    main()
+    sys.exit(main())

docs/guides/model_convert/convert_from_pytorch/tools/validate_pytorch_api_mapping.py

@@ -3,6 +3,7 @@
 import os
 import random
 import re
+import sys
 import time
 from collections import defaultdict
 from urllib.parse import urlparse
@@ -27,6 +28,13 @@
     "composite_implement": "组合替代实现",
 }
 
+white_list = [
+    "torch.nn.Linear",
+    "torch.nn.functional.linear",
+    "torch.nn.ZeroPad1d",
+    "torch.nn.ZeroPad3d",
+]
+
 # 反向映射（中文到英文）
 REVERSE_CATEGORY_MAP = {v: k for k, v in CATEGORY_MAP.items()}
 
@@ -388,6 +396,10 @@ def check_diff_doc_consistency(categories, base_dir):
             torch_api = torch_api.replace(r"\_", "_")
             if not torch_api:
                 continue
+            if not torch_api.startswith("torch."):
+                continue
+            if torch_api in white_list:
+                continue
 
             expected_apis[category_name].add(torch_api)
 
@@ -418,6 +430,9 @@ def check_diff_doc_consistency(categories, base_dir):
                 if filename.endswith(".md"):
                     torch_api = filename[:-3]  # 去掉.md后缀
 
+                    if torch_api in white_list:
+                        continue
+
                     # 检查这个API是否在对应类别的表格中
                     api_found = False
                     for category in categories:
@@ -781,8 +796,10 @@ def main():
             f"校验完成，共发现 {total_warnings} 个警告，请查看生成的警告文件。"
         )
         if total_errors > 0:
-            print("VALIDATE PYTORCH_API_MAPPING ERROR!")
+            return 1
+
+    return 0
 
 
 if __name__ == "__main__":
-    main()
+    sys.exit(main())