[llvm] [LV] Introduce isLegalMaskedLoadOrStore (NFC) (PR #195242)

[David Sherwood via llvm-commits]

================
@@ -650,15 +650,9 @@ class VFSelectionContext {
   /// of FP operations.
   bool useOrderedReductions(const RecurrenceDescriptor &RdxDesc) const;
 
-  /// Returns true if the target machine supports masked store operation
-  /// for the given \p DataType and kind of access to \p Ptr.
-  bool isLegalMaskedStore(Type *DataType, Value *Ptr, Align Alignment,
-                          unsigned AddressSpace) const;
-
-  /// Returns true if the target machine supports masked load operation
-  /// for the given \p DataType and kind of access to \p Ptr.
-  bool isLegalMaskedLoad(Type *DataType, Value *Ptr, Align Alignment,
-                         unsigned AddressSpace) const;
+  /// Returns true if the target machine can represent \p I as a masked load
+  /// or store.
+  bool isLegalMaskedLoadOrStore(Instruction *I, ElementCount VF) const;
----------------
david-arm wrote:

nit: Is it worth making this consistent with `isLegalGatherOrScatter` and passing in `Value *V` instead?

https://github.com/llvm/llvm-project/pull/195242