Fix setup scripts for OSS ROCm (pytorch#1000)

Summary:
Pull Request resolved: facebookresearch/FBGEMM#1000

- Sometimes we would like to test FBGEMM OSS inside FB devservers.  However, the scripts currently assume an OSS CI environment with Ubuntu, and so the setup scripts fail with error, which this diff fixes.

X-link: pytorch#3909

Reviewed By: sryap

Differential Revision: D72255342

Pulled By: q10

fbshipit-source-id: c6361e8439269b553fdc55ebf01a04d283fea873

Author: q10

.github/scripts/fbgemm_gpu_test.bash

@@ -363,7 +363,7 @@ test_setup_conda_environment () {
     install_cuda  "${env_name}" "${pytorch_variant_version}"                                            || return 1
     install_cudnn "${env_name}" "${HOME}/cudnn-${pytorch_variant_version}" "${pytorch_variant_version}" || return 1
   # Install ROCm tools and runtime
-  elif [ "$pytorch_variant_type" == "rocm" ]; then
+  elif [[ "$pytorch_variant_type" == "rocm" ]] && ! [[ "$(hostname)" =~ ^.*facebook.com$ ]]; then
     install_rocm_ubuntu     "${env_name}" "${pytorch_variant_version}"  || return 1
   fi
 

.github/scripts/utils_system.bash

@@ -133,7 +133,8 @@ print_gpu_info () {
   if [[ "${BUILD_FROM_NOVA}" != '1' ]]; then
     echo "################################################################################"
     echo "[INFO] Printing general display info ..."
-    install_system_packages lshw
+    install_system_packages hostname lshw
+    print_exec hostname
     print_exec sudo lshw -C display
   fi