diff --git a/backend-comparison/src/lib.rs b/backend-comparison/src/lib.rs index 0f371fcc0..4263f6809 100644 --- a/backend-comparison/src/lib.rs +++ b/backend-comparison/src/lib.rs @@ -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::(&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::(&device, feature_name, url, token); }