From a37a36c21a0c94b0a7b356ff37df19d4fa89267b Mon Sep 17 00:00:00 2001
From: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Date: Tue, 27 Aug 2024 06:50:06 -0700
Subject: [PATCH] Hide non-necessary symbols from shared object (#1136)

Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
---
 transformer_engine/common/CMakeLists.txt                | 4 ++++
 transformer_engine/common/libtransformer_engine.version | 4 ++++
 2 files changed, 8 insertions(+)
 create mode 100644 transformer_engine/common/libtransformer_engine.version

diff --git a/transformer_engine/common/CMakeLists.txt b/transformer_engine/common/CMakeLists.txt
index 7fab75dca0..58bd4f828c 100644
--- a/transformer_engine/common/CMakeLists.txt
+++ b/transformer_engine/common/CMakeLists.txt
@@ -15,6 +15,10 @@ if (CMAKE_BUILD_TYPE STREQUAL "Debug")
   set(CMAKE_CUDA_FLAGS_DEBUG "${CMAKE_CUDA_FLAGS_DEBUG} -G")
 endif()
 
+# Hide non-necessary symbols in shared object.
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wl,--version-script=${CMAKE_CURRENT_SOURCE_DIR}/libtransformer_engine.version")
+set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Wl,--version-script=${CMAKE_CURRENT_SOURCE_DIR}/libtransformer_engine.version")
+
 # Transformer Engine library
 project(transformer_engine LANGUAGES CUDA CXX)
 
diff --git a/transformer_engine/common/libtransformer_engine.version b/transformer_engine/common/libtransformer_engine.version
new file mode 100644
index 0000000000..0683ec01ea
--- /dev/null
+++ b/transformer_engine/common/libtransformer_engine.version
@@ -0,0 +1,4 @@
+{
+	global: *nvte*; *transformer_engine*;
+	local: *;
+};