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

gpu: implement ComputePassEncoder.dispatch

Browse source 
Signed-off-by: Stephen Gutekanst <stephen@hexops.com>
This commit is contained in:
Stephen Gutekanst 2022-03-17 10:21:43 -07:00 • committed by Stephen Gutekanst
parent 243827d232
commit 061ca10f78
2 changed files with 31 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

18
gpu/src/ComputePassEncoder.zig
View file

@ -13,8 +13,7 @@ vtable: *const VTable,
pub const VTable = struct {
reference: fn (ptr: *anyopaque) void,
release: fn (ptr: *anyopaque) void,
// dispatch: fn (ptr: *anyopaque, workgroup_count_x: u32, workgroup_count_y: u32, workgroup_count_z: u32) void,
// WGPU_EXPORT void wgpuComputePassEncoderDispatch(WGPUComputePassEncoder computePassEncoder, uint32_t workgroupCountX, uint32_t workgroupCountY, uint32_t workgroupCountZ);
dispatch: fn (ptr: *anyopaque, workgroup_count_x: u32, workgroup_count_y: u32, workgroup_count_z: u32) void,
// dispatchIndirect: fn (ptr: *anyopaque, indirect_buffer: Buffer, indirect_offset: u64) void,
// WGPU_EXPORT void wgpuComputePassEncoderDispatchIndirect(WGPUComputePassEncoder computePassEncoder, WGPUBuffer indirectBuffer, uint64_t indirectOffset);
end: fn (ptr: *anyopaque) void,
@ -35,6 +34,15 @@ pub inline fn release(enc: ComputePassEncoder) void {
enc.vtable.release(enc.ptr);
}
pub inline fn dispatch(
enc: ComputePassEncoder,
workgroup_count_x: u32,
workgroup_count_y: u32,
workgroup_count_z: u32,
) void {
enc.vtable.dispatch(enc.ptr, workgroup_count_x, workgroup_count_y, workgroup_count_z);
}
pub inline fn end(enc: ComputePassEncoder) void {
enc.vtable.end(enc.ptr);
}
@ -81,8 +89,14 @@ test {
_ = VTable;
_ = reference;
_ = release;
_ = dispatch;
_ = end;
_ = insertDebugMarker;
_ = popDebugGroup;
_ = pushDebugGroup;
_ = setBindGroup;
_ = setLabel;
_ = setPipeline;
_ = writeTimestamp;
_ = Descriptor;
}

15
gpu/src/NativeInstance.zig
View file

@ -1610,6 +1610,21 @@ const compute_pass_encoder_vtable = ComputePassEncoder.VTable{
c.wgpuComputePassEncoderRelease(@ptrCast(c.WGPUComputePassEncoder, ptr));
}
}).release,
.dispatch = (struct {
pub fn dispatch(
ptr: *anyopaque,
workgroup_count_x: u32,
workgroup_count_y: u32,
workgroup_count_z: u32,
) void {
c.wgpuComputePassEncoderDispatch(
@ptrCast(c.WGPUComputePassEncoder, ptr),
workgroup_count_x,
workgroup_count_y,
workgroup_count_z,
);
}
}).dispatch,
.end = (struct {
pub fn end(ptr: *anyopaque) void {
c.wgpuComputePassEncoderEnd(@ptrCast(c.WGPUComputePassEncoder, ptr));