mirror of https://github.com/llvm/torch-mlir
d803ab4eeb
To keep things simple in shape functions, `Scalar` inputs are considered `float`s. This means that when inserting the shape functions into the IR, we must cast any `!torch.number`s into `float`s so that the operand type matches the expected type in the shape function. This commit adds the cast from `Scalar` to `float`. |
||
---|---|---|
.. | ||
Torch | ||
TorchConversion |