gpu: implement ComputePassEncoder.dispatch

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

View file

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

View file

@ -1610,6 +1610,21 @@ const compute_pass_encoder_vtable = ComputePassEncoder.VTable{
c.wgpuComputePassEncoderRelease(@ptrCast(c.WGPUComputePassEncoder, ptr)); c.wgpuComputePassEncoderRelease(@ptrCast(c.WGPUComputePassEncoder, ptr));
} }
}).release, }).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 { .end = (struct {
pub fn end(ptr: *anyopaque) void { pub fn end(ptr: *anyopaque) void {
c.wgpuComputePassEncoderEnd(@ptrCast(c.WGPUComputePassEncoder, ptr)); c.wgpuComputePassEncoderEnd(@ptrCast(c.WGPUComputePassEncoder, ptr));