adapt the benchmark script for sp1

Author: lockedloop

benchmark-sp1-cuda.sh

@@ -11,33 +11,44 @@ echo "╚═══════════════════════
 echo
 
 # Check if CUDA is available
-if command -v nvidia-smi &> /dev/null; then
-    echo "✅ NVIDIA GPU detected:"
-    nvidia-smi --query-gpu=name,memory.total --format=csv,noheader
-    echo
-else
-    echo "⚠️  No NVIDIA GPU detected. Will run on CPU."
-    echo
+if ! command -v nvidia-smi &> /dev/null; then
+    echo "❌ No NVIDIA GPU detected. This script requires a GPU."
+    echo "   Please run on a GPU-enabled instance."
+    exit 1
 fi
 
-# Set environment variables for optimal SP1 performance
+echo "✅ NVIDIA GPU detected:"
+nvidia-smi --query-gpu=name,memory.total --format=csv,noheader
+echo
+
+# Set environment variables for SP1 CUDA proving
+export SP1_PROVER=cuda
+export CUDA_VISIBLE_DEVICES=0
 export RUST_LOG=info
+export SP1_CUDA=1
+export RUSTFLAGS="-C target-cpu=native -C target-feature=+avx2"
+
+echo "🎯 SP1 CUDA environment configured:"
+echo "   SP1_PROVER=$SP1_PROVER"
+echo "   CUDA_VISIBLE_DEVICES=$CUDA_VISIBLE_DEVICES"
+echo "   SP1_CUDA=$SP1_CUDA"
+echo
 
 echo "🔨 Building SP1 guest program..."
 cd crates/sp1/guest
 cargo prove build
 cd ../../..
 
-echo "📦 Building SP1 host with CUDA support..."
-cargo build --release --features cuda -p sp1-host
+echo "📦 Building SP1 host for CUDA..."
+cargo build --release -p sp1-host
 
 echo
-echo "🚀 Running benchmarks with CUDA acceleration..."
+echo "🚀 Running benchmarks..."
 echo "   Sample size: 10 iterations per benchmark"
 echo
 
 # Run the Criterion benchmark
-cargo bench --features cuda -p sp1-host
+cargo bench -p sp1-host
 
 echo
 echo "📊 Benchmark results saved to:"