From 4d163a4d9b3b7f0b628610589d8a3aa63ccb71a1 Mon Sep 17 00:00:00 2001
From: Violet <c01368481@gmail.com>
Date: Wed, 30 Jul 2025 16:34:21 -0700
Subject: [PATCH] Implement `cuModuleGetGlobal_v2` (#454)

---
 zluda/src/impl/module.rs | 9 +++++++++
 zluda/src/lib.rs         | 1 +
 2 files changed, 10 insertions(+)

diff --git a/zluda/src/impl/module.rs b/zluda/src/impl/module.rs
index 981ab06..5e8f371 100644
--- a/zluda/src/impl/module.rs
+++ b/zluda/src/impl/module.rs
@@ -105,6 +105,15 @@ pub(crate) fn get_function(
     unsafe { hipModuleGetFunction(hfunc, hmod.base, name) }
 }
 
+pub(crate) fn get_global_v2(
+    dptr: *mut hipDeviceptr_t,
+    bytes: *mut usize,
+    hmod: &Module,
+    name: *const ::core::ffi::c_char,
+) -> hipError_t {
+    unsafe { hipModuleGetGlobal(dptr, bytes, hmod.base, name) }
+}
+
 pub(crate) fn get_loading_mode(mode: &mut cuda_types::cuda::CUmoduleLoadingMode) -> CUresult {
     *mode = cuda_types::cuda::CUmoduleLoadingMode::CU_MODULE_EAGER_LOADING;
     Ok(())
diff --git a/zluda/src/lib.rs b/zluda/src/lib.rs
index 1fdc1ea..b90d4d6 100644
--- a/zluda/src/lib.rs
+++ b/zluda/src/lib.rs
@@ -112,6 +112,7 @@ cuda_macros::cuda_function_declarations!(
             cuMemsetD32_v2,
             cuMemsetD8_v2,
             cuModuleGetFunction,
+            cuModuleGetGlobal_v2,
             cuModuleGetLoadingMode,
             cuModuleLoadData,
             cuModuleUnload,