Closed Todorbsc closed 4 weeks ago
The StableHLO spec says that stablehlo.abs
should return a f64
tensor in this case, so I don't know why are inferring a complex<f64>
tensor? maybe our implementation of Base.abs(::TracedRArray)
is wrong when we set the returning MLIR type?
CC @mofeing
In the following simple example, I tried to compile (
@code_hlo
) a loss function that computes the distance (element-wise) between two arrays:params
andexpected
. In the first case, the compilation results to an error when using complex element types. However, in the second case, there are no errors when the arrays element type are real numbers.