diff options
| author | Johannes Gäßler <johannesg@5d6.de> | 2024-02-13 12:38:37 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-13 12:38:37 +0100 |
| commit | ad014bba97ef6ef6c3e2f78b2fc463e91ae94579 (patch) | |
| tree | e18dbcf3b0f596fd73694ac58b4aa3cf25864c10 | |
| parent | 49cc1f7d67de2da99f3ac185f9ff1319b7bf35f8 (diff) | |
make: add error message for bad CUDA version (#5444)
* make: add error message for bad CUDA version
* Update Makefile
Co-authored-by: Jared Van Bortel <cebtenzzre@gmail.com>
---------
Co-authored-by: Jared Van Bortel <cebtenzzre@gmail.com>
| -rw-r--r-- | Makefile | 8 |
1 files changed, 8 insertions, 0 deletions
@@ -569,6 +569,14 @@ $(info I CC: $(shell $(CC) --version | head -n 1)) $(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 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) +endif # CUDA_POWER_ARCH +endif # CUDA_DOCKER_ARCH +endif # eq ($(shell echo "$(CUDA_VERSION) < 11.7" | bc),1) endif # LLAMA_CUBLAS $(info ) |