check input shape is static

Author: p-lanza

src/Conversion/ONNXToTOSA/Tensor/Gather.cpp

@@ -46,7 +46,7 @@ class ONNXGatherLoweringToTOSA : public OpConversionPattern<ONNXGatherOp> {
     if (!onnx_mlir::isRankedShapedType(inputType))
       return rewriter.notifyMatchFailure(op, "input is not a ranked tensor");
 
-    if (!hasStaticShape(result.getType()))
+    if (!hasStaticShape(inputType) || !hasStaticShape(result.getType()))
       return rewriter.notifyMatchFailure(op, "dynamic shapes not supported");
 
     auto resultTy = dyn_cast<TensorType>(op.getType());

test/mlir/conversion/onnx_to_tosa/Tensor/Gather.mlir

@@ -175,3 +175,12 @@ func.func @test_gather_dynamic_shape_indices_i32(%arg0 : tensor<?x4xf32>, %indic
 // CHECK-LABEL: test_gather_dynamic_shape_indices_i32
 // CHECK: onnx.Gather
 }
+
+// -----
+
+func.func @test_gather_dynamic_input_static_output(%arg0 : tensor<?x2xf32>, %indices: tensor<?xi64>) -> tensor<1x2xf32> {
+  %0 = "onnx.Gather"(%arg0, %indices) {axis = 0 : si64, onnx_node_name = "/Gather_16"} : (tensor<?x2xf32>, tensor<?xi64>) -> tensor<1x2xf32>
+  "func.return"(%0) : (tensor<1x2xf32>) -> ()
+// CHECK-LABEL: test_gather_dynamic_input_static_output
+// CHECK: onnx.Gather
+}