dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

DefaultDevice should be an alias of BestAvailable (#2443) 

* BestAvailable should be an alias

* Fix an old reference
This commit is contained in:
Arthur Brussee 2024-10-31 16:03:52 +00:00 committed by GitHub
parent 6d085099f8
commit 8e466d7ce1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
burn-book/src/getting-started.md
View File

@ -157,7 +157,7 @@ Tensor {
[[3.0, 4.0],
[5.0, 6.0]],
shape: [2, 2],
device: BestAvailable,
device: DefaultDevice,
backend: "wgpu",
kind: "Float",
dtype: "f32",

3
crates/burn-tensor/src/lib.rs
View File

@ -65,7 +65,7 @@ mod cube_wgpu {
WgpuDevice::IntegratedGpu(index) => DeviceId::new(1, *index as u32),
WgpuDevice::VirtualGpu(index) => DeviceId::new(2, *index as u32),
WgpuDevice::Cpu => DeviceId::new(3, 0),
WgpuDevice::BestAvailable => DeviceId::new(4, 0),
WgpuDevice::BestAvailable | WgpuDevice::DefaultDevice => DeviceId::new(4, 0),
// For an existing device, use the 64 bit wgpu device ID as the burn DeviceID.
// We're only storing 32 bits, so wrap the the 64 bit value to 32 bits. This
// might collide - but a 1 in 4 billion chance seems ok given there's only a few
@ -73,7 +73,6 @@ mod cube_wgpu {
WgpuDevice::Existing(id) => {
DeviceId::new(5, (id.inner() % (u32::MAX as u64)) as u32)
}
WgpuDevice::DefaultDevice => DeviceId::new(6, 0),
}
}
}