multi_nic_send_buffer

#!/bin/bash
#
# Wrapper script for multi-NIC RDMA send buffer test
# Runs as client (sender) 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
RECV_IP=""

# First, check if receiver IP is provided as positional argument
if [[ $# -gt 0 ]] && [[ ! "$1" =~ ^-- ]]; then
    RECV_IP="$1"
    shift
fi

# Parse remaining arguments
while [[ $# -gt 0 ]]; do
    case $1 in
        --allow-nvlink)
            USE_NVLINK=1
            shift
            ;;
        --help|-h)
            echo "Usage: $0 <RECEIVER_IP> [OPTIONS]"
            echo ""
            echo "Arguments:"
            echo "  RECEIVER_IP        IP address of the receiver (required)"
            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 10.52.48.100                # Send with NICs-only mode"
            echo "  $0 10.52.48.100 --allow-nvlink # Send with NVLink enabled"
            exit 0
            ;;
        *)
            echo "Unknown option: $1"
            echo "Use --help for usage information"
            exit 1
            ;;
    esac
done

# Check if receiver IP is provided
if [[ -z "$RECV_IP" ]]; then
    echo "Error: Receiver IP address is required"
    echo "Usage: $0 <RECEIVER_IP> [OPTIONS]"
    echo "Use --help for more information"
    exit 1
fi

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

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

# Execute
exec ${CMD}