torch-mlir/python
Gaurav Shukla eddc09aa55 [TORCH][MLIR] Add E2E support for `aten.eq` and `aten.lt` ops
- Added E2E support for `aten.eq.Tensor` and `aten.lt.Tensor` ops. Both
  the operands are expected to be of the same type, i.e., type promotion
  is not addressed as a part of this commit.
- Added E2E support for `aten.eq.Scalar` and `aten.lt.Scalar` ops.
  Tensor operand type to Scalar operand type promotion has not been
  handled in this commit.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
2021-12-16 18:47:22 +05:30
..
test Make error reporting a bit better. 2021-10-28 13:20:16 -07:00
torch_mlir/dialects [TORCH][MLIR] Add E2E support for `aten.eq` and `aten.lt` ops 2021-12-16 18:47:22 +05:30
torch_mlir_e2e_test Add support for passing & returning memref of bool types 2021-12-09 00:23:38 +05:30
CMakeLists.txt Enable python extensions when building out of tree (#363) 2021-10-27 17:04:12 -07:00
TorchMLIRModule.cpp Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00