Skip to content

Fix: handle num_o_neurons == 1 in pulp_nn_linear_int8#11

Open
marceloribeiro19 wants to merge 1 commit intopulp-platform:paper_versionfrom
marceloribeiro19:fix-single-neuron-bug
Open

Fix: handle num_o_neurons == 1 in pulp_nn_linear_int8#11
marceloribeiro19 wants to merge 1 commit intopulp-platform:paper_versionfrom
marceloribeiro19:fix-single-neuron-bug

Conversation

@marceloribeiro19
Copy link

This PR adds a fix for the edge case where num_o_neurons == 1.

The original code returns invalid results for a single output neuron, as the optimized SIMD loop expects at least 2 neurons.

-This patch adds a simple scalar fallback.
-Tested manually against expected values.
-Backward-compatible.

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.

1 participant