1 files changed, 15 insertions, 0 deletions
diff --git a/candle-core/examples/cuda_basics.rs b/candle-core/examples/cuda_basics.rs
new file mode 100644
index 00000000..aeee541a
--- /dev/null
+++ b/candle-core/examples/cuda_basics.rs
@@ -0,0 +1,15 @@
+#[cfg(feature = "mkl")]
+extern crate intel_mkl_src;
+
+use anyhow::Result;
+use candle::{Device, Tensor};
+
+fn main() -> Result<()> {
+    let device = Device::new_cuda(0)?;
+    let t = Tensor::new(&[[1f32, 2., 3., 4.2]], &device)?;
+    let sum = t.sum(&[0])?;
+    println!("{sum}");
+    let sum = t.sum(&[1])?;
+    println!("{sum}");
+    Ok(())
+}