conversion tritonGPUToLLVM TargetInfoBase decoupling

Author: toudefu

third_party/iluvatar/backend/flagtree_backend_specialization/include/flagtree_spec.h

@@ -2,3 +2,4 @@
 #include "triton/Analysis/iluvatar_Membar.h"
 #include "triton/Analysis/iluvatar_Utility.h"
 #include "triton/Conversion/TritonGPUToLLVM/iluvatar_ElementwiseOpToLLVMBase.h"
+#include "triton/Conversion/TritonGPUToLLVM/iluvatar_TargetInfoBase.h"

third_party/iluvatar/backend/flagtree_backend_specialization/include/triton/Conversion/TritonGPUToLLVM/iluvatar_TargetInfoBase.h

@@ -0,0 +1,6 @@
+#ifndef ILUVATAR_TRITON_CONVERSION_TRITONGPU_TO_LLVM_TARGETINFOBASE_H
+#define ILUVATAR_TRITON_CONVERSION_TRITONGPU_TO_LLVM_TARGETINFOBASE_H
+
+#define FLAGTREE_SPEC_TargetInfoBase_function
+
+#endif // ILUVATAR_TRITON_CONVERSION_TRITONGPU_TO_LLVM_TARGETINFOBASE_H

third_party/iluvatar/include/triton/Conversion/TritonGPUToLLVM/TargetInfoBase.h

@@ -13,10 +13,18 @@ class TargetInfoBase {
   virtual Value ballot(ConversionPatternRewriter &rewriter, Location loc,
                        Type type, Value cmp) const = 0;
 
+#ifndef FLAGTREE_SPEC_TargetInfoBase_function
+  virtual void storeShared(ConversionPatternRewriter &rewriter, Location loc,
+                           Value ptr, Value val, Value pred) const = 0;
+  virtual Value loadShared(ConversionPatternRewriter &rewriter, Location loc,
+                           const TypeConverter *converter, Value ptr,
+                           Type elemTy, Value pred) const = 0;
+#else
   virtual Value storeShared(ConversionPatternRewriter &rewriter, Location loc,
                             Value ptr, Value val, Value pred) const = 0;
   virtual Value loadShared(ConversionPatternRewriter &rewriter, Location loc,
                            Value ptr, Type elemTy, Value pred) const = 0;
+#endif
 
   virtual Value shuffleXor(ConversionPatternRewriter &rewriter, Location loc,
                            Value val, int i) const = 0;