Skip to content

Added vpu optimized mean int8 - #221

Open
LeslieXMOS wants to merge 4 commits into
xmos:developfrom
LeslieXMOS:feature/mean8_xs3_vpu
Open

Added vpu optimized mean int8#221
LeslieXMOS wants to merge 4 commits into
xmos:developfrom
LeslieXMOS:feature/mean8_xs3_vpu

Conversation

@LeslieXMOS

Copy link
Copy Markdown
Contributor

Accelerate mean_int8 when running in xs3 instruction with VPU

mean_int8 will use mean_in8_mean when

  • the mean_axis is the last axis, which means end_dim_size =1
  • mean_dim_size needs to be divided by 4 (word alignment needed by xs3 VPU)

Inside mean_int8 it will use s32_to_f32 from xcore_math to quickly convert int32_t data to float.

This PR also provided a 2x faster asm for rounding float to int8_t

Comment thread lib_nn/src/asm/mean8.S Outdated
Comment thread lib_nn/src/asm/mean8.S
Comment thread lib_nn/src/asm/mean8.S Outdated
Comment thread lib_nn/src/asm/mean8.S Outdated
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants