From e86db9381a441bb0b4e41f81124dcf81625e3d9c Mon Sep 17 00:00:00 2001 From: jmorganca Date: Sat, 18 May 2024 23:53:29 -0700 Subject: [PATCH] `avx2` should only add `avx2` --- llama/llama.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama/llama.go b/llama/llama.go index 9fb52022..78333519 100644 --- a/llama/llama.go +++ b/llama/llama.go @@ -9,8 +9,8 @@ package llama // #cgo windows LDFLAGS: -lmsvcrt // #cgo avx CFLAGS: -mavx // #cgo avx CXXFLAGS: -mavx -// #cgo avx2 CFLAGS: -mavx -mavx2 -mfma -// #cgo avx2 CXXFLAGS: -mavx -mavx2 -mfma +// #cgo avx2 CFLAGS: -mavx2 -mfma +// #cgo avx2 CXXFLAGS: -mavx2 -mfma // #cgo cuda CFLAGS: -DGGML_USE_CUDA -DGGML_CUDA_DMMV_X=32 -DGGML_CUDA_PEER_MAX_BATCH_SIZE=128 -DGGML_MULTIPLATFORM -DGGML_CUDA_MMV_Y=1 -DGGML_BUILD=1 // #cgo cuda CXXFLAGS: -std=c++11 -DGGML_USE_CUDA -DGGML_CUDA_DMMV_X=32 -DGGML_CUDA_PEER_MAX_BATCH_SIZE=128 -DGGML_MULTIPLATFORM -DGGML_CUDA_MMV_Y=1 -DGGML_BUILD=1 // #cgo rocm CXXFLAGS: -std=c++11 -DGGML_USE_CUDA -DGGML_USE_HIPBLAS -DGGML_CUDA_DMMV_X=32 -DGGML_CUDA_PEER_MAX_BATCH_SIZE=128 -DGGML_MULTIPLATFORM -DGGML_CUDA_MMV_Y=1 -DGGML_BUILD=1