use obrah::runtime::Env;
use obrah::data::Buffer;
use obrah::kernel::{make_kernel, setarg, run_kernel};
fn main() {
let mut env = Env::new();
env.use_kernel("kernel.cl");
env.program();
make_kernel(&mut env, "vec_add");
let mut a = vec![1.0f32, 2.0, 3.0, 4.0];
let mut b = vec![10.0f32, 20.0, 30.0, 40.0];
let mut result = vec![0.0f32; a.len()];
let mut buf_a = Buffer::new(&mut env, &mut a);
let mut buf_b = Buffer::new(&mut env, &mut b);
let mut buf_result = Buffer::new(&mut env, &mut result);
buf_a.to(&a, &mut env);
buf_b.to(&b, &mut env);
setarg(&env, &mut buf_a, 0);
setarg(&env, &mut buf_b, 1);
setarg(&env, &mut buf_result, 2);
run_kernel(&mut env, a.len());
buf_result.from(&mut result, &mut env);
println!("Result: {:?}", result);
}