Skip to content

Layer norm throws on dimension mismatch#2870

Merged
awni merged 2 commits into
mainfrom
layer_norm_throws
Dec 4, 2025
Merged

Layer norm throws on dimension mismatch#2870
awni merged 2 commits into
mainfrom
layer_norm_throws

Conversation

@awni

@awni awni commented Dec 4, 2025

Copy link
Copy Markdown
Member

Layer norm didn't check the dimensions of x matched the weights and biases which can result in a segfault / incorrect results.

@awni awni requested review from angeloskath and zcbenz December 4, 2025 16:04

@angeloskath angeloskath left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

🙏

@awni awni merged commit 941cfe2 into main Dec 4, 2025
12 checks passed
@awni awni deleted the layer_norm_throws branch December 4, 2025 19:21
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