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

PyTorch like display implementation.

Browse Source
This commit is contained in:
laurent
2023-06-27 21:16:35 +01:00
parent 934655a60d
commit 8c81a70170
3 changed files with 265 additions and 258 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/utils.rs
View File

@ -1,6 +1,6 @@
use std::str::FromStr;
pub(crate) fn get_num_threads() -> usize {
pub fn get_num_threads() -> usize {
// Respond to the same environment variable as rayon.
match std::env::var("RAYON_NUM_THREADS")
.ok()