huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Cudarc static-linking enabled.

Browse Source
This commit is contained in:
Nicolas Patry
2025-03-29 09:27:53 +01:00
parent 2c0f6b008e
commit ec6d7ca773
16 changed files with 36 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-core/benches/benchmarks/mod.rs
View File

@ -20,9 +20,9 @@ impl BenchDevice for Device {
match self {
Device::Cpu => Ok(()),
Device::Cuda(device) => {
#[cfg(feature = "cuda")]
#[cfg(feature = "_cuda")]
return Ok(device.synchronize()?);
#[cfg(not(feature = "cuda"))]
#[cfg(not(feature = "_cuda"))]
panic!("Cuda device without cuda feature enabled: {:?}", device)
}
Device::Metal(device) => {
@ -61,7 +61,7 @@ impl BenchDeviceHandler {
let mut devices = Vec::new();
if cfg!(feature = "metal") {
devices.push(Device::new_metal(0)?);
} else if cfg!(feature = "cuda") {
} else if cfg!(feature = "_cuda") {
devices.push(Device::new_cuda(0)?);
}
devices.push(Device::Cpu);