cuda: support download toolchains (#7459)

* cuda: support download toolchains

* limit plat

Author: star-hengxing

packages/c/cuda/xmake.lua

@@ -1,17 +1,24 @@
 package("cuda")
-
+    set_kind("toolchain")
     set_homepage("https://developer.nvidia.com/cuda-zone/")
     set_description("CUDA® is a parallel computing platform and programming model developed by NVIDIA for general computing on graphical processing units (GPUs).")
 
+    if is_plat("windows") then
+        add_urls("https://developer.download.nvidia.com/compute/cuda/$(version)_windows.exe", {
+            version = function (version)
+                local driver_version_map = {
+                    ["12.6.3"] = "561.17",
+                }
+                return format("%s/local_installers/cuda_%s_%s", version, version, driver_version_map[tostring(version)])
+            end})
+
+        add_versions("12.6.3", "d73e937c75aaa8114da3aff4eee96f9cae03d4b9d70a30b962ccf3c9b4d7a8e1")
+    end
+
     add_configs("utils", {description = "Enabled cuda utilities.", default = {}, type = "table"})
+    add_configs("debug", {description = "Enable debug symbols.", default = false, type = "boolean", readonly = true})
 
-    on_load(function (package)
-        import("detect.sdks.find_cuda")
-        local cuda = find_cuda()
-        if cuda then
-            package:addenv("PATH", cuda.bindir)
-        end
-    end)
+    set_policy("package.precompiled", false)
 
     on_fetch(function (package, opt)
         if opt.system then
@@ -35,3 +42,37 @@ package("cuda")
             end
         end
     end)
+
+    on_load("windows", function (package)
+        package:mark_as_pathenv("CUDA_PATH")
+        package:setenv("CUDA_PATH", ".")
+    end)
+
+    on_install("windows|x64", function(package)
+        import("lib.detect.find_tool")
+        import("lib.detect.find_directory")
+
+        if package:is_plat("windows") then
+            local z7 = assert(find_tool("7z"), "7z tool not found!")
+            os.vrunv(z7.program, {"x", "-y", package:originfile()})
+
+            -- reference: https://github.com/ScoopInstaller/Main/blob/master/bucket/cuda.json
+            local names = {"bin", "extras", "include", "lib", "libnvvp", "nvml", "nvvm", "compute-sanitizer"}
+            for _, dir in ipairs(os.dirs("*")) do
+                if dir:startswith("cuda_") or dir:startswith("lib") then
+                    for _, name in ipairs(names) do
+                        local util_dir = find_directory(name, path.join(dir, "*"))
+                        if util_dir then
+                            os.vcp(path.join(util_dir, "*"), package:installdir(name))
+                        end
+                    end
+                end
+            end
+        end
+    end)
+
+    on_test(function (package)
+        if not package:is_cross() then
+            os.vrun("nvcc -V")
+        end
+    end)