Implement randn (CPU-> device)

2023-12-17 19:09:08 +01:00 · 2023-12-17 19:09:08 +01:00 · 0a6e0a8c9a
parent 972903021c
commit 0a6e0a8c9a
1 changed files with 3 additions and 4 deletions
--- 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))
            }
        }
    }