llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/lib/Dialect/Torch/Utils
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
..
CMakeLists.txt Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
TorchUpstream.cpp Add type promotion code to refine types. 2021-10-29 11:17:39 -04:00
Utils.cpp Fix type promotion code for scalar only operations 2022-05-07 10:37:21 -04:00