add --generate-line-info ptxas flag

trxcllnt · Aug 13, 2024 · 7edbeb4 · 7edbeb4
1 parent 3aee341
commit 7edbeb4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/compiler/cicc.rs b/src/compiler/cicc.rs
@@ -116,7 +116,8 @@ where
         match arg {
             Ok(arg) => {
                 let args = match arg.get_data() {
-                    Some(PassThrough(_)) => {
+                    Some(PassThroughFlag) |
+                        Some(PassThrough(_)) => {
                         take_next = false;
                         &mut common_args
                     },
@@ -279,6 +280,7 @@ ArgData! { pub
     Output(PathBuf),
     ExtraInput(PathBuf),
     ExtraOutput(PathBuf),
+    PassThroughFlag,
     PassThrough(OsString),
     Unhashed(OsString),
 }

diff --git a/src/compiler/ptxas.rs b/src/compiler/ptxas.rs
@@ -99,6 +99,7 @@ impl CCompilerImpl for Ptxas {
 use cicc::ArgData::*;
 
 counted_array!(pub static ARGS: [ArgInfo<cicc::ArgData>; _] = [
+    flag!("--generate-line-info", PassThroughFlag),
     take_arg!("-arch", OsString, CanBeSeparated, PassThrough),
     take_arg!("-m", OsString, CanBeSeparated, PassThrough),
     take_arg!("-o", PathBuf, Separated, Output),