Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

don't generate lazy iota/eye/tri/delta with omnistaging #4535

Merged
merged 5 commits into from
Oct 14, 2020

Conversation

mattjj
Copy link
Collaborator

@mattjj mattjj commented Oct 11, 2020

This is disabling, but not removing the device constant part of #1668 when omnistaging is enabled. We can do this because after #3370 and #4038 omnistaging removes the need for lazy device constants in a jitted context. (They could still in principle be useful in an op-by-op context, but the power:weight isn't worthwhile anymore.)

After this change, with omnistaging enabled the only parts of the lazy sublanguage that are used are those to do with broadcasts and transposes.

#4536 is a follow-up that removes the lazy device constant stuff (rather than just disabling it when omnistaging is enabled).

fyi @jblespiau

@mattjj mattjj changed the title don't generate lazy iota/eye/tri/delta omnistaging don't generate lazy iota/eye/tri/delta with omnistaging Oct 11, 2020
@mattjj mattjj marked this pull request as ready for review October 11, 2020 04:56
@mattjj mattjj added the pull ready Ready for copybara import and testing label Oct 13, 2020
@mattjj mattjj force-pushed the lazy-simplification branch from 43be942 to 990dc57 Compare October 14, 2020 21:52
@copybara-service copybara-service bot merged commit fb01f59 into master Oct 14, 2020
@mattjj mattjj deleted the lazy-simplification branch October 14, 2020 22:32
copybara-service bot pushed a commit that referenced this pull request Oct 15, 2020
…an internal test.

PiperOrigin-RevId: 337216255
copybara-service bot pushed a commit that referenced this pull request Oct 15, 2020
…an internal test.

PiperOrigin-RevId: 337219426
mattjj added a commit that referenced this pull request Oct 22, 2020
mattjj added a commit that referenced this pull request Oct 22, 2020
mattjj added a commit that referenced this pull request Oct 22, 2020
copybara-service bot pushed a commit that referenced this pull request Oct 23, 2020
--
ced333d by Matthew Johnson <mattjj@google.com>:

redo #4535 lazy simplification

PiperOrigin-RevId: 338603510
copybara-service bot pushed a commit that referenced this pull request Oct 23, 2020
--
ced333d by Matthew Johnson <mattjj@google.com>:

redo #4535 lazy simplification

PiperOrigin-RevId: 338603510
copybara-service bot pushed a commit that referenced this pull request Oct 23, 2020
--
ced333d by Matthew Johnson <mattjj@google.com>:

redo #4535 lazy simplification

PiperOrigin-RevId: 338606348
copybara-service bot pushed a commit that referenced this pull request Oct 23, 2020
--
ced333d by Matthew Johnson <mattjj@google.com>:

redo #4535 lazy simplification

PiperOrigin-RevId: 338667969
copybara-service bot pushed a commit that referenced this pull request Oct 23, 2020
--
ced333d by Matthew Johnson <mattjj@google.com>:

redo #4535 lazy simplification

PiperOrigin-RevId: 338667969
copybara-service bot pushed a commit that referenced this pull request Oct 23, 2020
--
ced333d by Matthew Johnson <mattjj@google.com>:

redo #4535 lazy simplification

PiperOrigin-RevId: 338670328
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla: yes pull ready Ready for copybara import and testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants