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

[TOSA] Extend Torch to TOSA reduction ops legalization #3710

Merged
merged 1 commit into from
Sep 16, 2024

Commits on Sep 16, 2024

  1. [TOSA] Extend Torch to TOSA reduction ops legalization

    - Add Torch to TOSA legalization for the following reduction ops:
      + aten.min.dim
      + aten.min
      + aten.max
      + aten.prod
      + aten.prod.dim_int
      + aten.all.dim
    - Add dtype casting support for reduce sum and prod ops
    - Extend aten.max.dim legalization to a template to support
    aten.min.dim legalization
    - Update end-to-end tests sets in xfail_sets.py
    - Update basic.mlir with new legalized ops
    
    Signed-off-by: Justin Ngo <justin.ngo@arm.com>
    Change-Id: I854dd6c0c55e570c1fb7242f20c85cf64d6e7fe0
    justin-ngo-arm committed Sep 16, 2024
    Configuration menu
    Copy the full SHA
    3fdc969 View commit details
    Browse the repository at this point in the history