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/eager_mode
History
Ashay Rane 9208bf0eb6
llvm: bump tag to e1318078 (#781) 
The updated LLVM code includes a patch to create bfloat16 array
attributes, thus enabling a different patch to torch-mlir to flesh out
support for the bfloat16 type.
 		2022-04-26 12:27:51 -07:00
..
CMakeLists.txt 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
__init__.py 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
ir_building.py llvm: bump tag to e1318078 (#781) 2022-04-26 12:27:51 -07:00
torch_mlir_dispatch.py Fix test fails due to upstream PyTorch change (#793) 2022-04-25 12:34:32 -07:00
torch_mlir_tensor.py Add link to forum post describing `__torch_dispatch__` 2022-04-01 10:10:43 -07:00