Refine CUDA support.

Instead of counting all source files we count only .cu files, ones with device code that actually require the special treatment.
rust-lang · Nov 22, 2022 · dbda5d5 · dbda5d5
1 parent 2cb2792
commit dbda5d5
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -1367,7 +1367,14 @@ impl Build {
         if !msvc || !is_asm || !is_arm {
             cmd.arg("-c");
         }
-        if self.cuda && self.files.len() > 1 {
+        if self.cuda
+            && self
+                .files
+                .iter()
+                .filter(|file| file.extension() == Some(OsStr::new("cu")))
+                .count()
+                > 1
+        {
             cmd.arg("--device-c");
         }
         if is_asm {
@@ -2035,7 +2042,14 @@ impl Build {
             self.assemble_progressive(dst, chunk)?;
         }
 
-        if self.cuda {
+        if self.cuda
+            && self
+                .files
+                .iter()
+                .filter(|file| file.extension() == Some(OsStr::new("cu")))
+                .count()
+                > 0
+        {
             // Link the device-side code and add it to the target library,
             // so that non-CUDA linker can link the final binary.