byte-6174 commited on
Commit
5d4a28f
·
unverified ·
1 Parent(s): 1807a31

make : update cuBLAS build both x86 and aarch64 (#1015)

Browse files

make cuBLAS compilation compatible with x86 as well as aarch64.

Files changed (1) hide show
  1. Makefile +3 -3
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/x86_64-linux/include
164
- CXXFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include -I/opt/cuda/include -I$(CUDA_PATH)/targets/x86_64-linux/include
165
- LDFLAGS += -lcublas -lculibos -lcudart -lcublasLt -lpthread -ldl -lrt -L/usr/local/cuda/lib64 -L/opt/cuda/lib64 -L$(CUDA_PATH)/targets/x86_64-linux/lib
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