Update Makefile

Co-authored-by: Jared Van Bortel <cebtenzzre@gmail.com>
This commit is contained in:
Johannes Gäßler 2024-02-11 11:29:01 +01:00 committed by GitHub
parent d02245cd76
commit f67ba9a674
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -570,7 +570,7 @@ $(info I CXX: $(shell $(CXX) --version | head -n 1))
ifdef LLAMA_CUBLAS
$(info I NVCC: $(shell $(NVCC) --version | tail -n 1))
CUDA_VERSION := $(shell nvcc --version | grep -oP 'release (\K[0-9]+\.[0-9])')
ifeq ($(shell echo "$(CUDA_VERSION) < 11.7" | bc),1)
ifeq ($(shell awk -v "v=$(CUDA_VERSION)" 'BEGIN { print (v < 11.7) }'),1)
ifndef CUDA_DOCKER_ARCH
ifndef CUDA_POWER_ARCH
$(error I ERROR: For CUDA versions < 11.7 a target CUDA architecture must be explicitly provided via CUDA_DOCKER_ARCH)