From 6ee20d3bfc0a9273af2b18fabf60070e37e35991 Mon Sep 17 00:00:00 2001 From: Cai Yudong Date: Fri, 14 Jun 2024 11:35:50 +0800 Subject: [PATCH] [skip ci] Update script prepare_gpu_build.sh for nightly build (#648) Signed-off-by: Cai Yudong --- scripts/prepare_gpu_build.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/prepare_gpu_build.sh b/scripts/prepare_gpu_build.sh index b37d0dda4..931938457 100755 --- a/scripts/prepare_gpu_build.sh +++ b/scripts/prepare_gpu_build.sh @@ -3,5 +3,6 @@ # This script is to modify CMakeLists.txt to build knowhere for certain GPU serials # GTX 1060 - sm_61 # GTX 1660 - sm_75 -sed 's/set(CMAKE_CUDA_ARCHITECTURES .*$/set(CMAKE_CUDA_ARCHITECTURES \"61-real;75-real\")/' CMakeLists.txt > tmp +# GTX 2080 SUPER - sm_75 +sed 's/set(CMAKE_CUDA_ARCHITECTURES .*$/set(CMAKE_CUDA_ARCHITECTURES \"75-real\")/' CMakeLists.txt > tmp mv tmp CMakeLists.txt