add cuda.nodevlink rule

1 files changed, 127 insertions, 0 deletions

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)
+