From a42d2ff6c4b54bc63a4126c9f61ad83b7cb0f18a Mon Sep 17 00:00:00 2001 From: ruki Date: Wed, 12 Jun 2019 23:00:12 +0800 Subject: add cuda.nodevlink rule --- xmake/rules/cuda/device_link/xmake.lua | 116 ------------------------------ xmake/rules/cuda/devlink/xmake.lua | 127 +++++++++++++++++++++++++++++++++ xmake/rules/cuda/xmake.lua | 2 +- 3 files changed, 128 insertions(+), 117 deletions(-) delete mode 100644 xmake/rules/cuda/device_link/xmake.lua create mode 100644 xmake/rules/cuda/devlink/xmake.lua diff --git a/xmake/rules/cuda/device_link/xmake.lua b/xmake/rules/cuda/device_link/xmake.lua deleted file mode 100644 index 358c42d93..000000000 --- a/xmake/rules/cuda/device_link/xmake.lua +++ /dev/null @@ -1,116 +0,0 @@ ---!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 - 2019, TBOOX Open Source Group. --- --- @author ruki --- @file xmake.lua --- - --- define rule: device-link -rule("cuda.device_link") - - -- add rule: cuda environment - add_deps("cuda.env") - - -- clean files - after_clean(function (target) - os.tryrm(target:objectfile(path.join(".cuda", "devlink", target:basename() .. "_gpucode.cu"))) - os.tryrm(target:dependfile(targetfile)) - end) - - -- @see https://devblogs.nvidia.com/separate-compilation-linking-cuda-device-code/ - before_link(function (target, opt) - - -- only for binary/shared - local targetkind = target:targetkind() - if targetkind ~= "binary" and targetkind ~= "shared" then - return - end - - -- imports - import("core.base.option") - import("core.theme.theme") - import("core.project.config") - import("core.project.depend") - import("core.tool.linker") - - -- load linker instance - local linkinst = linker.load("gpucode", "cu", {target = target}) - - -- init culdflags - local culdflags = {"-dlink"} - if targetkind == "shared" then - table.insert(culdflags, "-shared") - end - - -- get link flags - local linkflags = linkinst:linkflags({target = target, configs = {force = {culdflags = culdflags}}}) - - -- get target file - local targetfile = target:objectfile(path.join(".cuda", "devlink", target:basename() .. "_gpucode.cu")) - - -- get object files - local objectfiles = nil - for sourcekind, sourcebatch in pairs(target:sourcebatches()) do - if sourcekind == "cu" then - objectfiles = sourcebatch.objectfiles - end - end - if not objectfiles then - return - end - - -- insert gpucode.o to the object files - table.insert(target:objectfiles(), targetfile) - - -- load dependent info - local dependfile = target:dependfile(targetfile) - local dependinfo = option.get("rebuild") and {} or (depend.load(dependfile) or {}) - - -- need build this target? - local depfiles = objectfiles - local depvalues = {linkinst:program(), linkflags} - if not depend.is_changed(dependinfo, {lastmtime = os.mtime(target:targetfile()), values = depvalues, files = depfiles}) then - return - end - - -- is verbose? - local verbose = option.get("verbose") - - -- trace progress info - cprintf("${color.build.progress}" .. theme.get("text.build.progress_format") .. ":${clear} ", opt.progress) - if verbose then - cprint("${dim color.build.target}devlinking.$(mode) %s", path.filename(targetfile)) - else - cprint("${color.build.target}devlinking.$(mode) %s", path.filename(targetfile)) - end - - -- trace verbose info - if verbose then - print(linkinst:linkcmd(objectfiles, targetfile, {linkflags = linkflags})) - end - - -- flush io buffer to update progress info - io.flush() - - -- link it - assert(linkinst:link(objectfiles, targetfile, {linkflags = linkflags})) - - -- update files and values to the dependent file - dependinfo.files = depfiles - dependinfo.values = depvalues - depend.save(dependinfo, dependfile) - end) - diff --git a/xmake/rules/cuda/devlink/xmake.lua b/xmake/rules/cuda/devlink/xmake.lua new file mode 100644 index 000000000..f3ae08bc2 --- /dev/null +++ b/xmake/rules/cuda/devlink/xmake.lua @@ -0,0 +1,127 @@ +--!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 - 2019, TBOOX Open Source Group. +-- +-- @author ruki +-- @file xmake.lua +-- + +-- define rule: disable the default device-link +rule("cuda.nodevlink") + before_load(function (target) + target:data_set("cuda.nodevlink", true) + end) + +-- define rule: device-link (default) +rule("cuda.devlink") + + -- add rule: cuda environment + add_deps("cuda.env") + + -- clean files + after_clean(function (target) + os.tryrm(target:objectfile(path.join(".cuda", "devlink", target:basename() .. "_gpucode.cu"))) + os.tryrm(target:dependfile(targetfile)) + end) + + -- @see https://devblogs.nvidia.com/separate-compilation-linking-cuda-device-code/ + before_link(function (target, opt) + + -- disable devlink? + if target:data("cuda.nodevlink") then + return + end + + -- only for binary/shared + local targetkind = target:targetkind() + if targetkind ~= "binary" and targetkind ~= "shared" then + return + end + + -- imports + import("core.base.option") + import("core.theme.theme") + import("core.project.config") + import("core.project.depend") + import("core.tool.linker") + + -- load linker instance + local linkinst = linker.load("gpucode", "cu", {target = target}) + + -- init culdflags + local culdflags = {"-dlink"} + if targetkind == "shared" then + table.insert(culdflags, "-shared") + end + + -- get link flags + local linkflags = linkinst:linkflags({target = target, configs = {force = {culdflags = culdflags}}}) + + -- get target file + local targetfile = target:objectfile(path.join(".cuda", "devlink", target:basename() .. "_gpucode.cu")) + + -- get object files + local objectfiles = nil + for sourcekind, sourcebatch in pairs(target:sourcebatches()) do + if sourcekind == "cu" then + objectfiles = sourcebatch.objectfiles + end + end + if not objectfiles then + return + end + + -- insert gpucode.o to the object files + table.insert(target:objectfiles(), targetfile) + + -- load dependent info + local dependfile = target:dependfile(targetfile) + local dependinfo = option.get("rebuild") and {} or (depend.load(dependfile) or {}) + + -- need build this target? + local depfiles = objectfiles + local depvalues = {linkinst:program(), linkflags} + if not depend.is_changed(dependinfo, {lastmtime = os.mtime(target:targetfile()), values = depvalues, files = depfiles}) then + return + end + + -- is verbose? + local verbose = option.get("verbose") + + -- trace progress info + cprintf("${color.build.progress}" .. theme.get("text.build.progress_format") .. ":${clear} ", opt.progress) + if verbose then + cprint("${dim color.build.target}devlinking.$(mode) %s", path.filename(targetfile)) + else + cprint("${color.build.target}devlinking.$(mode) %s", path.filename(targetfile)) + end + + -- trace verbose info + if verbose then + print(linkinst:linkcmd(objectfiles, targetfile, {linkflags = linkflags})) + end + + -- flush io buffer to update progress info + io.flush() + + -- link it + assert(linkinst:link(objectfiles, targetfile, {linkflags = linkflags})) + + -- update files and values to the dependent file + dependinfo.files = depfiles + dependinfo.values = depvalues + depend.save(dependinfo, dependfile) + end) + diff --git a/xmake/rules/cuda/xmake.lua b/xmake/rules/cuda/xmake.lua index 0ec1ddaee..09e4c3fbe 100644 --- a/xmake/rules/cuda/xmake.lua +++ b/xmake/rules/cuda/xmake.lua @@ -22,5 +22,5 @@ rule("cuda") -- add rules - add_deps("cuda.device_link", "cuda.gencodes") + add_deps("cuda.devlink", "cuda.gencodes") -- cgit v1.3.1