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

[inductor] Lower diagonal, diagonal_copy and diagonal_scatter #103755

Closed
wants to merge 9 commits into from

Commits on Jun 16, 2023

  1. [inductor] Decompose diagonal_copy and diagonal_scatter

    This decomposes `diagonal_copy` into `as_strided_copy` and `diagonal_scatter`
    into `as_strided_scatter`.
    
    Currently these are decomposed into mutations, which can't be used by inductor
    because the `_scatter` and `_copy` variants are output by functionalization to
    remove mutations from the graph. So by re-introducing mutations in their
    decomposition, we break the assumptions of the graph.
    
    [ghstack-poisoned]
    peterbell10 committed Jun 16, 2023
    Configuration menu
    Copy the full SHA
    39cbbc8 View commit details
    Browse the repository at this point in the history
  2. Update on "[inductor] Lower diagonal, diagonal_copy and diagonal_scat…

    …ter"
    
    Currently these are decomposed into `as_strided`, which forces a buffer to be
    realized. Instead, this lowers them into a native inductor view node and so
    doesn't require any buffers to be realized.
    
    [ghstack-poisoned]
    peterbell10 committed Jun 16, 2023
    Configuration menu
    Copy the full SHA
    eda593e View commit details
    Browse the repository at this point in the history
  3. Update on "[inductor] Lower diagonal, diagonal_copy and diagonal_scat…

    …ter"
    
    Currently these are decomposed into `as_strided`, which forces a buffer to be
    realized. Instead, this lowers them into a native inductor view node and so
    doesn't require any buffers to be realized.
    
    [ghstack-poisoned]
    peterbell10 committed Jun 16, 2023
    Configuration menu
    Copy the full SHA
    1a43b63 View commit details
    Browse the repository at this point in the history
  4. Update on "[inductor] Lower diagonal, diagonal_copy and diagonal_scat…

    …ter"
    
    Currently these are decomposed into `as_strided`, which forces a buffer to be
    realized. Instead, this lowers them into a native inductor view node and so
    doesn't require any buffers to be realized.
    
    [ghstack-poisoned]
    peterbell10 committed Jun 16, 2023
    Configuration menu
    Copy the full SHA
    5d8b9f9 View commit details
    Browse the repository at this point in the history

Commits on Jun 17, 2023

  1. Update on "[inductor] Lower diagonal, diagonal_copy and diagonal_scat…

    …ter"
    
    Currently these are decomposed into `as_strided`, which forces a buffer to be
    realized. Instead, this lowers them into a native inductor view node and so
    doesn't require any buffers to be realized.
    
    [ghstack-poisoned]
    peterbell10 committed Jun 17, 2023
    Configuration menu
    Copy the full SHA
    c24eb21 View commit details
    Browse the repository at this point in the history

Commits on Jun 19, 2023

  1. Update on "[inductor] Lower diagonal, diagonal_copy and diagonal_scat…

    …ter"
    
    Currently these are decomposed into `as_strided`, which forces a buffer to be
    realized. Instead, this lowers them into a native inductor view node and so
    doesn't require any buffers to be realized.
    
    [ghstack-poisoned]
    peterbell10 committed Jun 19, 2023
    Configuration menu
    Copy the full SHA
    598e1e0 View commit details
    Browse the repository at this point in the history

Commits on Jun 20, 2023

  1. Update on "[inductor] Lower diagonal, diagonal_copy and diagonal_scat…

    …ter"
    
    Currently these are decomposed into `as_strided`, which forces a buffer to be
    realized. Instead, this lowers them into a native inductor view node and so
    doesn't require any buffers to be realized.
    
    [ghstack-poisoned]
    peterbell10 committed Jun 20, 2023
    Configuration menu
    Copy the full SHA
    e7a0890 View commit details
    Browse the repository at this point in the history

Commits on Jun 21, 2023

  1. Update on "[inductor] Lower diagonal, diagonal_copy and diagonal_scat…

    …ter"
    
    Currently these are decomposed into `as_strided`, which forces a buffer to be
    realized. Instead, this lowers them into a native inductor view node and so
    doesn't require any buffers to be realized.
    
    [ghstack-poisoned]
    peterbell10 committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    cb4f71a View commit details
    Browse the repository at this point in the history
  2. Fix small error on "[inductor] Lower diagonal, diagonal_copy and diag…

    …onal_scatter"
    
    Currently these are decomposed into `as_strided`, which forces a buffer to be
    realized. Instead, this lowers them into a native inductor view node and so
    doesn't require any buffers to be realized.
    
    cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx ipiszy ngimel yf225 chenyang78
    
    [ghstack-poisoned]
    peterbell10 committed Jun 21, 2023
    Configuration menu
    Copy the full SHA
    bbc046e View commit details
    Browse the repository at this point in the history