Skip to main content

vm_spirv_poly/
poly.rs

1use anyhow::Result;
2
3fn main() -> Result<()> {
4    let source = std::fs::read("zusts/gpu/poly.zs")?;
5    let kernel = vm_spirv::compile_source_with_workgroup_size(source, "poly", "main", [32, 32, 1])?;
6    let bytes = kernel.spirv.words().iter().flat_map(|word| word.to_le_bytes()).collect::<Vec<_>>();
7    std::fs::write("poly.spv", bytes)?;
8    std::fs::write("poly.spvasm", kernel.spirv.disassemble())?;
9    println!("wrote poly.spv ({} words)", kernel.spirv.words().len());
10    Ok(())
11}