mirror of
https://github.com/italicsjenga/vello.git
synced 2025-01-10 20:51:29 +11:00
ba2b27cc3c
The current plan is to more or less follow the wgpu/wgpu-hal approach. In the mux/backend layer (which corresponds fairly strongly to wgpu-hal), there isn't explicit construction of a compute encoder, but there are new methods for beginning and ending a compute pass. At the hub layer (which corresponds to wgpu) there will be a ComputeEncoder object. That said, there will be some differences. The WebGPU "end" method on a compute encoder is implemented in wgpu as Drop, and that is not ideal. Also, the wgpu-hal approach to timer queries (still based on write_timestamp) is not up to the task of Metal timer queries, where the query offsets have to be specified at compute encoder creation. That's why there are different projects :) WIP: current state is that stage-style queries work on Apple Silicon, but non-Metal backends are broken, and piet-gpu is not yet updated to use new API.
40 lines
1.7 KiB
Rust
40 lines
1.7 KiB
Rust
use piet_gpu_hal::{include_shader, BindType, ComputePassDescriptor};
|
|
use piet_gpu_hal::{BufferUsage, Instance, InstanceFlags, Session};
|
|
|
|
fn main() {
|
|
let (instance, _) = Instance::new(None, InstanceFlags::empty()).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_compute_pipeline(code, &[BindType::Buffer])
|
|
.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);
|
|
let mut pass = cmd_buf.begin_compute_pass(&ComputePassDescriptor::timer(&query_pool, 0, 1));
|
|
pass.dispatch(&pipeline, &descriptor_set, (256, 1, 1), (1, 1, 1));
|
|
pass.end();
|
|
cmd_buf.finish_timestamps(&query_pool);
|
|
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);
|
|
}
|
|
}
|