tl_cpu 0.4.9

CPU tensor backend for TL
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

//! CpuStream — CPU バックエンドの GpuStream 実装
//!
//! CPU には非同期ストリームの概念はないが、
//! バックエンド互換性のために実装する。

use crate::graph::CpuGraph;
use tl_backend::stream::GpuStream;

/// CPU コマンドストリーム
pub struct CpuStream {
    capturing: Option<CpuGraph>,
}

impl CpuStream {
    pub fn new() -> Self {
        CpuStream { capturing: None }
    }
}

impl GpuStream for CpuStream {
    type Graph = CpuGraph;

    fn synchronize(&mut self) {
        // CPU は同期実行なので何もしない
    }

    fn needs_sync(&self) -> bool {
        false // CPU は常に同期済み
    }

    fn begin_capture(&mut self) {
        self.capturing = Some(CpuGraph::new());
    }

    fn end_capture(&mut self) -> CpuGraph {
        self.capturing.take()
            .expect("end_capture called without begin_capture")
    }

    fn is_capturing(&self) -> bool {
        self.capturing.is_some()
    }
}