ferrotensor-backend 0.2.0

wgpu compute backend with hand-written WGSL kernels for the Ferrotensor ML framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

//! Create the wgpu context (adapter, device and queue).
//!
//! Run with: `cargo run -p ferrotensor-backend --example init_wgpu`
//!
//! Output:
//!
//! ```text
//! Initializing WGPU Context...
//! WGPU Context initialized successfully!
//! ```
//!
//! Set `RUST_LOG=info` to see which adapter was picked.

use ferrotensor_backend::WgpuContext;

fn main() {
    env_logger::init();
    pollster::block_on(async {
        println!("Initializing WGPU Context...");
        let _ctx = WgpuContext::new().await;
        println!("WGPU Context initialized successfully!");
    });
}