some small changes
This commit is contained in:
parent
8a9870fe26
commit
7fd740b8fc
1 changed files with 11 additions and 2 deletions
11
src/main.zig
11
src/main.zig
|
@ -25,7 +25,7 @@ pub fn main() void {
|
|||
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 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]);
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue