relearn 0.3.1

A Reinforcement Learning library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

//! Details concerning torch backends.
use tch::Cuda;

/// A RAII guard to enable / disable cuDNN and restore the previous state after being dropped.
pub struct WithCudnnEnabled {
    previous_state: bool,
}

impl WithCudnnEnabled {
    #[must_use]
    pub fn new(enabled: bool) -> Self {
        let previous_state = Cuda::user_enabled_cudnn();
        Cuda::set_user_enabled_cudnn(enabled);
        Self { previous_state }
    }
}

impl Drop for WithCudnnEnabled {
    fn drop(&mut self) {
        Cuda::set_user_enabled_cudnn(self.previous_state);
    }
}