Test build time for sm 8.0 only (#3068)

Summary: Pull Request resolved: #3068 Differential Revision: D62157280
pytorch · Sep 4, 2024 · 2c8f48b · 2c8f48b
1 parent 225ac16
commit 2c8f48b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/scripts/fbgemm_gpu_build.bash b/.github/scripts/fbgemm_gpu_build.bash
@@ -199,7 +199,7 @@ __configure_fbgemm_gpu_build_cuda () {
       #   https://github.com/NVIDIA/nvbench/discussions/129
       #   https://github.com/vllm-project/vllm/blob/main/CMakeLists.txt#L187
       #   https://github.com/NVIDIA/cutlass/blob/main/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized.hpp#L224
-      local arch_list="7.0;8.0;9.0;9.0a"
+      local arch_list="7.0;8.0;9.0a"
     else
       local arch_list="7.0;8.0"
     fi

diff --git a/.github/scripts/nova_dir.bash b/.github/scripts/nova_dir.bash
@@ -16,5 +16,5 @@ export BUILD_FROM_NOVA=1
 ## Overwrite existing ENV VAR in Nova
 if [[ "$CONDA_ENV" != "" ]]; then export CONDA_RUN="conda run --no-capture-output -p ${CONDA_ENV}" && echo "$CONDA_RUN"; fi
 if [[ "$CU_VERSION" == "cu118" ]]; then export TORCH_CUDA_ARCH_LIST='7.0;8.0' && echo "$TORCH_CUDA_ARCH_LIST"; fi
-if [[ "$CU_VERSION" == "cu121" ]]; then export TORCH_CUDA_ARCH_LIST='7.0;8.0;9.0;9.0a' && echo "$TORCH_CUDA_ARCH_LIST"; fi
-if [[ "$CU_VERSION" == "cu124" ]]; then export TORCH_CUDA_ARCH_LIST='7.0;8.0;9.0;9.0a' && echo "$TORCH_CUDA_ARCH_LIST"; fi
+if [[ "$CU_VERSION" == "cu121" ]]; then export TORCH_CUDA_ARCH_LIST='7.0;8.0;9.0a' && echo "$TORCH_CUDA_ARCH_LIST"; fi
+if [[ "$CU_VERSION" == "cu124" ]]; then export TORCH_CUDA_ARCH_LIST='7.0;8.0;9.0a' && echo "$TORCH_CUDA_ARCH_LIST"; fi