|
@@ -597,7 +597,7 @@ $(info I CC: $(shell $(CC) --version | head -n 1))
|
|
|
$(info I CXX: $(shell $(CXX) --version | head -n 1))
|
|
$(info I CXX: $(shell $(CXX) --version | head -n 1))
|
|
|
ifdef LLAMA_CUBLAS
|
|
ifdef LLAMA_CUBLAS
|
|
|
$(info I NVCC: $(shell $(NVCC) --version | tail -n 1))
|
|
$(info I NVCC: $(shell $(NVCC) --version | tail -n 1))
|
|
|
-CUDA_VERSION := $(shell nvcc --version | grep -oP 'release (\K[0-9]+\.[0-9])')
|
|
|
|
|
|
|
+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)
|
|
ifeq ($(shell awk -v "v=$(CUDA_VERSION)" 'BEGIN { print (v < 11.7) }'),1)
|
|
|
ifndef CUDA_DOCKER_ARCH
|
|
ifndef CUDA_DOCKER_ARCH
|
|
|
ifndef CUDA_POWER_ARCH
|
|
ifndef CUDA_POWER_ARCH
|