diff options
| author | slaren <2141330+slaren@users.noreply.github.com> | 2023-04-19 11:22:45 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-04-19 11:22:45 +0200 |
| commit | 8944a1329648c57bb7d66851170938230587a52c (patch) | |
| tree | a4b759e0a1ee93fa542dc708fc938b660ff13eac /Makefile | |
| parent | 66674012389f9537140044290f8517bc4a5e0b74 (diff) | |
Add NVIDIA cuBLAS support (#1044)
Diffstat (limited to 'Makefile')
| -rw-r--r-- | Makefile | 4 |
1 files changed, 4 insertions, 0 deletions
@@ -97,6 +97,10 @@ ifdef LLAMA_OPENBLAS CFLAGS += -DGGML_USE_OPENBLAS -I/usr/local/include/openblas LDFLAGS += -lopenblas endif +ifdef LLAMA_CUBLAS + CFLAGS += -DGGML_USE_CUBLAS -I/usr/local/cuda/include + LDFLAGS += -lcublas_static -lculibos -lcudart_static -lcublasLt_static -lpthread -ldl -L/usr/local/cuda/lib64 +endif ifdef LLAMA_GPROF CFLAGS += -pg CXXFLAGS += -pg |