[MLIR][TOSA] Fix converting tosa.clamp and tosa.relu to linalg
authorjungpark-mlir <jungwook.park@amd.com>
Mon, 11 Jul 2022 17:18:09 +0000 (17:18 +0000)
committerKrzysztof Drewniak <Krzysztof.Drewniak@amd.com>
Mon, 11 Jul 2022 17:18:47 +0000 (17:18 +0000)
commit6e8e91a7b63c51f487ddfbe2d6b2372ea1310faf
tree78709be4b3515ff8b1ba3e202a6482df1d8d4130
parentf20a3cbefd81943d0c2df9b8cf171896e78e38c1
[MLIR][TOSA] Fix converting tosa.clamp and tosa.relu to linalg

Tosa to Linalg conversion crashes when input tensor is a float type other than fp32.
Because tosa.clamp and tosa.reluN have fp32 min/max attribute which is converted as arith.constant with the attribute type.
This commit fixes the crash by correctly setting the float constant type from the input tensor.

Reviewed By: eric-k256

Differential Revision: https://reviews.llvm.org/D128630
mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp
mlir/test/Conversion/TosaToLinalg/tosa-to-linalg.mlir