Skip to content

Add oneDNN patch to fix torch.Linear regression#464

Merged
puneetmatharu merged 2 commits into
ARM-software:mainfrom
jondea:fix-torch-linear-regression
Jun 3, 2026
Merged

Add oneDNN patch to fix torch.Linear regression#464
puneetmatharu merged 2 commits into
ARM-software:mainfrom
jondea:fix-torch-linear-regression

Conversation

@jondea
Copy link
Copy Markdown
Contributor

@jondea jondea commented Jun 3, 2026

This restores the previous PyTorch+oneDNN behaviour of doing a single ba->blocked reorder rather than ba->ab then ab->blocked.

And restores the previous behaviour of PyTorch bmm heuristic by adding pytorch/pytorch@09a5778

@jondea jondea force-pushed the fix-torch-linear-regression branch from 7984cea to 2ae956b Compare June 3, 2026 07:55
@puneetmatharu
Copy link
Copy Markdown
Contributor

Reference PR: uxlfoundation/oneDNN#5156

Copy link
Copy Markdown
Contributor

@puneetmatharu puneetmatharu left a comment

Choose a reason for hiding this comment

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

Lgtm

@puneetmatharu puneetmatharu merged commit 944cd31 into ARM-software:main Jun 3, 2026
9 checks passed
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