Skip to content

[opt] Support lifting operations through OneHotSelect in more circumstances#4395

Draft
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_931343361
Draft

[opt] Support lifting operations through OneHotSelect in more circumstances#4395
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_931343361

Conversation

@copybara-service

Copy link
Copy Markdown

[opt] Support lifting operations through OneHotSelect in more circumstances

This change extends SelectLiftingPass to lift operations through OneHotSelect, including comparisons and shifts. It introduces a safety analysis based on what's known about the OneHotSelect's selector's state (zero-hot, one-hot, multi-hot) and the behavior of the operations.

…tances

This change extends SelectLiftingPass to lift operations through OneHotSelect, including comparisons and shifts. It introduces a safety analysis based on what's known about the OneHotSelect's selector's state (zero-hot, one-hot, multi-hot) and the behavior of the operations.

PiperOrigin-RevId: 931343361
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