mirror of
https://github.com/italicsjenga/vello.git
synced 2025-01-11 04:51:32 +11:00
bae185efbd
Move types into the toplevel and hide implementation details. Remove deref of hub CmdBuf to mux. Restrict public visibility of internals. Most items have some docs, though improvements are still possible. In particular, there should be detailed safety info.
37 lines
1.5 KiB
Rust
37 lines
1.5 KiB
Rust
use piet_gpu_hal::include_shader;
|
|
use piet_gpu_hal::{BufferUsage, Instance, Session};
|
|
|
|
fn main() {
|
|
let (instance, _) = Instance::new(None).unwrap();
|
|
unsafe {
|
|
let device = instance.device(None).unwrap();
|
|
let session = Session::new(device);
|
|
let usage = BufferUsage::MAP_READ | BufferUsage::STORAGE;
|
|
let src = (0..256).map(|x| x + 1).collect::<Vec<u32>>();
|
|
let buffer = session.create_buffer_init(&src, usage).unwrap();
|
|
let code = include_shader!(&session, "./shader/gen/collatz");
|
|
let pipeline = session.create_simple_compute_pipeline(code, 1).unwrap();
|
|
let descriptor_set = session
|
|
.create_simple_descriptor_set(&pipeline, &[&buffer])
|
|
.unwrap();
|
|
let query_pool = session.create_query_pool(2).unwrap();
|
|
let mut cmd_buf = session.cmd_buf().unwrap();
|
|
cmd_buf.begin();
|
|
cmd_buf.reset_query_pool(&query_pool);
|
|
cmd_buf.write_timestamp(&query_pool, 0);
|
|
cmd_buf.dispatch(&pipeline, &descriptor_set, (256, 1, 1), (1, 1, 1));
|
|
cmd_buf.write_timestamp(&query_pool, 1);
|
|
cmd_buf.host_barrier();
|
|
cmd_buf.finish();
|
|
let submitted = session.run_cmd_buf(cmd_buf, &[], &[]).unwrap();
|
|
submitted.wait().unwrap();
|
|
let timestamps = session.fetch_query_pool(&query_pool);
|
|
let mut dst: Vec<u32> = Default::default();
|
|
buffer.read(&mut dst).unwrap();
|
|
for (i, val) in dst.iter().enumerate().take(16) {
|
|
println!("{}: {}", i, val);
|
|
}
|
|
println!("{:?}", timestamps);
|
|
}
|
|
}
|