ndrs 0.2.0

A tensor library with GPU support
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

pub mod backend;
pub mod device;
pub mod dtype;
pub mod kernel;
pub mod macros;
pub mod stream;
pub mod tensor;
pub mod view;
pub mod view_ops;

pub use backend::cpu;
pub use backend::cuda;

// 统一设备/流 API
pub use device::Device;

// 其他导出保持不变
pub use dtype::{DType, DTYPE_FLOAT32, DTYPE_INT32};
pub use tensor::Tensor;
pub use view::{broadcast_shapes, ArcTensorView, RcTensorView, SliceInfo, TensorViewOps};

#[cfg(test)]
mod test_init {
    use ctor::ctor;

    #[ctor]
    fn init_logger() {
        let _ = env_logger::builder().is_test(true).try_init();
    }
}