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/test
History
Jae Hoon (Antonio) Kim 0af55781ae
Propagate device data names (#1157) 
* Propagate device data names

* Address PR comment

* Add example usage

* Add test for device data names

* Make TorchMlirComputation fields protected

* Add lazy backend device data name unit tests

* Disable lazy backend tests if LTC is disabled

* Add comments
 		2022-08-16 09:30:22 -04:00
..
compile_api torch_mlir.compile: Allow ignoring traced shapes 2022-08-04 10:18:34 -07:00
eager_mode Clean up Autogen (#1112) 2022-07-30 09:40:02 -04:00
lazy_backend Propagate device data names (#1157) 2022-08-16 09:30:22 -04:00
torchscript_e2e_test E2E framework: Report tensor dtype in summary 2022-08-05 10:05:18 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
annotations-sugar.py Update test to pass with newer versions of tanh (#990) 2022-06-28 20:28:13 -07:00
lit.cfg.py Propagate device data names (#1157) 2022-08-16 09:30:22 -04:00
lit.site.cfg.py.in Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00