Skip to content

Commit

Permalink
cgo: use O3
Browse files Browse the repository at this point in the history
  • Loading branch information
mxyng committed Jan 30, 2025
1 parent 5d75d83 commit bea1f1f
Show file tree
Hide file tree
Showing 11 changed files with 13 additions and 1 deletion.
1 change: 1 addition & 0 deletions discover/gpu.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
package discover

/*
#cgo CPPFLAGS: -O3
#cgo linux LDFLAGS: -lrt -lpthread -ldl -lstdc++ -lm
#cgo windows LDFLAGS: -lpthread
Expand Down
1 change: 1 addition & 0 deletions discover/gpu_darwin.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ package discover

/*
#cgo CFLAGS: -x objective-c
#cgo CPPFLAGS: -O3
#cgo LDFLAGS: -framework Foundation -framework CoreGraphics -framework Metal
#include "gpu_info_darwin.h"
*/
Expand Down
1 change: 1 addition & 0 deletions llama/llama.cpp/examples/llava/llava.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
package llava

// #cgo CXXFLAGS: -std=c++11
// #cgo CPPFLAGS: -O3
// #cgo CPPFLAGS: -I${SRCDIR}/../../include -I${SRCDIR}/../../common
// #cgo CPPFLAGS: -I${SRCDIR}/../../../../ml/backend/ggml/ggml/include
import "C"
1 change: 1 addition & 0 deletions llama/llama.cpp/src/llama.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
package llama

// #cgo CXXFLAGS: -std=c++17
// #cgo CPPFLAGS: -O3
// #cgo CPPFLAGS: -I${SRCDIR}/../include
// #cgo CPPFLAGS: -I${SRCDIR}/../../../ml/backend/ggml/ggml/include
// #cgo windows CPPFLAGS: -D_WIN32_WINNT=0x0602
Expand Down
1 change: 1 addition & 0 deletions llama/llama.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ package llama
/*
#cgo CFLAGS: -std=c11
#cgo CXXFLAGS: -std=c++17
#cgo CPPFLAGS: -O3
#cgo CPPFLAGS: -I${SRCDIR}/llama.cpp/include
#cgo CPPFLAGS: -I${SRCDIR}/llama.cpp/common
#cgo CPPFLAGS: -I${SRCDIR}/llama.cpp/examples/llava
Expand Down
1 change: 1 addition & 0 deletions ml/backend/ggml/ggml/src/ggml-blas/blas.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
package blas

// #cgo CXXFLAGS: -std=c++11
// #cgo CPPFLAGS: -O3
// #cgo CPPFLAGS: -DGGML_USE_BLAS
// #cgo CPPFLAGS: -I${SRCDIR}/.. -I${SRCDIR}/../../include
// #cgo darwin,arm64 CPPFLAGS: -DGGML_BLAS_USE_ACCELERATE -DACCELERATE_NEW_LAPACK -DACCELERATE_LAPACK_ILP64
Expand Down
4 changes: 3 additions & 1 deletion ml/backend/ggml/ggml/src/ggml-cpu/cpu.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,11 @@ package cpu

// #cgo CFLAGS: -Wno-implicit-function-declaration
// #cgo CXXFLAGS: -std=c++17
// #cgo CPPFLAGS: -I${SRCDIR}/amx -I${SRCDIR}/llamafile -I${SRCDIR}/.. -I${SRCDIR}/../../include
// #cgo CPPFLAGS: -O3
// #cgo CPPFLAGS: -DGGML_USE_LLAMAFILE
// #cgo CPPFLAGS: -I${SRCDIR}/amx -I${SRCDIR}/llamafile -I${SRCDIR}/.. -I${SRCDIR}/../../include
// #cgo linux CPPFLAGS: -D_GNU_SOURCE
// #cgo arm64 CPPFLAGS: -DGGML_USE_AARCH64
// #cgo darwin,arm64 CPPFLAGS: -DGGML_USE_ACCELERATE -DACCELERATE_NEW_LAPACK -DACCELERATE_LAPACK_ILP64
// #cgo darwin,arm64 LDFLAGS: -framework Accelerate
import "C"
Expand Down
1 change: 1 addition & 0 deletions ml/backend/ggml/ggml/src/ggml-cpu/llamafile/llamafile.go
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
package llamafile

// #cgo CXXFLAGS: -std=c++17
// #cgo CPPFLAGS: -O3
// #cgo CPPFLAGS: -I${SRCDIR}/.. -I${SRCDIR}/../.. -I${SRCDIR}/../../../include
import "C"
1 change: 1 addition & 0 deletions ml/backend/ggml/ggml/src/ggml-metal/metal.go
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ package metal

//go:generate sh -c "{ echo // Code generated $(date). DO NOT EDIT.; sed -e '/__embed_ggml-common.h__/r ../ggml-common.h' -e '/__embed_ggml-common.h__/d' -e '/#include \"ggml-metal-impl.h\"/r ggml-metal-impl.h' -e '/#include \"ggml-metal-impl.h\"/d' ggml-metal.metal; } >ggml-metal-embed.metal"

// #cgo CPPFLAGS: -O3
// #cgo CPPFLAGS: -DGGML_METAL_EMBED_LIBRARY -I.. -I../../include
// #cgo LDFLAGS: -framework Metal -framework MetalKit
import "C"
1 change: 1 addition & 0 deletions ml/backend/ggml/ggml/src/ggml.go
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
package ggml

// #cgo CXXFLAGS: -std=c++17
// #cgo CPPFLAGS: -O3
// #cgo CPPFLAGS: -DNDEBUG -DGGML_USE_CPU
// #cgo CPPFLAGS: -I${SRCDIR}/../include -I${SRCDIR}/ggml-cpu
// #cgo windows LDFLAGS: -lmsvcrt -static -static-libgcc -static-libstdc++
Expand Down
1 change: 1 addition & 0 deletions ml/backend/ggml/ggml/src/ggml_darwin_arm64.go
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
package ggml

// #cgo CPPFLAGS: -O3
// #cgo CPPFLAGS: -DGGML_USE_METAL -DGGML_USE_BLAS
// #cgo LDFLAGS: -framework Foundation
import "C"
Expand Down

0 comments on commit bea1f1f

Please sign in to comment.