try generating fat binaries for cuda

This commit is contained in:
Concedo 2024-01-20 22:05:08 +08:00
parent a137b6b9ff
commit 425849f387

View file

@ -147,7 +147,7 @@ ifdef CUDA_DOCKER_ARCH
NVCCFLAGS += -Wno-deprecated-gpu-targets -arch=$(CUDA_DOCKER_ARCH) NVCCFLAGS += -Wno-deprecated-gpu-targets -arch=$(CUDA_DOCKER_ARCH)
else else
ifdef LLAMA_PORTABLE ifdef LLAMA_PORTABLE
NVCCFLAGS += -Wno-deprecated-gpu-targets -arch=all-major NVCCFLAGS += -Wno-deprecated-gpu-targets -arch=all
else else
NVCCFLAGS += -arch=native NVCCFLAGS += -arch=native
endif endif