use PYTORCH_ROCM_ARCH to set GLOO_ROCM_ARCH (#40170)

Summary:
Previously it used the default arch set which may or may not coincide with the user's.

Pull Request resolved: pytorch/pytorch#40170

Differential Revision: D22400866

Pulled By: xw285cornell

fbshipit-source-id: 222ba684782024fa68f37bf7d4fdab9a2389bdea

cmake/Dependencies.cmake

@@ -1207,6 +1207,9 @@ if(USE_GLOO)
     set(__BUILD_BENCHMARK ${BUILD_BENCHMARK})
     set(BUILD_TEST OFF)
     set(BUILD_BENCHMARK OFF)
+    if(USE_ROCM)
+      set(ENV{GLOO_ROCM_ARCH} "${PYTORCH_ROCM_ARCH}")
+    endif()
     if(NOT USE_SYSTEM_GLOO)
       add_subdirectory(${CMAKE_CURRENT_LIST_DIR}/../third_party/gloo)
     else()