init v0.23.0

Signed-off-by: Sun Ruoxi <sunruoxi@4paradigm.com>
2026-08-27 15:11:51 +08:00
parent b582a8e7d1
commit 7f8a1b1f7a
2849 changed files with 712887 additions and 22001 deletions
--- a/csrc/cmake/scripts/util/ascendc_gen_options.py
+++ b/csrc/cmake/scripts/util/ascendc_gen_options.py
@@ -0,0 +1,77 @@
+#!/usr/bin/env python3
+# -----------------------------------------------------------------------------------------------------------
+# Copyright (c) 2025 Huawei Technologies Co., Ltd.
+# This program is free software, you can redistribute it and/or modify it under the terms and conditions of
+# CANN Open Software License Agreement Version 2.0 (the "License").
+# Please refer to the License for details. You may not use this file except in compliance with the License.
+# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
+# INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
+# See LICENSE in the root of the software repository for the full text of the License.
+# -----------------------------------------------------------------------------------------------------------
+
+import os
+import stat
+import sys
+
+
+def write_options_to_file(file_name: str, options_str: str, op_type: str, compute_unit: str, split_char: str):
+    flags = os.O_WRONLY | os.O_CREAT
+    modes = stat.S_IWUSR | stat.S_IRUSR
+    try:
+        with os.fdopen(os.open(file_name, flags, modes), "a") as fd:
+            fd.write(op_type + split_char + compute_unit + split_char + options_str + "\n")
+    except Exception as err:
+        print("write compile options config file failed")
+        raise (err)
+
+
+def gen_compile_options(compile_options_file: str, op_type: str, compute_unit: str, compile_options: list):
+    base_dir = os.path.dirname(compile_options_file)
+    opc_config_file = os.path.join(base_dir, "custom_opc_options.ini")
+    compile_opt = []
+    opc_debug_config = []
+    opc_tiling_keys = ""
+    for opts in compile_options:
+        if "oom" in opts:
+            if opts == "--oom":
+                opc_debug_config.append("oom")
+            else:
+                raise RuntimeError(f"Unknown oom option format {opts}")
+        elif "--save-temp-files" in opts:
+            opc_debug_config.append("dump_cce")
+        elif opts.startswith("--op_relocatable_kernel_binary") or opts.startswith("--op_super_kernel_options"):
+            opc_debug_config.append(opts)
+        elif "--tiling_key" in opts:
+            keys = opts.strip().split("=")[1].split(",")
+            keys_str = ";".join([key for key in keys])
+            opc_tiling_keys = keys_str
+        else:
+            compile_opt.append(opts)
+    if len(compile_opt) > 0:
+        options_str = ";".join([opt for opt in compile_opt])
+        write_options_to_file(compile_options_file, options_str, op_type, compute_unit, ",")
+    opc_config_str = ""
+    if opc_debug_config:
+        opc_config_str = "--op_debug_config=" + ";".join([opt for opt in opc_debug_config])
+    if len(opc_tiling_keys) > 0:
+        if opc_config_str != "":
+            opc_config_str += "@"
+        opc_config_str += "--tiling_key=" + opc_tiling_keys
+
+    if opc_config_str != "":
+        write_options_to_file(opc_config_file, opc_config_str, op_type, compute_unit, "@")
+
+
+if __name__ == "__main__":
+    if len(sys.argv) < 4:
+        raise RuntimeError("arguments must greater than 4")
+    compute_soc = ""
+    comp_options = []
+    for i in range(len(sys.argv) - 3):
+        if sys.argv[i + 3].upper().startswith("ASCEND"):
+            compute_soc += sys.argv[i + 3] + ";"
+        else:
+            comp_options.append(sys.argv[i + 3])
+    if compute_soc != "":
+        compute_soc = compute_soc[0:-1]
+    gen_compile_options(sys.argv[1], sys.argv[2], compute_soc, comp_options)