diff --git a/llama_cublas.go b/llama_cublas.go index 32bb072..efd1519 100644 --- a/llama_cublas.go +++ b/llama_cublas.go @@ -4,6 +4,6 @@ package llama /* -#cgo LDFLAGS: -lcublas +#cgo LDFLAGS: -lcublas -lcudart -L/usr/local/cuda/lib64/ */ import "C"