diff --git a/Makefile b/Makefile index 5647fd0..19b5b91 100644 --- a/Makefile +++ b/Makefile @@ -47,8 +47,8 @@ CC_cublasLt110 := -gencode arch=compute_75,code=sm_75 CC_cublasLt110 += -gencode arch=compute_80,code=sm_80 CC_cublasLt111 := -gencode arch=compute_75,code=sm_75 -#CC_cublasLt111 += -gencode arch=compute_80,code=sm_80 -#CC_cublasLt111 += -gencode arch=compute_86,code=sm_86 +CC_cublasLt111 += -gencode arch=compute_80,code=sm_80 +CC_cublasLt111 += -gencode arch=compute_86,code=sm_86 CC_ADA_HOPPER := -gencode arch=compute_89,code=sm_89 CC_ADA_HOPPER += -gencode arch=compute_90,code=sm_90 diff --git a/cuda_install.sh b/cuda_install.sh index 678f7ca..eae2cb5 100644 --- a/cuda_install.sh +++ b/cuda_install.sh @@ -13,6 +13,7 @@ URL117=https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installer URL118=https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run URL120=https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run URL121=https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run +URL122=https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run CUDA_VERSION=$1 @@ -65,6 +66,9 @@ if [[ -n "$CUDA_VERSION" ]]; then elif [[ "$CUDA_VERSION" -eq "121" ]]; then URL=$URL121 FOLDER=cuda-12.1 + elif [[ "$CUDA_VERSION" -eq "122" ]]; then + URL=$URL122 + FOLDER=cuda-12.2 else echo "argument error: No cuda version passed as input. Choose among versions 92 to 121" fi