1 files changed, 52 insertions, 0 deletions
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