Skip to content

[SDY][PadForDivisibility] Add DotGeneralOpPattern to enforce kZero padding#1359

Merged
copybara-service[bot] merged 1 commit into
mainfrom
test_914952809
May 21, 2026
Merged

[SDY][PadForDivisibility] Add DotGeneralOpPattern to enforce kZero padding#1359
copybara-service[bot] merged 1 commit into
mainfrom
test_914952809

Conversation

@copybara-service
Copy link
Copy Markdown

@copybara-service copybara-service Bot commented May 20, 2026

[SDY][PadForDivisibility] Add DotGeneralOpPattern to enforce kZero padding

  • Implement DotGeneralOpPattern to enforce kZero padding on operands of stablehlo.dot_general operations.
  • Exclude DotGeneralOp from GenericOpPattern to avoid ambiguity.
  • Add MLIR test case verifying correct padding propagation for dot_general.

Also adds a general util...

  • Create ensurePadding helper to insert select operations when cached padding differs from required padding.

…dding

- Implement DotGeneralOpPattern to enforce kZero padding on operands of stablehlo.dot_general operations.
- Exclude DotGeneralOp from GenericOpPattern to avoid ambiguity.
- Add MLIR test case verifying correct padding propagation for dot_general.

Also adds a general util...
- Create ensurePadding helper to insert select operations when cached padding differs from required padding.

PiperOrigin-RevId: 919119485
@copybara-service copybara-service Bot merged commit 02dad3d into main May 21, 2026
@copybara-service copybara-service Bot deleted the test_914952809 branch May 21, 2026 17:16
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