mirror of https://github.com/llvm/torch-mlir
0a788e0467
As titled, create a new decomposition for `aten.fmod.Tensor` to `aten.div`, `aten.trunc`, `aten.mul` and `aten.sub`. Note that we only use `aten.trunc` for floating point operations. This further gets decomposed to `aten.where` etc. by other existing decompositions. This decomposition now makes TOSA pass for a simple model with `aten.fmod` while it makes `stablehlo` fail. For now, we disallow this decomposition for `stablehlo` --------- Co-authored-by: Srinath Avadhanula <srinath.avadhanula@getcruise.com> |
||
---|---|---|
.. | ||
jit_ir_common | ||
ltc | ||
onnx_c_importer | ||
pt1 | ||
CMakeLists.txt |