added inference vstream loading
This commit is contained in:
parent
93fc8addc4
commit
c3112f90b0
1 changed files with 31 additions and 3 deletions
34
src/main.zig
34
src/main.zig
|
@ -13,10 +13,10 @@ const max_edge_layers = 32;
|
||||||
|
|
||||||
pub fn main() void {
|
pub fn main() void {
|
||||||
|
|
||||||
// var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
|
var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
|
||||||
// defer arena.deinit();
|
defer arena.deinit();
|
||||||
|
|
||||||
// const alloc = arena.allocator();
|
const alloc = arena.allocator();
|
||||||
|
|
||||||
std.fs.cwd().access(hef_file, .{ }) catch |e| {
|
std.fs.cwd().access(hef_file, .{ }) catch |e| {
|
||||||
std.debug.panic("Could not open hef file! '{any}'", .{ e });
|
std.debug.panic("Could not open hef file! '{any}'", .{ e });
|
||||||
|
@ -85,6 +85,34 @@ pub fn main() void {
|
||||||
std.debug.print("Output_vstream_size is: '{d}'\n", .{ output_vstream_size });
|
std.debug.print("Output_vstream_size is: '{d}'\n", .{ output_vstream_size });
|
||||||
std.debug.print("Input_vstream_size is: '{d}'\n", .{ input_vstream_size });
|
std.debug.print("Input_vstream_size is: '{d}'\n", .{ input_vstream_size });
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
var input_frame_size: usize = 0;
|
||||||
|
|
||||||
|
status = hlo.hailo_get_input_vstream_frame_size(input_vstreams, &input_frame_size);
|
||||||
|
assert(status == hlo.HAILO_SUCCESS);
|
||||||
|
|
||||||
|
const frame_count = input_frame_size / @sizeOf(u8);
|
||||||
|
std.debug.print("Trying to create {d} u8s\n", .{ frame_count });
|
||||||
|
var input_data = try alloc.create(u8, frame_count);
|
||||||
|
|
||||||
|
var rnd = std.Random.init(0);
|
||||||
|
|
||||||
|
for (0..frame_count) |i| {
|
||||||
|
input_data[i] = rnd.random().int(u8);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Here we simulate taking our data and chunking it into frame_size, then uploading each frame until we have uploaded
|
||||||
|
// all of our data
|
||||||
|
for (0..frame_count) |_| {
|
||||||
|
status = hlo.hailo_vstream_write_raw_buffer(input_vstreams, input_data, input_frame_size);
|
||||||
|
assert(status == hlo.HAILO_SUCCESS);
|
||||||
|
}
|
||||||
|
|
||||||
|
status = hlo.hailo_flush_input_vstream(input_vstreams);
|
||||||
|
assert(status == hlo.HAILO_SUCCESS);
|
||||||
|
|
||||||
|
|
||||||
_ = hlo.hailo_release_output_vstreams(&output_vstreams, output_vstream_size);
|
_ = hlo.hailo_release_output_vstreams(&output_vstreams, output_vstream_size);
|
||||||
_ = hlo.hailo_release_input_vstreams(&input_vstreams, input_vstream_size);
|
_ = hlo.hailo_release_input_vstreams(&input_vstreams, input_vstream_size);
|
||||||
_ = hlo.hailo_release_hef(hef);
|
_ = hlo.hailo_release_hef(hef);
|
||||||
|
|
Loading…
Reference in a new issue