[AutoBump] Merge with fixes of 93ffe17 (Jun 15)

Author: mgehre-amd

mlir/lib/Conversion/TosaToLinalg/TosaToLinalg.cpp

@@ -2483,7 +2483,10 @@ struct RFFT2dConverter final : public OpConversionPattern<RFFT2dOp> {
     auto loc = rfft2d.getLoc();
     auto input = operands.getInput();
     auto elementType =
-        cast<FloatType>(cast<ShapedType>(input.getType()).getElementType());
+        dyn_cast<FloatType>(cast<ShapedType>(input.getType()).getElementType());
+    if (!elementType)
+      return rewriter.notifyMatchFailure(rfft2d,
+                                         "only supports float element types");
 
     // Compute the output type and set of dynamic sizes
     llvm::SmallVector<Value> dynamicSizes;

mlir/test/Conversion/TosaToLinalg/tosa-to-linalg-invalid.mlir

@@ -36,3 +36,12 @@ func.func @clamp_on_large_int(%arg0: tensor<1xui64>) -> tensor<1xui64> {
   %0 = tosa.clamp %arg0 {min_int = -1 : i64, max_int = 5 : i64, min_fp = 1.0 : f32, max_fp = 5.0 : f32} : (tensor<1xui64>) -> tensor<1xui64>
   return %0 : tensor<1xui64>
 }
+
+// -----
+
+// CHECK-LABEL: @rfft2d_with_non_float_type
+func.func @rfft2d_with_non_float_type(%arg0 : tensor<1x1x1xi32>) -> (tensor<1x1x1xi32>, tensor<1x1x1xi32>) {
+  // expected-error@+1 {{failed to legalize operation 'tosa.rfft2d'}}
+  %real, %imag = tosa.rfft2d %arg0 : (tensor<1x1x1xi32>) -> (tensor<1x1x1xi32>, tensor<1x1x1xi32>)
+  return %real, %imag : tensor<1x1x1xi32>, tensor<1x1x1xi32>
+}