Add --keep-dir option

- On linux, requires an already created directory.

README.md

@@ -17,6 +17,7 @@ Compiles CUDA code using the Visual Studio CUDA Toolkit extension on Windows and
 - `cudaPath` (String) -> custom CUDA install path.
 - `cudaGenLineInfo` (Bool) -> generates line info.
 - `cudaIntDir` (String) -> Intermediary directory for CUDA files - Windows only.
+- `cudaKeepDir` (String) -> Directory to place cudaKeep files (on Linux, requires an existing directory).
 
 The following functions are provided:
 - `detectNvccVersion()` -> try to detect the default version of nvcc on the system.

src/cuda-exported-variables.lua

@@ -77,3 +77,9 @@ premake.api.register {
     scope = "project",
     kind = "string"
 }
+
+premake.api.register {
+    name = "cudaKeepDir",
+    scope = "project",
+    kind = "string"
+}

src/premake5-cuda-nvcc.lua

@@ -83,6 +83,13 @@ function nvcc.getcxxflags(cfg)
     flags = table.join(flags, { "--maxrregcount " .. cfg.cudaMaxRegCount})
   end
 
+  if cfg.cudaKeepDir ~= nil then
+    local e = { }
+    e.cfg = cfg
+    local v = premake.detoken.expand(cfg.cudaKeepDir, e)
+    flags = table.join(flags, { "--keep-dir " .. v})
+  end
+
   return flags
 end
 

src/premake5-cuda-vs.lua

@@ -71,6 +71,13 @@ local function addCompilerProps(cfg)
     writeString('CompileOut', path.translate(v .. "/%%(Filename)%%(Extension).obj", "\\"))
   end
 
+  if cfg.cudaKeepDir ~= nil and cfg.cudaKeepDir ~= '' then
+    local e = { }
+    e.cfg = cfg
+    local v = premake.detoken.expand(cfg.cudaKeepDir, e)
+    writeString('KeepDir', path.translate(v, "\\"))
+  end
+
   -- Code Generation is useless, when you can provide it directly in the compile flags
   premake.w('  <CodeGeneration></CodeGeneration>')
   premake.w('</CudaCompile>')

test/premake5.lua

@@ -70,6 +70,7 @@ project "ExampleProjectExe"
   cudaVerbosePTXAS "On"
   cudaMaxRegCount "32"
   cudaIntDir "out/bin/cudaobj/%{cfg.buildcfg}"
+  cudaKeepDir "%{cfg.objdir}"
 
   vpaths {
     ["Import/*"] = "include/**.hpp",