mirror of https://github.com/llvm/torch-mlir
28be6511d2
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. |
||
---|---|---|
.. | ||
torch | ||
TorchBinding.td |