#!/bin/bash

# 快速测试脚本 - 验证程序功能

echo "=========================================="
echo "MPI-OpenMP矩阵乘法快速测试"
echo "=========================================="

# 编译程序
echo "1. 编译程序..."
./build.sh
if [ $? -ne 0 ]; then
    echo "编译失败！"
    exit 1
fi
echo ""

# 测试串行版本
echo "2. 测试串行版本 (512x512x512)..."
./gemm_serial 512 512 512 0 | grep -E "(matmul|verification)"
echo ""

# 测试并行版本 - 单进程
echo "3. 测试并行版本 (1 MPI进程, 512x512x512)..."
mpirun -np 1 ./gemm_parallel 512 512 512 | grep -E "(mpi matmul|verification)"
echo ""

# 测试并行版本 - 多进程
echo "4. 测试并行版本 (4 MPI进程, 512x512x512)..."
mpirun -np 4 ./gemm_parallel 512 512 512 | grep -E "(mpi matmul|verification)"
echo ""

# 测试并行版本 - 混合并行
echo "5. 测试并行版本 (2 MPI进程, 2 OpenMP线程, 512x512x512)..."
export OMP_NUM_THREADS=2
mpirun -np 2 ./gemm_parallel 512 512 512 | grep -E "(mpi matmul|verification)"
echo ""

# 测试优化版本
echo "6. 测试优化版本 (4 MPI进程, 512x512x512)..."
mpirun -np 4 ./gemm_optimized 512 512 512 | grep -E "(optimized mpi matmul|verification)"
echo ""

echo "=========================================="
echo "快速测试完成！"
echo ""
echo "如果所有测试都显示 'PASSED'，说明程序工作正常。"
echo "接下来可以运行 ./run_experiments.sh 进行完整实验。"
echo "=========================================="