zig/mach
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

gpu: implement Device.createComputePipeline

Browse source 
Signed-off-by: Stephen Gutekanst <stephen@hexops.com>
This commit is contained in:
Stephen Gutekanst 2022-03-15 20:17:07 -07:00 • committed by Stephen Gutekanst
parent 38535c7b5f
commit c90fb09de1
2 changed files with 37 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

9
gpu/src/Device.zig
View file

@ -29,7 +29,7 @@ pub const VTable = struct {
// WGPU_EXPORT WGPUBindGroupLayout wgpuDeviceCreateBindGroupLayout(WGPUDevice device, WGPUBindGroupLayoutDescriptor const * descriptor); // WGPU_EXPORT WGPUBindGroupLayout wgpuDeviceCreateBindGroupLayout(WGPUDevice device, WGPUBindGroupLayoutDescriptor const * descriptor);
// WGPU_EXPORT WGPUBuffer wgpuDeviceCreateBuffer(WGPUDevice device, WGPUBufferDescriptor const * descriptor); // WGPU_EXPORT WGPUBuffer wgpuDeviceCreateBuffer(WGPUDevice device, WGPUBufferDescriptor const * descriptor);
createCommandEncoder: fn (ptr: *anyopaque, descriptor: ?*const CommandEncoder.Descriptor) CommandEncoder, createCommandEncoder: fn (ptr: *anyopaque, descriptor: ?*const CommandEncoder.Descriptor) CommandEncoder,
// WGPU_EXPORT WGPUComputePipeline wgpuDeviceCreateComputePipeline(WGPUDevice device, WGPUComputePipelineDescriptor const * descriptor); createComputePipeline: fn (ptr: *anyopaque, descriptor: *const ComputePipeline.Descriptor) ComputePipeline,
createComputePipelineAsync: fn ( createComputePipelineAsync: fn (
ptr: *anyopaque, ptr: *anyopaque,
descriptor: *const ComputePipeline.Descriptor, descriptor: *const ComputePipeline.Descriptor,
@ -97,6 +97,13 @@ pub inline fn createCommandEncoder(device: Device, descriptor: ?*const CommandEn
return device.vtable.createCommandEncoder(device.ptr, descriptor); return device.vtable.createCommandEncoder(device.ptr, descriptor);
} }
pub inline fn createComputePipeline(
device: Device,
descriptor: *const ComputePipeline.Descriptor,
) ComputePipeline {
return device.vtable.createComputePipeline(device.ptr, descriptor);
}
pub inline fn createComputePipelineAsync( pub inline fn createComputePipelineAsync(
device: Device, device: Device,
descriptor: *const ComputePipeline.Descriptor, descriptor: *const ComputePipeline.Descriptor,

41
gpu/src/NativeInstance.zig
View file

@ -382,24 +382,26 @@ const device_vtable = Device.VTable{
return wrapCommandEncoder(c.wgpuDeviceCreateCommandEncoder(@ptrCast(c.WGPUDevice, ptr), desc)); return wrapCommandEncoder(c.wgpuDeviceCreateCommandEncoder(@ptrCast(c.WGPUDevice, ptr), desc));
} }
}).createCommandEncoder, }).createCommandEncoder,
.createComputePipeline = (struct {
pub fn createComputePipeline(
ptr: *anyopaque,
descriptor: *const ComputePipeline.Descriptor,
) ComputePipeline {
const desc = convertComputePipelineDescriptor(descriptor);
return wrapComputePipeline(c.wgpuDeviceCreateComputePipeline(
@ptrCast(c.WGPUDevice, ptr),
&desc,
));
}
}).createComputePipeline,
.createComputePipelineAsync = (struct { .createComputePipelineAsync = (struct {
pub fn createComputePipelineAsync( pub fn createComputePipelineAsync(
ptr: *anyopaque, ptr: *anyopaque,
descriptor: *const ComputePipeline.Descriptor, descriptor: *const ComputePipeline.Descriptor,
callback: *ComputePipeline.CreateCallback, callback: *ComputePipeline.CreateCallback,
) void { ) void {
const desc = c.WGPUComputePipelineDescriptor{ const desc = convertComputePipelineDescriptor(descriptor);
.nextInChain = null,
.label = if (descriptor.label) |l| l else null,
.layout = @ptrCast(c.WGPUPipelineLayout, descriptor.layout.ptr),
.compute = c.WGPUProgrammableStageDescriptor{
.nextInChain = null,
.module = @ptrCast(c.WGPUShaderModule, descriptor.compute.module.ptr),
.entryPoint = descriptor.compute.entry_point,
.constantCount = if (descriptor.compute.constants) |v| @intCast(u32, v.len) else 0,
.constants = if (descriptor.compute.constants) |v| @ptrCast(*const c.WGPUConstantEntry, &v[0]) else null,
},
};
const cCallback = (struct { const cCallback = (struct {
pub fn cCallback( pub fn cCallback(
@ -436,6 +438,21 @@ const device_vtable = Device.VTable{
}).createRenderPipeline, }).createRenderPipeline,
}; };
inline fn convertComputePipelineDescriptor(descriptor: *const ComputePipeline.Descriptor) c.WGPUComputePipelineDescriptor {
return .{
.nextInChain = null,
.label = if (descriptor.label) |l| l else null,
.layout = @ptrCast(c.WGPUPipelineLayout, descriptor.layout.ptr),
.compute = c.WGPUProgrammableStageDescriptor{
.nextInChain = null,
.module = @ptrCast(c.WGPUShaderModule, descriptor.compute.module.ptr),
.entryPoint = descriptor.compute.entry_point,
.constantCount = if (descriptor.compute.constants) |v| @intCast(u32, v.len) else 0,
.constants = if (descriptor.compute.constants) |v| @ptrCast(*const c.WGPUConstantEntry, &v[0]) else null,
},
};
}
inline fn convertRenderPipelineDescriptor( inline fn convertRenderPipelineDescriptor(
d: *const RenderPipeline.Descriptor, d: *const RenderPipeline.Descriptor,
tmp_depth_stencil: *c.WGPUDepthStencilState, tmp_depth_stencil: *c.WGPUDepthStencilState,