multi_nic_receive_buffer

#!/bin/bash
#
# Wrapper script for multi-NIC RDMA receive buffer test
# Runs as server (receiver) with configurable NVLink support
#

# Hardcoded configuration
NICS="mlx5_0,mlx5_3,mlx5_4,mlx5_5,mlx5_6,mlx5_9,mlx5_10,mlx5_11"
GPUS="0,1,2,3,4,5,6,7"

# Parse arguments
USE_NVLINK=0

while [[ $# -gt 0 ]]; do
    case $1 in
        --allow-nvlink)
            USE_NVLINK=1
            shift
            ;;
        --help|-h)
            echo "Usage: $0 [OPTIONS]"
            echo ""
            echo "Options:"
            echo "  --allow-nvlink    Enable NVLink transfers (default: disabled, uses --nics-only)"
            echo "  --help, -h        Show this help message"
            echo ""
            echo "Examples:"
            echo "  $0                    # Receive with NICs-only mode"
            echo "  $0 --allow-nvlink    # Receive with NVLink enabled"
            exit 0
            ;;
        *)
            echo "Unknown option: $1"
            echo "Use --help for usage information"
            exit 1
            ;;
    esac
done

# Build command
CMD="./test_rdma_multi_qp -s --nics ${NICS} --gpus ${GPUS}"

# Add --nics-only flag if NVLink is not enabled
if [[ $USE_NVLINK -eq 0 ]]; then
    CMD="${CMD} --nics-only"
fi

# Execute
exec ${CMD}