tarry.singh/Radix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update loader example to demonstrate new HostBuffer helpers and type‑inferred buffer loading.

Browse Source
This commit is contained in:
Foke Singh 2024-10-25 10:20:04 +00:00
parent 048d7eb38e
commit 1540c6e85e
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/loader/main.zig
View File

@ -1,9 +1,8 @@
const std = @import("std"); const std = @import("std");
const asynk = @import("async");
const stdx = @import("stdx"); const stdx = @import("stdx");
const zml = @import("zml"); const zml = @import("zml");
const asynk = @import("async");
const asyncc = asynk.asyncc;
pub fn main() !void { pub fn main() !void {
try asynk.AsyncThread.main(std.heap.c_allocator, asyncMain); try asynk.AsyncThread.main(std.heap.c_allocator, asyncMain);
@ -39,9 +38,8 @@ pub fn asyncMain() !void {
var buffers = try gpa.allocator().alloc(zml.Buffer, buffer_store.buffers.count()); var buffers = try gpa.allocator().alloc(zml.Buffer, buffer_store.buffers.count());
defer { defer {
for (buffers) |*buf| { // Note we don't pass an allocator to buf.deinit() cause its allocated on the device.
buf.deinit(); for (buffers) |*buf| buf.deinit();
}
gpa.allocator().free(buffers); gpa.allocator().free(buffers);
} }
@ -54,7 +52,7 @@ pub fn asyncMain() !void {
while (it.next()) |entry| : (i += 1) { while (it.next()) |entry| : (i += 1) {
const host_buffer = entry.value_ptr.*; const host_buffer = entry.value_ptr.*;
total_bytes += host_buffer.data.len; total_bytes += host_buffer.shape().byteSize();
std.debug.print("Buffer: {s} ({any} / {any})\n", .{ entry.key_ptr.*, i + 1, buffer_store.buffers.count() }); std.debug.print("Buffer: {s} ({any} / {any})\n", .{ entry.key_ptr.*, i + 1, buffer_store.buffers.count() });
buffers[i] = try zml.Buffer.from(platform, host_buffer); buffers[i] = try zml.Buffer.from(platform, host_buffer);
} }