Merge pull request #430 from OpportunityLiu/dev

支持 nvcc -gencode 选项

.gitignore

@@ -41,3 +41,4 @@ core/.config.mak
 winenv
 xmake-ppa
 !xmake/actions/build
+.vscode/*

tests/projects/cuda/console/xmake.lua

@@ -1,3 +1,6 @@
+
+includes('add_cugencodes.lua')
+
 -- define target
 target("cuda_console")
 
@@ -11,14 +14,9 @@ target("cuda_console")
     add_files("src/*.cu")
 
     -- generate SASS code for each SM architecture
-    for _, sm in ipairs({"30", "35", "37", "50", "52", "60", "61", "70"}) do
-        add_cuflags("-gencode arch=compute_" .. sm .. ",code=sm_" .. sm)
-        add_ldflags("-gencode arch=compute_" .. sm .. ",code=sm_" .. sm)
-    end
+    add_cugencodes("sm_30", "sm_35", "sm_37", "sm_50", "sm_52", "sm_60", "sm_61", "sm_70")
 
     -- generate PTX code from the highest SM architecture to guarantee forward-compatibility
-    sm = "70"
-    add_cuflags("-gencode arch=compute_" .. sm .. ",code=compute_" .. sm)
-    add_ldflags("-gencode arch=compute_" .. sm .. ",code=compute_" .. sm)
-
+    add_cugencodes("compute_70")
+
 

xmake/includes/add_cugencodes.lua

@@ -0,0 +1,148 @@
+--!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      OpportunityLiu
+-- @file        add_cugencodes.lua
+--
+
+-- add cuda `-gencode` flags to target
+--
+-- the gpu arch format syntax
+-- - compute_xx                   --> `-gencode arch=compute_xx,code=compute_xx`
+-- - sm_xx                        --> `-gencode arch=compute_xx,code=sm_xx`
+-- - sm_xx,sm_yy                  --> `-gencode arch=compute_xx,code=[sm_xx,sm_yy]`
+-- - compute_xx,sm_yy             --> `-gencode arch=compute_xx,code=sm_yy`
+-- - compute_xx,sm_yy,sm_zz       --> `-gencode arch=compute_xx,code=[sm_yy,sm_zz]`
+-- - native                       --> match the fastest cuda device on current host,
+--                                    eg. for a Tesla P100, `-gencode arch=compute_60,code=sm_60` will be added,
+--                                    if no available device is found, no `-gencode` flags will be added
+--                                    @seealso xmake/modules/lib/detect/find_cudadevices
+--
+-- e.g.
+-- includes("add_cugencodes.lua")
+-- target("test")
+--    set_kind("binary")
+--    add_files("src/*.cu")
+--    add_cugencodes("native", "compute_50,sm_50", "compute_70")
+--
+
+
+-- define rule
+rule("cuda.add_gencodes")
+    before_load(function (target)
+
+        local function set (list)
+            local result = {}
+            for _, l in ipairs(list) do result[l] = true end
+            return result
+        end
+
+        -- sm_20 and compute_20 is supported until CUDA 8
+        local knownVArchs = set { 20, 30, 32, 35, 37, 50, 52, 53, 60, 61, 62, 70, 72, 75, }
+        local knownRArchs = set { 20, 30, 32, 35, 37, 50, 52, 53, 60, 61, 62, 70, 72, 75, }
+
+        local function nf_cugencode(archs)
+            if type(archs) ~= 'string' then
+                return nil
+            end
+
+            archs = archs:trim():lower()
+
+            if archs == 'native' then
+                import("lib.detect.find_cudadevices")
+                local device = find_cudadevices({ skip_compute_mode_prohibited = true, order_by_flops = true })[1]
+                if device then
+                    return nf_cugencode('sm_' .. device.major .. device.minor)
+                end
+                return nil
+            end
+
+            local vArch = nil
+            local rArchs = {}
+
+            local function parse_arch(value, prefix, knowList)
+                if not value:startswith(prefix) then
+                    return nil
+                end
+                local arch = tonumber(value:sub(#prefix + 1)) or tonumber(value:sub(#prefix + 2))
+                if arch == nil then
+                    raise("Unknown architecture: " .. value)
+                end
+                if not knowList[arch] then
+                    if arch <= table.maxn(knowList) then
+                        raise("Unknown architecture: " .. prefix .. "_" .. arch)
+                    else
+                        utils.warning("Unknown architecture: " .. prefix .. "_" .. arch)
+                    end
+                end
+                return arch
+            end
+
+            for _, v in ipairs(archs:split(',')) do
+                local arch = v:trim()
+                local tempRArch = parse_arch(arch, 'sm', knownRArchs)
+                if tempRArch then
+                    table.insert(rArchs, tempRArch)
+                end
+
+                local tempVArch = parse_arch(arch, 'compute', knownVArchs)
+                if tempVArch then
+                    if vArch ~= nil then
+                        raise("More than one virtual architecture is defined in one gpu gencode option: compute_" .. vArch .. " and compute_" .. tempVArch)
+                    end
+                    vArch = tempVArch
+                end
+
+                if not (tempRArch or tempVArch) then
+                    raise("Unknown architecture: " .. arch)
+                end
+            end
+
+            if vArch == nil and #rArchs == 0 then
+                return nil
+            end
+
+            if #rArchs == 0 then
+                return '-gencode arch=compute_' .. vArch .. ',code=compute_' .. vArch
+            end
+
+            rArchs = table.unique(rArchs)
+            vArch = vArch or math.min(unpack(rArchs))
+
+            if #rArchs == 1 then
+                return '-gencode arch=compute_' .. vArch .. ',code=sm_' .. rArchs[1]
+            else
+                return '-gencode arch=compute_' .. vArch .. ',code=[sm_' .. table.concat(rArchs, ',sm_') .. ']'
+            end
+        end
+
+        for _, v in ipairs(target:values("cuda.gencode")) do
+            local flag = nf_cugencode(v)
+            if flag then
+                target:add('cuflags', flag)
+                target:add('ldflags', flag)
+            end
+        end
+    end)
+rule_end()
+
+-- add cuda gencode to target
+function add_cugencodes(...)
+    -- apply rule
+    add_rules("cuda.add_gencodes")
+    add_values("cuda.gencode", ...)
+end
+