[mlir][Vector] Add check of supported reduction kind for ScanOp.

[jacquesguan]

Summary

This patch adds check of supported reduction kind for ScanOp to avoid using and/or/xor for floating point type.

mlir/lib/Dialect/Vector/IR/VectorOps.cpp

LogicalResult ScanOp::verify() {
   }
   if (llvm::any_of(llvm::zip(initialValueShapes, expectedShape),
                    [](std::tuple<int64_t, int64_t> s) {
                      return std::get<0>(s) != std::get<1>(s);
                    })) {
     return emitOpError("incompatible input/initial value shapes");
   }
 
+  // Verify supported reduction kind.
+  Type eltType = getDestType().getElementType();
+  if (!isSupportedCombiningKind(getKind(), eltType))
+    return emitOpError("unsupported reduction type ")
+           << eltType << " for kind '" << stringifyCombiningKind(getKind())
+           << "'";
+
   return success();
 }
 
 void mlir::vector::populateVectorToVectorCanonicalizationPatterns(
     RewritePatternSet &patterns) {
   patterns
       .add<CreateMaskFolder, MaskedLoadFolder, MaskedStoreFolder, GatherFolder,
            ScatterFolder, ExpandLoadFolder, CompressStoreFolder,

mlir/test/Dialect/Vector/invalid.mlir

func @scan_incompatible_shapes(%arg0: vector<2x3xi32>, %arg1: vector<5xi32>) -> vector<2x3xi32> {
   // expected-error@+1 {{incompatible input/initial value shapes}}
   %0:2 = vector.scan <add>, %arg0, %arg1 {inclusive = true, reduction_dim = 0} :
     vector<2x3xi32>, vector<5xi32>
   return %0#0 : vector<2x3xi32>
 }
 
 // -----
 
+func @scan_unsupported_kind(%arg0: vector<2x3xf32>, %arg1: vector<3xf32>) -> vector<2x3xf32> {
+  // expected-error@+1 {{'vector.scan' op unsupported reduction type 'f32' for kind 'xor'}}
+  %0:2 = vector.scan <xor>, %arg0, %arg1 {inclusive = true, reduction_dim = 0} :
+    vector<2x3xf32>, vector<3xf32>
+  return %0#0 : vector<2x3xf32>
+}
+
+// -----
+
 func @invalid_splat(%v : f32) {
   // expected-error@+1 {{invalid kind of type specified}}
   vector.splat %v : memref<8xf32>
   return
 }
 
 // -----