llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/python/torch_mlir
History
Qiang Fu f7c7bb800c
Add non-default dtype support for a few elementwise math ops. (#687) 
* fix type inference
* fix Torch2Linalg conversion
* add test cases
 		2022-03-23 13:35:43 -07:00
..
dialects Add non-default dtype support for a few elementwise math ops. (#687) 2022-03-23 13:35:43 -07:00
eager_mode This PR implements an eager mode backend for PyTorch through the torch-mlir framework. This is accomplished by overriding the `__torch_dispatch__` class method on wrapper subclass `TorchMLIRTensor(torch.Tensor)`. 2022-03-22 14:42:57 -07:00