From 0a6e0a8c9ae056684c0edd888acb3c691f889d33 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Sun, 17 Dec 2023 19:09:08 +0100
Subject: [PATCH] Implement randn (CPU-> device)

---
 candle-core/src/device.rs | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/candle-core/src/device.rs b/candle-core/src/device.rs
index 3eb7f8b7..1e33021b 100644
--- a/candle-core/src/device.rs
+++ b/candle-core/src/device.rs
@@ -201,10 +201,9 @@ impl Device {
                     Ok(Storage::Cuda(storage))
                 }
             }
-            Device::Metal(_device) => {
-                // let storage = device.rand_uniform(shape, dtype, lo, up)?;
-                // Ok(Storage::Metal(storage))
-                crate::bail!("Metal rand_uniform not implemented")
+            Device::Metal(device) => {
+                let storage = device.rand_uniform(shape, dtype, lo, up)?;
+                Ok(Storage::Metal(storage))
             }
         }
     }