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
Yi Zhang 28be6511d2 Fix type promotion code for scalar only operations 
Fix the type promotion code for scalar only operation to return
TorchType which is the type tracked in ValueKnowledge.scalarType.

- Fix `getPromotedResultScalarType` to return Torch type.
- Add `getBuiltInTypeForTorchScalar` helper to convert scalar type
to builtin type before passing to the next level type promotion
helper `updateResultTypeState`.
- Add `setScalarType` helper to make setting ValueKnowledge.scalarType
  easier.
 		2022-05-07 10:37:21 -04: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 Fix type promotion code for scalar only operations 2022-05-07 10:37:21 -04: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