From fa72dc9bfc2ed092842d7dce1f40c768d742d650 Mon Sep 17 00:00:00 2001 From: Guillaume Lagrange Date: Fri, 20 Sep 2024 11:32:18 -0400 Subject: [PATCH] Change enum usage --- backend-comparison/src/lib.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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); }