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.

Sign up for GitHub

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

Jump to bottom

Add Scalarization Patterns for AtenToDtypeOp, AtenNegOp, AtenRemainderTensorOp #3861

Merged
merged 11 commits into from
Nov 12, 2024
Merged

Add Scalarization Patterns for AtenToDtypeOp, AtenNegOp, AtenRemainderTensorOp #3861

merged 11 commits into from
Nov 12, 2024
+302 −30

Commits on Nov 7, 2024

  1. Add propagation for aten.to.dtype

    @zjgarvey
    zjgarvey committed Nov 7, 2024
    Configuration menu
    Copy the full SHA
    f20f7bd View commit details
    Browse the repository at this point in the history

  2. Merge remote-tracking branch 'upstream/main' into prop_todtype

    @zjgarvey
    zjgarvey committed Nov 7, 2024
    Configuration menu
    Copy the full SHA
    13f6a68 View commit details
    Browse the repository at this point in the history

Commits on Nov 9, 2024

  1. Add scalarization for NegTensor and RemainderTensor 

    Add canonicalization to remove complex dynamic shape calculations from unflatten ops whenever these could just be -1
Add lowering to arith for neg.int and remainder.int ops
    @zjgarvey
    zjgarvey committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    7873430 View commit details
    Browse the repository at this point in the history

  2. fix failing lit tests

    @zjgarvey
    zjgarvey committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    9c7e3aa View commit details
    Browse the repository at this point in the history

  3. don't do something dumb

    @zjgarvey
    zjgarvey committed Nov 9, 2024
    Configuration menu
    Copy the full SHA
    db19a25 View commit details
    Browse the repository at this point in the history

Commits on Nov 11, 2024

  1. add better handling of rank 0 tensors; fix a bug in unflatten decomp

    @zjgarvey
    zjgarvey committed Nov 11, 2024
    Configuration menu
    Copy the full SHA
    441beaa View commit details
    Browse the repository at this point in the history

  2. add item ops to worklist if they get consumed by slice ops

    @zjgarvey
    zjgarvey committed Nov 11, 2024
    Configuration menu
    Copy the full SHA
    0c080d2 View commit details
    Browse the repository at this point in the history

  3. move -1 unflatten inference to scalarize shapes

    @zjgarvey
    zjgarvey committed Nov 11, 2024
    Configuration menu
    Copy the full SHA
    5bea9eb View commit details
    Browse the repository at this point in the history

  4. fix lit test

    @zjgarvey
    zjgarvey committed Nov 11, 2024
    Configuration menu
    Copy the full SHA
    5a5adcd View commit details
    Browse the repository at this point in the history

  5. add some tests and remove unused scalar cast ops

    @zjgarvey
    zjgarvey committed Nov 11, 2024
    Configuration menu
    Copy the full SHA
    28299c4 View commit details
    Browse the repository at this point in the history

Commits on Nov 12, 2024

  1. address comments

    @zjgarvey
    zjgarvey committed Nov 12, 2024
    Configuration menu
    Copy the full SHA
    0088a40 View commit details
    Browse the repository at this point in the history