Add some 'cuda-if-available' helper function. (#172)

candle-core/src/device.rs

@@ -109,6 +109,14 @@ impl Device {
         }
     }
 
+    pub fn cuda_if_available(ordinal: usize) -> Result<Self> {
+        if crate::utils::cuda_is_available() {
+            Self::new_cuda(ordinal)
+        } else {
+            Ok(Self::Cpu)
+        }
+    }
+
     pub(crate) fn rand_uniform(
         &self,
         shape: &Shape,

candle-core/src/utils.rs

@@ -17,3 +17,10 @@ pub fn has_mkl() -> bool {
     #[cfg(not(feature = "mkl"))]
     return false;
 }
+
+pub fn cuda_is_available() -> bool {
+    #[cfg(feature = "cuda")]
+    return true;
+    #[cfg(not(feature = "cuda"))]
+    return false;
+}