diff options
| author | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-11-10 15:35:46 +0100 |
|---|---|---|
| committer | Nicolas Patry <nicolas@Nicolass-MacBook-Pro.local> | 2023-11-20 14:12:57 +0100 |
| commit | d46670f7c0a64ae3824546364b05614e49ecb70a (patch) | |
| tree | 236bbdc6377bda790f7b81a6b39f2ee52283de4b /candle-core/src/device.rs | |
| parent | f710fab02e911b31ab15af80d42923f6a56317b2 (diff) | |
| download | candle-d46670f7c0a64ae3824546364b05614e49ecb70a.tar.gz candle-d46670f7c0a64ae3824546364b05614e49ecb70a.tar.bz2 candle-d46670f7c0a64ae3824546364b05614e49ecb70a.zip | |
Tmp state.
Diffstat (limited to 'candle-core/src/device.rs')
| -rw-r--r-- | candle-core/src/device.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/device.rs b/candle-core/src/device.rs index 73eb9640..3eb7f8b7 100644 --- a/candle-core/src/device.rs +++ b/candle-core/src/device.rs @@ -8,7 +8,7 @@ use crate::{CpuStorage, DType, Result, Shape, Storage, WithDType}; pub enum DeviceLocation { Cpu, Cuda { gpu_id: usize }, - Metal, + Metal { gpu_id: usize }, } #[derive(Debug, Clone)] |