build : add compile option to force use of MMQ kernels

This commit is contained in:
Georgi Gerganov 2023-10-27 13:21:04 +03:00
parent a4e15a36e4
commit 49af767fad
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
3 changed files with 11 additions and 0 deletions

View file

@ -397,6 +397,9 @@ endif # CUDA_DOCKER_ARCH
ifdef LLAMA_CUDA_FORCE_DMMV
NVCCFLAGS += -DGGML_CUDA_FORCE_DMMV
endif # LLAMA_CUDA_FORCE_DMMV
ifdef LLAMA_CUDA_FORCE_MMQ
NVCCFLAGS += -DGGML_CUDA_FORCE_MMQ
endif # LLAMA_CUDA_FORCE_MMQ
ifdef LLAMA_CUDA_DMMV_X
NVCCFLAGS += -DGGML_CUDA_DMMV_X=$(LLAMA_CUDA_DMMV_X)
else