go : fix CUDA build (#2416)

* Fixed go cuda bindings building

* Added note to go bindings Readme to build using cuda support

---------

Co-authored-by: Binozo <redacted>

diff --git a/bindings/go/Makefile b/bindings/go/Makefile
index 20a054c116fe2637a6bbe1708a96f4fc303146d6..ca39de20555f1c62f507d347d1df055bc97653c9 100644 (file)
--- a/bindings/go/Makefile
+++ b/bindings/go/Makefile
@@ -17,6 +17,11 @@ EXAMPLES_DIR := $(wildcard examples/*)
 INCLUDE_PATH := $(abspath ../../include):$(abspath ../../ggml/include)
 LIBRARY_PATH := $(abspath ../..)
 
+ifeq ($(GGML_CUDA),1)
+       LIBRARY_PATH := $(LIBRARY_PATH):$(CUDA_PATH)/targets/$(UNAME_M)-linux/lib/
+       BUILD_FLAGS := -ldflags "-extldflags '-lcudart -lcuda -lcublas'"
+endif
+
 ifeq ($(UNAME_S),Darwin)
        EXT_LDFLAGS := -framework Foundation -framework Metal -framework MetalKit
 endif

diff --git a/bindings/go/README.md b/bindings/go/README.md
index 1968cfd24700be7acdf5d54b9044b520e1425911..6958ede80f25afdcf33e53633d90faed7641d707 100644 (file)
--- a/bindings/go/README.md
+++ b/bindings/go/README.md
@@ -62,6 +62,12 @@ This will compile a static `libwhisper.a` in a `build` folder, download a model
 make examples
 ```
 
+To build using cuda support add `GGML_CUDA=1`:
+
+```bash
+GGML_CUDA=1 make examples
+```
+
 The examples are placed in the `build` directory. Once built, you can download all the models with the following command:
 
 ```bash