Skip to content

4-bit support for convolution#85

Merged
mansnils merged 1 commit intoARM-software:mainfrom
ArmRyan:experimental/conv_int4
Nov 14, 2023
Merged

4-bit support for convolution#85
mansnils merged 1 commit intoARM-software:mainfrom
ArmRyan:experimental/conv_int4

Conversation

@ArmRyan
Copy link
Copy Markdown
Collaborator

@ArmRyan ArmRyan commented Nov 14, 2023

  • Adds arm_convolve_wrapper function for 4bit weights
  • Adds arm_convolve_1x1 normal and fast variant for 4bit weights
  • Adds arm_nn_mat_mult_kernel_s4_s16 for multiply 4bit weights with 16bit input
  • Adds mat_mult_nt_t_s4 function for dsp and scalar
  • Adds scalar and dsp implementation for arm_convolve with 4bit weights
  • Adds Unit tests for 4bit weight convolutions

Change-Id: Idea55432fdab2db05a033889d7c39dd0ea69f8ad

 - Adds arm_convolve_wrapper function for 4bit weights
 - Adds arm_convolve_1x1 normal and fast variant for 4bit weights
 - Adds arm_nn_mat_mult_kernel_s4_s16 for multiply 4bit weights with
   16bit input
 - Adds mat_mult_nt_t_s4 function for dsp and scalar
 - Adds scalar and dsp implementation for arm_convolve with 4bit weights
 - Adds Unit tests for 4bit weight convolutions

Change-Id: Idea55432fdab2db05a033889d7c39dd0ea69f8ad
Signed-off-by: Ryan O'Shea <ryan.oshea3@arm.com>
@mansnils mansnils added the feature New features label Nov 14, 2023
@mansnils mansnils merged commit bfc54ed into ARM-software:main Nov 14, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

feature New features

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants