EmbarkStudios / rust-gpu

🐉 Making Rust a first-class language and ecosystem for GPU shaders 🚧

Home Page:https://shader.rs

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to get spir-v with OpCapability Kernel?

jczaja opened this issue · comments

Hi,

I have a dummy compute kernel:

#[spirv(compute(threads(768)))]
pub fn main_cs(
    #[spirv(global_invocation_id)] id: UVec3,
    #[spirv(storage_buffer, descriptor_set = 0, binding = 0)] prime_indices: &mut [u32],
) {
    let index = id.x as usize;
    prime_indices[index] = collatz(prime_indices[index]).1u32;
}

I can run it using Vulkano abd it works, but I wanted to run this spir-v kernel using Intel Level Zero and this did not work. Problem is that Intel Level zero is OpenCL like API and it require that SPIR-V is :

OpCapability Kernel
OpMemoryModel Physical64 OpenCL
..

while my dummy compute kernel as SPIR-V contains:

OpCapability Shader 
OpMemoryModel Logical Simple 
...

How can I compiler my rust compute shader so it is of SPIR-V form OpCapability Kernel ?

I believe you do it in the spirv builder interface.

SpirvBuilder::new(<path_to_crate>, <target>)
        .capability(Capability::Kernel)
        .build()?