llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test
History
Ramiro Leal-Cavazos 96f90efd16
Add shape info to `rand_like` + support for `dtype` flag (#851) 
The op `aten.rand_like` was missing a shape function, unit tests, and
the `dtype` argument was being ignored in its decomposition. This
commit fixes all three things.
 		2022-05-12 16:00:59 -07:00
..
Conversion [MLIR][TORCH] E2E support for [ge|ceil].float, [ge|ne|gt].float_int op 2022-05-05 21:48:35 +05:30
Dialect Add shape info to `rand_like` + support for `dtype` flag (#851) 2022-05-12 16:00:59 -07:00
RefBackend [Torch][Linalg] Add basic support for RNG 2022-01-31 18:56:42 -05:00
python Add tracing suport to `torch_mlir.compile`. 2022-05-03 09:08:40 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
lit.cfg.py Add tool substitutions to support out-of-tree builds 2021-10-07 21:16:43 -07:00
lit.site.cfg.py.in Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00