[ROCm] Require rocm_smi package (pytorch#175648)

Fixes pytorch#158725

This is essentially @AngryLoki  patch:
https://github.com/gentoo/gentoo/blob/8cdbe88fa388ce264d1d70047222fcad190fec3d/sci-ml/caffe2/files/caffe2-2.9.0-rocm-distributed-link.patch

Pull Request resolved: pytorch#175648
Approved by: https://github.com/jeffdaily, https://github.com/mlazos

(cherry picked from commit 9bff6e1)

Author: naromero77amd

cmake/Dependencies.cmake

@@ -1089,6 +1089,13 @@ if(USE_ROCM)
       )
     endif()
 
+    # ROCM-SMI needed to support symmetric memory
+    if(USE_DISTRIBUTED AND UNIX)
+      list(APPEND Caffe2_PUBLIC_HIP_DEPENDENCY_LIBS
+        rocm_smi64
+      )
+    endif()
+
     # ---[ Kernel asserts
     # Kernel asserts is disabled for ROCm by default.
     # It can be turned on by turning on the env USE_ROCM_KERNEL_ASSERT to the build system.

cmake/public/LoadHIP.cmake

@@ -198,6 +198,7 @@ if(HIP_FOUND)
   if(UNIX)
     find_package_and_print_version(rccl)
     find_package_and_print_version(hsa-runtime64 REQUIRED)
+    find_package_and_print_version(rocm_smi REQUIRED)
   endif()
 
   # Optional components.