diskann-tools 0.51.0

DiskANN is a fast approximate nearest neighbor search library for high dimensional data
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

/*
 * Copyright (c) Microsoft Corporation.
 * Licensed under the MIT license.
 */

use num_cpus;

pub fn get_num_threads(num_threads: Option<usize>) -> usize {
    match num_threads {
        Some(n) => n,
        None => num_cpus::get(),
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_get_num_threads_with_some() {
        assert_eq!(get_num_threads(Some(4)), 4);
        assert_eq!(get_num_threads(Some(1)), 1);
        assert_eq!(get_num_threads(Some(16)), 16);
    }

    #[test]
    fn test_get_num_threads_with_none() {
        let result = get_num_threads(None);
        // Should return the number of CPUs, which is at least 1
        assert!(result >= 1);
        // Should match num_cpus::get()
        assert_eq!(result, num_cpus::get());
    }
}