Fixed #675: Can now query (the global) CUDA module loading mode

eyalroz · Sep 6, 2024 · 0f445ac · 0f445ac
1 parent fc415b6
commit 0f445ac
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/cuda/api/module.hpp b/src/cuda/api/module.hpp
@@ -91,6 +91,17 @@ module_t create(
 	ContiguousContainer  module_data);
 ///@}
 
+#if CUDA_VERSION >= 12030
+using loading_mode_t = CUmoduleLoadingMode;
+
+inline loading_mode_t loading_mode() {
+	loading_mode_t result;
+	auto status = cuModuleGetLoadingMode(&result);
+	throw_if_error_lazy(status, "Failed obtaining CUDA module loading mode");
+	return result;
+}
+#endif
+
 } // namespace module
 
 /**