From 558e5d2fc3d610be6d926cf5ec1eba32e4f121cb Mon Sep 17 00:00:00 2001 From: as Date: Wed, 2 Apr 2025 17:51:29 +0800 Subject: [PATCH] chore: add ggml-cuda into library path for go bindings --- Makefile | 2 +- bindings/go/Makefile | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/Makefile b/Makefile index dbda58acc44..be87c336491 100644 --- a/Makefile +++ b/Makefile @@ -4,7 +4,7 @@ .PHONY: build build: - cmake -B build + cmake -B build -DGGML_CUDA=$(GGML_CUDA) cmake --build build --config Release # download a few audio samples into folder "./samples": diff --git a/bindings/go/Makefile b/bindings/go/Makefile index edcc0166b74..07155b6e593 100644 --- a/bindings/go/Makefile +++ b/bindings/go/Makefile @@ -16,10 +16,12 @@ MODELS_DIR := models EXAMPLES_DIR := $(wildcard examples/*) INCLUDE_PATH := $(abspath ../../include):$(abspath ../../ggml/include) LIBRARY_PATH := $(abspath ../../${BUILD_DIR}/src:$(abspath ../../${BUILD_DIR}/ggml/src)) +CUDA_PATH := $(CUDA_PATH:-/usr/local/cuda) ifeq ($(GGML_CUDA),1) + LIBRARY_PATH := $(LIBRARY_PATH):$(abspath ../../${BUILD_DIR}/ggml/src/ggml-cuda) LIBRARY_PATH := $(LIBRARY_PATH):$(CUDA_PATH)/targets/$(UNAME_M)-linux/lib/ - BUILD_FLAGS := -ldflags "-extldflags '-lcudart -lcuda -lcublas'" + BUILD_FLAGS := -ldflags "-extldflags '-lggml-cuda -lcudart -lcuda -lcublas'" endif ifeq ($(UNAME_S),Darwin) @@ -30,6 +32,7 @@ all: clean whisper examples whisper: mkdir cmake -S ../.. -B ../../${BUILD_DIR} \ + -DGGML_CUDA=$(GGML_CUDA) \ -DCMAKE_BUILD_TYPE=Release \ -DBUILD_SHARED_LIBS=OFF cmake --build ../../${BUILD_DIR} --target whisper