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

Change enum usage

This commit is contained in:
Guillaume Lagrange 2024-09-20 11:32:18 -04:00
parent d58770efde
commit fa72dc9bfc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend-comparison/src/lib.rs
View File

@ -114,7 +114,7 @@ macro_rules! bench_on_backend {
use burn::backend::candle::CandleDevice;
use burn::backend::Candle;
let device = CandleDevice::Cuda(0);
let device = CandleDevice::cuda(0);
bench::<Candle>(&device, feature_name, url, token);
}
@ -123,7 +123,7 @@ macro_rules! bench_on_backend {
use burn::backend::candle::CandleDevice;
use burn::backend::Candle;
let device = CandleDevice::Metal(0);
let device = CandleDevice::metal(0);
bench::<Candle>(&device, feature_name, url, token);
}