Optional linking with build tags

README.md

@@ -25,16 +25,28 @@ cd go-llama.cpp
 make libbinding.a
 ```
 
-To build with OpenBLAS, for example:
+Now you can run the example with:
+
+```
+LIBRARY_PATH=$PWD C_INCLUDE_PATH=$PWD go run ./examples -m "/model/path/here" -t 14
+```
+
+## OpenBLAS accelleration
+
+To build and run with OpenBLAS, for example:
 
 ```
 CMAKE_ARGS="-DLLAMA_OPENBLAS=ON" make libbinding.a
+LIBRARY_PATH=$PWD C_INCLUDE_PATH=$PWD go run -tags openblas ./examples -m "/model/path/here" -t 14
 ```
 
-Now you can run the example with:
+## GPU
+
+To build with CuBLAS:
 
 ```
-LIBRARY_PATH=$PWD C_INCLUDE_PATH=$PWD go run ./examples -m "/model/path/here" -t 14
+CMAKE_ARGS="-DLLAMA_CUBLAS=ON" make libbinding.a
+LIBRARY_PATH=$PWD C_INCLUDE_PATH=$PWD go run -tags cublas ./examples -m "/model/path/here" -t 14
 ```
 
 Enjoy!

llama_cublas.go

@@ -0,0 +1,9 @@
+//go:build cublas
+// +build cublas
+
+package llama
+
+/*
+#cgo LDFLAGS: -lcublas
+*/
+import "C"

llama_openblas.go

@@ -0,0 +1,9 @@
+//go:build openblas
+// +build openblas
+
+package llama
+
+/*
+#cgo LDFLAGS: -lopenblas
+*/
+import "C"