40 lines
1.3 KiB
Rust
40 lines
1.3 KiB
Rust
#![allow(dead_code)]
|
|
|
|
// use std::rc::Rc;
|
|
// use std::sync::{Arc, RwLock};
|
|
//
|
|
// use coaster::backend::Backend;
|
|
// use coaster::IFramework;
|
|
// use coaster::frameworks::cuda::{Cuda, get_cuda_backend};
|
|
// use coaster::frameworks::native::{Cpu, Native};
|
|
// use coaster::tensor::SharedTensor;
|
|
// use juice::layer::{LayerConfig, Layer, ComputeOutput, ILayer};
|
|
// use juice::layers::{LinearConfig, SequentialConfig, Sequential};
|
|
// use juice::util::ArcLock;
|
|
|
|
// mod drafts;
|
|
// use drafts::write_to_memory;
|
|
mod simple;
|
|
|
|
|
|
fn main() {
|
|
simple::test_simple();
|
|
// let backend: Rc<Backend<Cuda>> = Rc::new(get_cuda_backend());
|
|
// let native: Native = Native::new();
|
|
// let cpu: Cpu = native.new_device(native.hardwares()).unwrap();
|
|
//
|
|
// let mut net_cfg: SequentialConfig = SequentialConfig::default();
|
|
// net_cfg.add_input("data", &vec![1, 3, 1]);
|
|
// net_cfg.add_layer(
|
|
// LayerConfig::new("linear1", LinearConfig { output_size: 2 })
|
|
// );
|
|
// let mut net = Sequential::from_config(backend.clone(), &net_cfg);
|
|
//
|
|
// let mut x: SharedTensor<f32> = SharedTensor::<f32>::new(&(3, 1));
|
|
// let x_values: &Vec<f32> = &vec![1f32, -2.0, 0.2];
|
|
// write_to_memory(x.write_only(&cpu).unwrap(), x_values);
|
|
// let x_lock: ArcLock<SharedTensor<f32>> = Arc::new(RwLock::new(x));
|
|
//
|
|
// net.forward(&backend, &[x_lock],)
|
|
}
|