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/dialects/torch
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
..
importer Fix type promotion code for scalar only operations 2022-05-07 10:37:21 -04:00
__init__.py Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00