Skip to content

Commit

Permalink
[TOSA] Expand Torch to TOSA legalization coverage (llvm#3827)
Browse files Browse the repository at this point in the history 
- Add/Extend Torch to TOSA legalization for the following ops:
  + Add aten.threshold_backward
  + Fix aten.threshold
  + Re-implement aten.broadcast_to using tosa.reshape and tosa.tile
  + Add support for rank 0 index for aten.index_select
  + Fix aten.index_put.hacked_twin
  + Add aten.uniform
  + Add aten.logical_and
- Update xfail_sets.py with new e2e results
- Add LIT tests to basic.mlir for newly added ops


Change-Id: I8910564a049d18293284fe2e55e82bc1d2cf10e3

Signed-off-by: Justin Ngo <[email protected]>
  • Loading branch information
@justin-ngo-arm
justin-ngo-arm authored Oct 30, 2024
1 parent a6292f3 commit 4dd213b
Show file tree
Hide file tree
Showing 3 changed files with 399 additions and 224 deletions.
Loading

0 comments on commit 4dd213b

Please sign in to comment.