[AOT] Add invariant attribute for memory and global pointer

Signed-off-by: Shen-Ta Hsieh <beststeve@secondstate.io>
DecFox · Nov 1, 2023 · e63e8ee · e63e8ee
1 parent a3a3474
commit e63e8ee
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/lib/aot/compiler.cpp b/lib/aot/compiler.cpp
@@ -453,6 +453,8 @@ struct WasmEdge::AOT::Compiler::CompileContext {
     auto VPtr = Builder.createLoad(
         Int8PtrTy, Builder.createInBoundsGEP1(Int8PtrTy, Array,
                                               LLContext.getInt64(Index)));
+    VPtr.setMetadata(LLContext, LLVM::Core::InvariantGroup,
+                     LLVM::Metadata(LLContext, {}));
     return Builder.createBitCast(VPtr, Int8PtrTy);
   }
   std::pair<LLVM::Type, LLVM::Value> getGlobal(LLVM::Builder &Builder,
@@ -463,6 +465,8 @@ struct WasmEdge::AOT::Compiler::CompileContext {
     auto VPtr = Builder.createLoad(
         Int128PtrTy, Builder.createInBoundsGEP1(Int8PtrTy, Array,
                                                 LLContext.getInt64(Index)));
+    VPtr.setMetadata(LLContext, LLVM::Core::InvariantGroup,
+                     LLVM::Metadata(LLContext, {}));
     auto Ptr = Builder.createBitCast(VPtr, Ty.getPointerTo());
     return {Ty, Ptr};
   }
@@ -492,7 +496,7 @@ struct WasmEdge::AOT::Compiler::CompileContext {
     auto PtrTy = Ty.getPointerTo();
     auto PtrPtrTy = PtrTy.getPointerTo();
     auto IT = Builder.createLoad(IntrinsicsTablePtrTy, IntrinsicsTable);
-    IT.setMetadata(LLContext, LLVM::Core::InvariantLoad,
+    IT.setMetadata(LLContext, LLVM::Core::InvariantGroup,
                    LLVM::Metadata(LLContext, {}));
     auto VPtr =
         Builder.createInBoundsGEP2(IntrinsicsTableTy, IT, LLContext.getInt64(0),

diff --git a/lib/aot/llvm.h b/lib/aot/llvm.h
@@ -87,7 +87,7 @@ class Core {
   static inline unsigned int ReadOnly = 0;
   static inline unsigned int StrictFP = 0;
 
-  static inline unsigned int InvariantLoad = 0;
+  static inline unsigned int InvariantGroup = 0;
 
 private:
   static inline std::once_flag Once;
@@ -162,7 +162,7 @@ class Core {
     ReadOnly = getEnumAttributeKind("readonly"sv);
     StrictFP = getEnumAttributeKind("strictfp"sv);
 
-    InvariantLoad = getMetadataKind("invariant.load");
+    InvariantGroup = getMetadataKind("invariant.group");
   }
 
   template <typename... ArgsT>