avx flags

succinctlabs · Sep 25, 2024 · 0a4fbc9 · 0a4fbc9
1 parent dbdc26b
commit 0a4fbc9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/.gitignore b/.gitignore
@@ -15,4 +15,5 @@ pgo-data.profdata
 COMMIT_HASH
 
 # Benchmark results
-**/.csv
+**/.csv
+**/*.csv
diff --git a/eval.sh b/eval.sh
@@ -44,10 +44,14 @@ else
   GPU_EXISTS=false
 fi
 
-# Set the compilation flags.
-# if [ "$GPU_EXISTS" = false ]; then
-#   export RUSTFLAGS='-C target-cpu=native'
-# fi
+# Check for AVX-512 support
+if lscpu | grep -q avx512; then
+  # If AVX-512 is supported, add the specific features to RUSTFLAGS
+  export RUSTFLAGS="-C target-cpu=native -C target-feature=+avx512ifma,+avx512vl"
+else
+  # If AVX-512 is not supported, just set target-cpu=native
+  export RUSTFLAGS="-C target-cpu=native"
+fi
 
 # Set the logging level.
 export RUST_LOG=info