Refine CUDA support.

Instead of counting all source files we count only .cu files, ones with
device code that actually require the special treatment.

Author: dot-asm

src/lib.rs

@@ -1369,7 +1369,7 @@ impl Build {
         if !msvc || !is_asm || !is_arm {
             cmd.arg("-c");
         }
-        if self.cuda && self.files.len() > 1 {
+        if self.cuda && self.cuda_file_count() > 1 {
             cmd.arg("--device-c");
         }
         if is_asm {
@@ -2037,7 +2037,7 @@ impl Build {
             self.assemble_progressive(dst, chunk)?;
         }
 
-        if self.cuda {
+        if self.cuda && self.cuda_file_count() > 0 {
             // Link the device-side code and add it to the target library,
             // so that non-CUDA linker can link the final binary.
 
@@ -3012,6 +3012,13 @@ impl Build {
         cache.insert(sdk.into(), ret.clone());
         Ok(ret)
     }
+
+    fn cuda_file_count(&self) -> usize {
+        self.files
+            .iter()
+            .filter(|file| file.extension() == Some(OsStr::new("cu")))
+            .count()
+    }
 }
 
 impl Default for Build {