Update

[ghstack-poisoned]
pytorch · huydhn · Jul 17, 2025 · Jul 17, 2025 · Jul 17, 2025 · Jul 17, 2025
commit 444df0faa1902563f8913f68aa128f3f3e34d659
diff --git a/.ci/pytorch/build.sh b/.ci/pytorch/build.sh
@@ -309,13 +309,21 @@ else
     if [[ -n "${BUILD_ADDITIONAL_PACKAGES}" ]]; then
       if [[ "${BUILD_ADDITIONAL_PACKAGES}" == *vision* ]]; then
         install_torchvision
-      elif [[ "${BUILD_ADDITIONAL_PACKAGES}" == *audio* ]]; then
+      fi
+
+      if [[ "${BUILD_ADDITIONAL_PACKAGES}" == *audio* ]]; then
         install_torchaudio
-      elif [[ "${BUILD_ADDITIONAL_PACKAGES}" == *text* ]]; then
+      fi
+
+      if [[ "${BUILD_ADDITIONAL_PACKAGES}" == *text* ]]; then
         install_torchtext
-      elif [[ "${BUILD_ADDITIONAL_PACKAGES}" == *torchrec* || "${BUILD_ADDITIONAL_PACKAGES}" == *fbgemm* ]]; then
+      fi
+
+      if [[ "${BUILD_ADDITIONAL_PACKAGES}" == *torchrec* || "${BUILD_ADDITIONAL_PACKAGES}" == *fbgemm* ]]; then
         install_torchrec_and_fbgemm
-      elif [[ "${BUILD_ADDITIONAL_PACKAGES}" == *torchao* ]]; then
+      fi
+
+      if [[ "${BUILD_ADDITIONAL_PACKAGES}" == *torchao* ]]; then
         install_torchao
       fi
     fi

@@ -1670,11 +1670,7 @@ elif [[ "${TEST_CONFIG}" == verify_cachebench ]]; then
   checkout_install_torchbench nanogpt
   PYTHONPATH=$(pwd)/torchbench test_verify_cachebench
 elif [[ "${TEST_CONFIG}" == *torchbench* ]]; then
-  if [[ "${TEST_CONFIG}" == *cpu* ]]; then
-    install_torchaudio
-  else
-    install_torchaudio
-  fi
+  install_torchaudio
   install_torchvision
   install_torchao
   id=$((SHARD_NUMBER-1))