# Get parameters
PARAMS=($ARGUMENTS)
N_AGENTS=${PARAMS[0]:-24}
TOP_K=${PARAMS[1]:-5}
PRESET=${PARAMS[2]:-"profiling-nsight-cuda"}

# Detect GPU
GPU_ARCH=$(nvidia-smi -L 2>/dev/null | grep -oP 'NVIDIA \K[^ ]+' | tr '[:lower:]' '[:upper:]')

echo "=== GPU Profiling Specialist ==="
echo "Preset: $PRESET"
echo "Top K: $TOP_K"
echo "=============================="

# Determine which agents to profile
# In a real scenario, this would read from benchmark results
# For now, assume we profile the first TOP_K worktrees that have successful builds
for i in $(seq -f "%02g" 1 $TOP_K); do
  WORKTREE="/home/sbstndbs/subsetix_kokkos_optimized_opt${i}"

  if [ ! -d "$WORKTREE" ]; then
    echo "⚠️  Worktree optimized_opt${i} not found, skipping"
    continue
  fi

  echo "=== Profiling optimized_opt${i} ==="
  cd "$WORKTREE"

  # Clean previous build
  rm -rf build-profiling-*

  # Configure with profiling preset
  cmake --preset $PRESET -DKokkos_ARCH_${GPU_ARCH}=ON

  # Build with profiling
  cmake --build $PRESET -j4

  # Run profiling
  if [ "$PRESET" = "profiling-nsight-cuda" ]; then
    # Use Nsight Compute for kernel analysis
    ncu --set full \
        --export profile_optimized_opt${i} \
        ./experimental/benchmarks/experimental_unified_comparison_benchmark \
          --benchmark_filter="3D_Large" \
          --benchmark_repetitions=3

    echo "Profile saved to profile_optimized_opt${i}.ncu-rep"
  elif [ "$PRESET" = "profiling-nsight-cuda-release" ]; then
    # Release + symbols for better performance
    ncu --set full \
        --export profile_optimized_opt${i}_release \
        ./experimental/benchmarks/experimental_unified_comparison_benchmark \
          --benchmark_filter="3D_Large" \
          --benchmark_repetitions=3

    echo "Profile saved to profile_optimized_opt${i}_release.ncu-rep"
  fi

  # Extract key metrics from profile
  echo "Key metrics for optimized_opt${i}:"
  # Parse ncu output to extract occupancy, memory bandwidth, etc.
  # (This would typically use ncu --csv to export metrics)

  echo ""
done

echo "=== Profiling Summary ==="
echo "Profiled $TOP_K agents with preset: $PRESET"
echo "Profiles saved in respective worktree directories"

{
  "experimental-perf-serial": "Linux perf, experimental only",
  "experimental-perf-openmp": "Linux perf + OpenMP",
  "experimental-serial-profile": "Kokkos profiling tools",
  "profiling-nsight-cuda": "Nsight GPU profiling",
  "profiling-nsight-cuda-release": "Nsight with Release + symbols"
}