mirror of https://github.com/llvm/torch-mlir
f1e3701caf
* use lhs tensor's element type as compute type when rhs is scalar. * previously `a != 1.0`(a is a fp32 tensor) will lowering to `%6 = stablehlo.compare EQ, %4, %5, FLOAT : (tensor<2x5xf64>, tensor<2x5xf64>) -> tensor<2x5xi1>` * now it will lowering to `%6 = stablehlo.compare EQ, %4, %5, FLOAT : (tensor<2x5xf32>, tensor<2x5xf32>) -> tensor<2x5xi1>` |
||
---|---|---|
.. | ||
Basic.cpp | ||
CMakeLists.txt | ||
GatherScatter.cpp | ||
Linear.cpp | ||
Pooling.cpp | ||
PopulatePatterns.h | ||
Reduction.cpp | ||
Rng.cpp | ||
StablehloLegalizeUtils.cpp | ||
TorchToStablehlo.cpp | ||
Utils.cpp | ||
Utils.h | ||
ViewLike.cpp |