[fbgemm_gpu] Remove errant RPATHs from binaries

- Remove errant RPATHs from binaries (#3098)
pytorch · Sep 23, 2024 · b9dc5ec · b9dc5ec
1 parent 2889f3a
commit b9dc5ec
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/.github/scripts/fbgemm_gpu_build.bash b/.github/scripts/fbgemm_gpu_build.bash
@@ -416,6 +416,9 @@ __print_library_infos () {
 
     echo "[CHECK] Listing out external shared libraries linked:"
     print_exec ldd "${library}"
+
+    echo "[CHECK] Displaying ELF information:"
+    print_exec readelf -d "${library}"
     echo "################################################################################"
     echo ""
     echo ""

diff --git a/.github/scripts/utils_build.bash b/.github/scripts/utils_build.bash
@@ -297,6 +297,7 @@ install_build_tools () {
     ncurses \
     ninja \
     openblas \
+    patchelf \
     scikit-build \
     wheel) || return 1
 

diff --git a/fbgemm_gpu/CMakeLists.txt b/fbgemm_gpu/CMakeLists.txt
@@ -112,3 +112,7 @@ if(NOT FBGEMM_CPU_ONLY AND NOT USE_ROCM)
   #
   add_subdirectory(experimental/gen_ai)
 endif()
+
+
+install(CODE "execute_process(COMMAND sh ../.github/scripts/fbgemm_gpu_postbuild.bash")
+# ; patchelf --remove-rpath *.so)