Spaces:
Sleeping
Sleeping
byte-6174
commited on
make : update cuBLAS build both x86 and aarch64 (#1015)
Browse filesmake cuBLAS compilation compatible with x86 as well as aarch64.
Makefile
CHANGED
|
@@ -160,9 +160,9 @@ ifdef WHISPER_OPENBLAS
|
|
| 160 |
endif
|
| 161 |
|
| 162 |
ifdef WHISPER_CUBLAS
|
| 163 |
-
CFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets
|
| 164 |
-
CXXFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets
|
| 165 |
-
LDFLAGS += -lcublas -lculibos -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64 -L/opt/cuda/lib64 -L$(CUDA_PATH)/targets
|
| 166 |
WHISPER_OBJ += ggml-cuda.o
|
| 167 |
NVCC = nvcc
|
| 168 |
NVCCFLAGS = --forward-unknown-to-host-compiler -arch=any
|
|
|
|
| 160 |
endif
|
| 161 |
|
| 162 |
ifdef WHISPER_CUBLAS
|
| 163 |
+
CFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets/$(UNAME_M)-linux/include
|
| 164 |
+
CXXFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets/$(UNAME_M)-linux/include
|
| 165 |
+
LDFLAGS += -lcublas -lculibos -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64 -L/opt/cuda/lib64 -L$(CUDA_PATH)/targets/$(UNAME_M)-linux/lib
|
| 166 |
WHISPER_OBJ += ggml-cuda.o
|
| 167 |
NVCC = nvcc
|
| 168 |
NVCCFLAGS = --forward-unknown-to-host-compiler -arch=any
|