summaryrefslogtreecommitdiff
path: root/xmake/toolchains/ascendc/check.lua
diff options
context:
space:
mode:
authorwuzhenqing <[email protected]>2026-05-04 11:15:50 +0000
committerwuzhenqing <[email protected]>2026-05-04 11:18:09 +0000
commitfa807b737b4b6b3004cccafc0dfdfdd66fcdce2d (patch)
tree558dc6bc9e5749180fa97a569aab8bb6c2e54fec /xmake/toolchains/ascendc/check.lua
parent8bbbb2860fdd1910db466fe318ee8d465728ca32 (diff)
feat(ascendc): add Huawei Ascend C toolchain support
Add toolchain, language, rules, and compiler modules for the bisheng compiler driver from CANN (Compute Architecture for Neural Networks) toolkit. Supports .asc (Ascend C kernel) and .aicpu (AI-CPU operator) source kinds with binary/static/shared target types. - Toolchain: SDK auto-detection via ASCEND_HOME_PATH env or --sdk - Compiler: bisheng module inherits from clang, adds add_sourceflags for explicit sourcekind overrides - Rules: ascendc.env (default C++17), npuarchs (--npu-arch), build - Language: check_main detection, add_ascnpuarchs API - Tests: unit tests for check_main, project tests for libs/mixed Fixes #7506
Diffstat (limited to 'xmake/toolchains/ascendc/check.lua')
-rw-r--r--xmake/toolchains/ascendc/check.lua90
1 files changed, 90 insertions, 0 deletions
diff --git a/xmake/toolchains/ascendc/check.lua b/xmake/toolchains/ascendc/check.lua
new file mode 100644
index 000000000..60df001db
--- /dev/null
+++ b/xmake/toolchains/ascendc/check.lua
@@ -0,0 +1,90 @@
+--!A cross-platform build utility based on Lua
+--
+-- Licensed under the Apache License, Version 2.0 (the "License");
+-- you may not use this file except in compliance with the License.
+-- You may obtain a copy of the License at
+--
+-- http://www.apache.org/licenses/LICENSE-2.0
+--
+-- Unless required by applicable law or agreed to in writing, software
+-- distributed under the License is distributed on an "AS IS" BASIS,
+-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+-- See the License for the specific language governing permissions and
+-- limitations under the License.
+--
+-- Copyright (C) 2015-present, Xmake Open Source Community.
+--
+-- @author wuzhenqing
+-- @file check.lua
+--
+
+-- imports
+import("lib.detect.find_tool")
+
+-- map host architecture to CANN host tool directory
+function _host_archdir(arch)
+ local host_archdirs = {
+ x86_64 = "x86_64-linux"
+ , x64 = "x86_64-linux"
+ , arm64 = "aarch64-linux"
+ , aarch64 = "aarch64-linux"
+ }
+ return host_archdirs[arch]
+end
+
+-- check the ascendc toolchain
+function main(toolchain)
+ if not toolchain:is_plat("linux") then
+ return false
+ end
+
+ -- resolve sdkdir: --sdk= > ASCEND_HOME_PATH > ASCEND_TOOLKIT_HOME
+ local sdkroot = toolchain:sdkdir()
+ if not sdkroot then
+ sdkroot = os.getenv("ASCEND_HOME_PATH") or os.getenv("ASCEND_TOOLKIT_HOME")
+ end
+ if not sdkroot or not os.isdir(sdkroot) then
+ return false
+ end
+ sdkroot = path.absolute(sdkroot)
+
+ -- map host arch to CANN host directory
+ local host_arch = os.arch()
+ local host_archdir = _host_archdir(host_arch)
+ if not host_archdir then
+ return false
+ end
+
+ local hostroot = path.join(sdkroot, host_archdir)
+ if not os.isdir(hostroot) then
+ return false
+ end
+
+ -- check required executables
+ local bindir = path.join(hostroot, "bin")
+ local bisheng_bin = path.join(bindir, "bisheng")
+ local llvm_ar = path.join(bindir, "llvm-ar")
+ if not os.isexec(bisheng_bin) or not os.isexec(llvm_ar) then
+ return false
+ end
+
+ -- ensure bisheng can load its own shared libraries during version check
+ local host_libdir = path.join(hostroot, "lib64")
+ local ld_library_path = os.getenv("LD_LIBRARY_PATH") or ""
+ local envs = {
+ LD_LIBRARY_PATH = ld_library_path ~= "" and
+ (host_libdir .. path.envsep() .. ld_library_path) or host_libdir
+ }
+
+ -- use find_tool (unified interface) instead of find_bisheng directly
+ local result = find_tool("bisheng", {program = bisheng_bin, version = true, envs = envs})
+ if not result or not result.program then
+ return false
+ end
+
+ toolchain:config_set("sdkdir", sdkroot)
+ toolchain:config_set("bindir", bindir)
+ toolchain:config_set("hostroot", hostroot)
+ cprint("checking for Huawei Ascend C Toolchain (host: %s) ... ${color.success}${text.success}", host_arch)
+ return true
+end