llama_cublas.go 125 B

123456789
  1. //go:build cublas
  2. // +build cublas
  3. package llama
  4. /*
  5. #cgo LDFLAGS: -lcublas -lcudart -L/usr/local/cuda/lib64/
  6. */
  7. import "C"