diff options
| author | wuzhenqing <[email protected]> | 2026-05-04 11:15:50 +0000 |
|---|---|---|
| committer | wuzhenqing <[email protected]> | 2026-05-04 11:18:09 +0000 |
| commit | fa807b737b4b6b3004cccafc0dfdfdd66fcdce2d (patch) | |
| tree | 558dc6bc9e5749180fa97a569aab8bb6c2e54fec /xmake/toolchains | |
| parent | 8bbbb2860fdd1910db466fe318ee8d465728ca32 (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.lua | 90 | ||||
| -rw-r--r-- | xmake/toolchains/ascendc/load.lua | 52 | ||||
| -rw-r--r-- | xmake/toolchains/ascendc/xmake.lua | 46 |
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") |
