[mlir][sparse] convenience util to test for "sparse" op

[cota]

Thanks

mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensor.h

 Level getCOOStart(SparseTensorEncodingAttr enc);
 
 /// Helpers to setup a COO type.
 RankedTensorType getCOOFromTypeWithOrdering(RankedTensorType src,
                                             AffineMap ordering, bool ordered);
 
 RankedTensorType getCOOFromType(RankedTensorType src, bool ordered);
 
+/// Returns true iff MLIR operand has any sparse operand or result.
+inline bool hasAnySparseOperandOrResult(Operation *op) {
+  bool anySparseIn = llvm::any_of(op->getOperands().getTypes(), [](Type t) {
+    return getSparseTensorEncoding(t) != nullptr;
+  });
+  bool anySparseOut = llvm::any_of(op->getResults().getTypes(), [](Type t) {
+    return getSparseTensorEncoding(t) != nullptr;
+  });
+  return anySparseIn || anySparseOut;
+}
+
 //
 // Reordering.
 //
 
 // This CPP guard is to disable deprecation warnings for the LLVM
 // build-bot, while making it easy to re-enable it for local development.
 #if 0
 #define DEPRECATED                                                             \