nove_tensor 0.1.2

An easy-to-use, lightweight deep learning library wrapped around Candle Tensor.
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

use nove::tensor::Device;

#[test]
fn test_eq() {
    let device1 = Device::cpu();
    let device2 = Device::cpu();
    assert_eq!(device1, device2);
}

#[cfg(feature = "cuda")]
#[test]
fn test_cuda_with_invalid_index() {
    // Test cuda with an invalid index
    let device = Device::cuda(usize::MAX);
    assert_eq!(device.is_err(), true);
}

#[cfg(feature = "cuda")]
#[test]
fn test_cuda_if_available() {
    // Try to get a CUDA device
    let cuda_if_available = Device::cuda_if_available(0);

    // Get the CPU device used to compare with
    let cpu = Device::cpu();

    // Judge whether the CUDA device is available
    match Device::cuda(0) {
        // If CUDA device is available, cuda_if_available should be equal to it
        Ok(cuda) => assert_eq!(cuda, cuda_if_available),
        // If CUDA device is not available, cuda_if_available should be equal to cpu
        Err(_) => assert_eq!(cuda_if_available, cpu),
    }
}

#[cfg(feature = "metal")]
#[test]
fn test_metal_with_invalid_index() {
    // Test metal with an invalid index
    let device = Device::metal(usize::MAX);
    assert_eq!(device.is_err(), true);
}

#[cfg(feature = "metal")]
#[test]
fn test_metal_if_available() {
    // Try to get a Metal device
    let metal_if_available = Device::metal_if_available(0);

    // Get the CPU device used to compare with
    let cpu = Device::cpu();

    // Judge whether the Metal device is available
    match Device::metal(0) {
        // If Metal device is available, metal_if_available should be equal to it
        Ok(metal) => assert_eq!(metal, metal_if_available),
        // If Metal device is not available, metal_if_available should be equal to cpu
        Err(_) => assert_eq!(metal_if_available, cpu),
    }
}