make IMPORTED rocshmem library

Author: alextmagro

transformer_engine/common/CMakeLists.txt

@@ -395,13 +395,23 @@ else()
     else()
       set(ROCSHMEM_HOME "/opt/rocm" CACHE STRING "Location of ROCSHMEM installation (default)")
     endif()
+    set(ROCSHMEM_LIBRARY_PATH "${ROCSHMEM_HOME}/lib/librocshmem.a")
+    if (EXISTS ${ROCSHMEM_LIBRARY_PATH})
+      add_library(rocshmem STATIC IMPORTED)
+      set_target_properties(rocshmem PROPERTIES
+        IMPORTED_LOCATION "${ROCSHMEM_LIBRARY_PATH}"
+        IMPORTED_LINK_INTERFACE_LANGUAGES "CXX"
+      )
+    else()
+      message(FATAL_ERROR "ROCSHMEM library not found at ${ROCSHMEM_LIBRARY_PATH}. PLease set ROCSHMEM_HOME.")
+    endif()
     target_link_options(transformer_engine PRIVATE
         -fgpu-rdc
     )
     target_link_libraries(transformer_engine PUBLIC 
         -Wl,--whole-archive
         rocshmemapi
-        "${ROCSHMEM_HOME}/lib/librocshmem.a"
+        rocshmem
         -Wl,--no-whole-archive
     )
   endif()