summaryrefslogtreecommitdiff
path: root/xmake/toolchains
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
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')
-rw-r--r--xmake/toolchains/ascendc/check.lua90
-rw-r--r--xmake/toolchains/ascendc/load.lua52
-rw-r--r--xmake/toolchains/ascendc/xmake.lua46
3 files changed, 188 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
diff --git a/xmake/toolchains/ascendc/load.lua b/xmake/toolchains/ascendc/load.lua
new file mode 100644
index 000000000..696541c9d
--- /dev/null
+++ b/xmake/toolchains/ascendc/load.lua
@@ -0,0 +1,52 @@
+--!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 load.lua
+--
+
+function main(toolchain)
+ local sdkdir = toolchain:config("sdkdir")
+ local hostroot = toolchain:config("hostroot")
+ if not sdkdir or not hostroot then
+ raise("ascendc toolchain not checked")
+ end
+
+ -- add run environments
+ toolchain:add("runenvs", "LD_LIBRARY_PATH", path.join(hostroot, "lib64"))
+
+ -- add toolchain-level search paths (picked up by language nameflags)
+ toolchain:add("includedirs", path.join(hostroot, "include"))
+ toolchain:add("includedirs", path.join(hostroot, "asc", "include"))
+ toolchain:add("includedirs", path.join(hostroot, "asc", "include", "aicpu_api"))
+ toolchain:add("includedirs", path.join(sdkdir, "include"))
+ toolchain:add("linkdirs", path.join(hostroot, "lib64"))
+ toolchain:add("linkdirs", path.join(hostroot, "devlib"))
+ toolchain:add("linkdirs", path.join(sdkdir, "lib64"))
+ toolchain:add("rpathdirs", path.join(hostroot, "lib64"))
+ toolchain:add("rpathdirs", path.join(hostroot, "devlib"))
+ toolchain:add("rpathdirs", path.join(sdkdir, "lib64"))
+ toolchain:add("syslinks", "ascendcl", "acl_rt", "dl")
+
+ -- aicpu-specific compiler flags (SDK-path-dependent)
+ toolchain:add("aicpuflags", "-fvisibility-inlines-hidden")
+ toolchain:add("aicpuflags", "-D_GLIBCXX_USE_CXX11_ABI=0")
+ toolchain:add("aicpuflags", "-D_GNU_SOURCE")
+ toolchain:add("aicpuflags", "--cce-aicpu-L" .. path.join(hostroot, "lib64", "device", "lib64"))
+ toolchain:add("aicpuflags", "--cce-aicpu-laicpu_api")
+ toolchain:add("aicpuflags", "--cce-aicpu-toolkit-path=" .. path.join(sdkdir, "toolkit", "toolchain", "hcc", "bin"))
+ toolchain:add("aicpuflags", "--cce-aicpu-sysroot=" .. path.join(sdkdir, "toolkit", "toolchain", "hcc", "sysroot"))
+end
diff --git a/xmake/toolchains/ascendc/xmake.lua b/xmake/toolchains/ascendc/xmake.lua
new file mode 100644
index 000000000..6071a6840
--- /dev/null
+++ b/xmake/toolchains/ascendc/xmake.lua
@@ -0,0 +1,46 @@
+--!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 xmake.lua
+--
+
+-- Huawei Ascend C / bisheng toolchain
+--
+-- @code
+-- target("example")
+-- set_toolchains("ascendc")
+-- add_ascnpuarchs("dav-2201")
+-- @endcode
+--
+toolchain("ascendc")
+
+ -- set homepage
+ set_homepage("https://www.hiascend.com/")
+ set_description("Huawei Ascend C (bisheng compiler driver)")
+
+ -- set toolset
+ set_toolset("asc", "bisheng")
+ set_toolset("aicpu", "bisheng")
+ set_toolset("ascld", "bisheng")
+ set_toolset("ascsh", "bisheng")
+ set_toolset("ar", "llvm-ar", "ar")
+
+ -- check toolchain
+ on_check("check")
+
+ -- on load
+ on_load("load")