Add address space attribute to LLVMIR's GlobalOp.

PiperOrigin-RevId: 270012505

1 files changed, 6 insertions, 3 deletions

diff --git a/mlir/lib/Target/LLVMIR/ModuleTranslation.cpp b/mlir/lib/Target/LLVMIR/ModuleTranslation.cpp
index cbdc6c27ea1..5b5c06a1e86 100644
--- a/mlir/lib/Target/LLVMIR/ModuleTranslation.cpp
+++ b/mlir/lib/Target/LLVMIR/ModuleTranslation.cpp
@@ -326,9 +326,12 @@ void ModuleTranslation::convertGlobals() {
       cst = getLLVMConstant(type, op.value(), op.getLoc());
     }
 
-    auto *var = new llvm::GlobalVariable(*llvmModule, type, op.constant(),
-                                         llvm::GlobalValue::InternalLinkage,
-                                         cst, op.sym_name());
+    auto addrSpace = op.addr_space().getLimitedValue();
+    auto *var = new llvm::GlobalVariable(
+        *llvmModule, type, op.constant(), llvm::GlobalValue::InternalLinkage,
+        cst, op.sym_name(), /*InsertBefore=*/nullptr,
+        llvm::GlobalValue::NotThreadLocal, addrSpace);
+
     globalsMapping.try_emplace(op, var);
   }
 }