llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/external
History
Yi Zhang 603e068e45 E2e implementation for `aten.cat`,`aten.gather`, `aten.bmm` 
Also contains the following changes:
- Remove derefineOp canonicalizer because it's not safe.
- Support for optional tensor and list tensors in reduceOpVariant. This
only works for some special detected and easy to handle cases. For list,
it covers the case list is got from a `ListConstruct`. For optional, it
covers the case optional is constructed from a `DerefineOp`.
- Remove the `inferReturnTypes` for `FromBuiltinTensorOp` because it's
not safe to deduce types from the input. For example, a built-in tensor
of i8 could be converted to si8 or ui8. It's better to let the user
specify the return type explicitly.
 		2021-09-22 19:15:01 -04:00
..
llvm-project@6e60bb6883 E2e implementation for `aten.cat`,`aten.gather`, `aten.bmm` 2021-09-22 19:15:01 -04:00
torch-mlir E2e implementation for `aten.cat`,`aten.gather`, `aten.bmm` 2021-09-22 19:15:01 -04:00