huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 11:37:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use num-cpus to enable parallelism.

Browse Source
This commit is contained in:
laurent
2023-06-27 14:42:26 +01:00
parent 8371890996
commit ca6aa8ff12
4 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-core/src/cpu_backend.rs
View File

@ -669,7 +669,7 @@ impl CpuStorage {
// conj_rhs: bool,
true,
// parallelism: Parallelism
Parallelism::None,
Parallelism::Rayon(crate::utils::get_num_threads()),
)
}
}
@ -721,11 +721,10 @@ impl CpuStorage {
// conj_rhs: bool,
true,
// parallelism: Parallelism
Parallelism::None,
Parallelism::Rayon(crate::utils::get_num_threads()),
)
}
}
Ok(Self::F32(dst))
}
(CpuStorage::F64(lhs), CpuStorage::F64(rhs)) => {
@ -773,7 +772,7 @@ impl CpuStorage {
// conj_rhs: bool,
true,
// parallelism: Parallelism
Parallelism::None,
Parallelism::Rayon(crate::utils::get_num_threads()),
)
}
}