mirror of
https://github.com/italicsjenga/vello.git
synced 2024-10-17 23:11:30 +11:00
e4b16e706a
These function, but can use some work. First, the buffer situation is worse than it should be. It should be possible to create a single readback buffer rather then copy from gpu-local to host-coherent. Second, the command buffer `finish_timestamps` call doesn't correlate to anything in Vulkan, so needs plumbing up through the hub in one form or other when that happens. I'm inclined to make it ergonomic by doing a bit of resource tracking that will trigger the appropriate call (and subsequent host barrier) in the `finish` method on the command buffer.
86 lines
2.4 KiB
Rust
86 lines
2.4 KiB
Rust
//! An example to exercise the dx12 backend, while it's being developed.
|
|
//! This will probably go away when it's fully implemented and we can
|
|
//! just use the hub.
|
|
|
|
use piet_gpu_hal::{dx12, CmdBuf, Device, Error, MemFlags};
|
|
|
|
const SHADER_CODE: &str = r#"RWByteAddressBuffer _53 : register(u0, space0);
|
|
|
|
static uint3 gl_GlobalInvocationID;
|
|
struct SPIRV_Cross_Input
|
|
{
|
|
uint3 gl_GlobalInvocationID : SV_DispatchThreadID;
|
|
};
|
|
|
|
uint collatz_iterations(inout uint n)
|
|
{
|
|
uint i = 0u;
|
|
while (n != 1u)
|
|
{
|
|
if ((n & 1u) == 0u)
|
|
{
|
|
n /= 2u;
|
|
}
|
|
else
|
|
{
|
|
n = (3u * n) + 1u;
|
|
}
|
|
i++;
|
|
}
|
|
return i;
|
|
}
|
|
|
|
void comp_main()
|
|
{
|
|
uint index = gl_GlobalInvocationID.x;
|
|
uint param = _53.Load(index * 4 + 0);
|
|
uint _61 = collatz_iterations(param);
|
|
_53.Store(index * 4 + 0, _61);
|
|
}
|
|
|
|
[numthreads(256, 1, 1)]
|
|
void main(SPIRV_Cross_Input stage_input)
|
|
{
|
|
gl_GlobalInvocationID = stage_input.gl_GlobalInvocationID;
|
|
comp_main();
|
|
}
|
|
"#;
|
|
|
|
fn toy() -> Result<(), Error> {
|
|
let instance = dx12::Dx12Instance::new()?;
|
|
let device = instance.device()?;
|
|
let buf = device.create_buffer(1024, MemFlags::host_coherent())?;
|
|
let dev_buf = device.create_buffer(1024, MemFlags::device_local())?;
|
|
let data: Vec<u32> = (1..257).collect();
|
|
let query_pool = device.create_query_pool(2)?;
|
|
unsafe {
|
|
device.write_buffer(&buf, &data)?;
|
|
let pipeline = device.create_simple_compute_pipeline(SHADER_CODE, 1, 0)?;
|
|
let ds = device.create_descriptor_set(&pipeline, &[&dev_buf], &[])?;
|
|
let mut cmd_buf = device.create_cmd_buf()?;
|
|
let fence = device.create_fence(false)?;
|
|
cmd_buf.begin();
|
|
cmd_buf.copy_buffer(&buf, &dev_buf);
|
|
cmd_buf.memory_barrier();
|
|
cmd_buf.write_timestamp(&query_pool, 0);
|
|
cmd_buf.dispatch(&pipeline, &ds, (1, 1, 1));
|
|
cmd_buf.write_timestamp(&query_pool, 1);
|
|
cmd_buf.memory_barrier();
|
|
cmd_buf.copy_buffer(&dev_buf, &buf);
|
|
cmd_buf.finish_timestamps(&query_pool);
|
|
cmd_buf.host_barrier();
|
|
cmd_buf.finish();
|
|
device.run_cmd_buf(&cmd_buf, &[], &[], Some(&fence))?;
|
|
device.wait_and_reset(&[fence])?;
|
|
let mut readback: Vec<u32> = Vec::new();
|
|
device.read_buffer(&buf, &mut readback)?;
|
|
println!("{:?}", readback);
|
|
println!("{:?}", device.fetch_query_pool(&query_pool));
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
fn main() {
|
|
toy().unwrap();
|
|
}
|