huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 12:20:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Avoid some unnecessary returns.

Browse Source
This commit is contained in:
Laurent
2024-01-08 07:19:59 +01:00
parent ad075a5f7e
commit fb05af4c42
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
candle-core/benches/utils.rs
View File

@ -25,13 +25,13 @@ impl BenchDevice for Device {
}
pub(crate) fn device() -> Result<Device> {
return if cfg!(feature = "metal") {
if cfg!(feature = "metal") {
Device::new_metal(0)
} else if cfg!(feature = "cuda") {
Device::new_cuda(0)
} else {
Ok(Device::Cpu)
};
}
}
pub(crate) fn bench_name<S: Into<String>>(name: S) -> String {
@ -39,7 +39,7 @@ pub(crate) fn bench_name<S: Into<String>>(name: S) -> String {
}
const fn device_variant() -> &'static str {
return if cfg!(feature = "metal") {
if cfg!(feature = "metal") {
"metal"
} else if cfg!(feature = "cuda") {
"cuda"
@ -49,5 +49,5 @@ const fn device_variant() -> &'static str {
"mkl"
} else {
"cpu"
};
}
}