From 7fd740b8fc594abf8527a545b7e824b009b27aea Mon Sep 17 00:00:00 2001 From: Nickiel12 Date: Fri, 11 Oct 2024 23:54:36 +0000 Subject: [PATCH] some small changes --- src/main.zig | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/src/main.zig b/src/main.zig index fc76f39..d67a869 100644 --- a/src/main.zig +++ b/src/main.zig @@ -24,8 +24,8 @@ pub fn main() void { var hef : hlo.hailo_hef = undefined; var device_params: hlo.hailo_vdevice_params_t = undefined; var network_group: hlo.hailo_configured_network_group = undefined; - const input_vstream_params : [max_edge_layers]hlo.hailo_input_vstream_params_by_name_t = undefined; - const output_vstream_params: [max_edge_layers]hlo.hailo_output_vstream_params_by_name_t = undefined; + const input_vstream_params : [max_edge_layers]hlo.hailo_input_vstream_params_by_name_t = undefined; + const output_vstream_params : [max_edge_layers]hlo.hailo_output_vstream_params_by_name_t = undefined; const output_vstreams : [max_edge_layers]hlo.hailo_output_vstream = undefined; const input_vstreams : [max_edge_layers]hlo.hailo_input_vstream = undefined; var input_vstream_size = max_edge_layers; @@ -54,6 +54,8 @@ pub fn main() void { output_vstream_params, &output_vstream_size); assert(status == hlo.HAILO_SUCCESS); + assert(input_vstream_size <= max_edge_layers); + status = hlo.hailo_create_input_vstreams(network_group, input_vstream_params, input_vstream_size, input_vstreams); assert(status == hlo.HAILO_SUCCESS); @@ -63,6 +65,13 @@ pub fn main() void { status = hlo.activate_network_group(network_group, null, &activated_network_group); assert(status == hlo.HAILO_SUCCESS); + defer hlo.hailo_release_output_vstreams(output_vstreams, output_vstream_size); + defer hlo.hailo_release_input_vstreams(input_vstreams, input_vstream_size); + defer hlo.hailo_release_hef(hef); + defer hlo.hailo_release_vdevice(vdevice); + + std.debug.print("output vstream size is: {any}", .{output_vstream_size}); + for (0..output_vstream_size) |i| { hlo.hailo_get_output_vstream_info(output_vstreams[i], &output_vstreams_info[i]); }