Add class member function support for addSet

Add class member function support

Author: jiaolu

include/llvm-dialects/Dialect/Visitor.h

@@ -414,6 +414,12 @@ class VisitorBuilder : private detail::VisitorBuilderBase {
     return *this;
   }
 
+  template <typename... OpTs>
+  VisitorBuilder &addSet(void (PayloadT::*fn)(llvm::Instruction &)) {
+    addSetMemberCase(detail::VisitorKey::opSet<OpTs...>(), fn);
+    return *this;
+  }
+
   VisitorBuilder &addSet(const OpSet &opSet,
                          VisitorResult (*fn)(PayloadT &,
                                              llvm::Instruction &I)) {
@@ -512,6 +518,15 @@ class VisitorBuilder : private detail::VisitorBuilderBase {
     VisitorBuilderBase::add(key, &VisitorBuilder::setForwarder<ReturnT>, data);
   }
 
+  template <typename ReturnT>
+  void addSetMemberCase(detail::VisitorKey key,
+                        ReturnT (PayloadT::*fn)(llvm::Instruction &)) {
+    detail::VisitorCallbackData data{};
+    static_assert(sizeof(fn) <= sizeof(data.data));
+    memcpy(&data.data, &fn, sizeof(fn));
+    VisitorBuilderBase::add(key, &VisitorBuilder::setForwarder<ReturnT>, data);
+  }
+
   template <typename OpT, typename ReturnT>
   void addMemberFnCase(detail::VisitorKey key, ReturnT (PayloadT::*fn)(OpT &)) {
     detail::VisitorCallbackData data{};