* Copyright (c) 2005 Topspin Communications. All rights reserved.
* Copyright (c) 2005 Mellanox Technologies Ltd. All rights reserved.
* Copyright (c) 2009 HNR Consulting. All rights reserved.
*
* This software is available to you under a choice of one of two
* licenses. You may choose to be licensed under the terms of the GNU
* General Public License (GPL) Version 2, available from the file
* COPYING in the main directory of this source tree, or the
* OpenIB.org BSD license below:
*
* Redistribution and use in source and binary forms, with or
* without modification, are permitted provided that the following
* conditions are met:
*
* - Redistributions of source code must retain the above
* copyright notice, this list of conditions and the following
* disclaimer.
*
* - Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the following
* disclaimer in the documentation and/or other materials
* provided with the distribution.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*
* $Id$
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "perftest_communication.h"
#include "perftest_parameters.h"
#include "perftest_resources.h"
******************************************************************************/
int main(int argc, char *argv[])
{
int ret_parser, i = 0, rc;
struct ibv_device *ib_dev = NULL;
struct pingpong_context ctx;
struct pingpong_dest *my_dest, *rem_dest;
struct perftest_parameters user_param;
struct perftest_comm user_comm;
struct bw_report_data my_bw_rep, rem_bw_rep;
int rdma_cm_flow_destroyed = 0;
memset(&user_param, 0, sizeof(struct perftest_parameters));
memset(&user_comm, 0, sizeof(struct perftest_comm));
memset(&ctx, 0, sizeof(struct pingpong_context));
user_param.verb = WRITE;
user_param.tst = BW;
strncpy(user_param.version, VERSION, sizeof(user_param.version));
ret_parser = parser(&user_param, argv, argc);
if (ret_parser) {
if (ret_parser != VERSION_EXIT && ret_parser != HELP_EXIT)
fprintf(stderr, " Parser function exited with Error\n");
goto return_error;
}
if ((user_param.connection_type == DC || user_param.use_xrc) && user_param.duplex) {
user_param.num_of_qps *= 2;
}
ib_dev = ctx_find_dev(&user_param.ib_devname);
if (!ib_dev) {
fprintf(stderr, " Unable to find the Infiniband/RoCE device\n");
goto return_error;
}
ctx.context = ctx_open_device(ib_dev, &user_param);
if (!ctx.context) {
fprintf(stderr, " Couldn't get context for the device\n");
goto free_devname;
}
{
int need_context_extend = 0;
#ifdef CUST
if (user_param.is_use_direct_async) {
need_context_extend = 1;
}
#endif
#ifdef HYPER_VEROCE_EN
if (user_param.use_hyper_roce) {
need_context_extend = 1;
}
#endif
if (need_context_extend) {
ctx.context_extend = ctx_open_device_extend(ctx.context);
if (!ctx.context_extend) {
fprintf(stderr, "Failed to create context_extend\n");
free(user_param.ib_devname);
return FAILURE;
}
}
}
* the function will print the relevent error info. */
if (verify_params_with_device_context(ctx.context, &user_param)) {
fprintf(stderr, " Couldn't get context for the device\n");
goto free_devname;
}
if (check_link(ctx.context, &user_param)) {
fprintf(stderr, " Couldn't get context for the device\n");
goto free_devname;
}
if (create_comm_struct(&user_comm, &user_param)) {
fprintf(stderr, " Unable to create RDMA_CM resources\n");
goto free_devname;
}
if (user_param.output == FULL_VERBOSITY && user_param.machine == SERVER) {
printf("\n************************************\n");
printf("* Waiting for client to connect... *\n");
printf("************************************\n");
}
if (establish_connection(&user_comm)) {
fprintf(stderr, " Unable to init the socket connection\n");
dealloc_comm_struct(&user_comm, &user_param);
goto free_devname;
}
sleep(1);
exchange_versions(&user_comm, &user_param);
check_version_compatibility(&user_param);
check_sys_data(&user_comm, &user_param);
if (check_mtu(ctx.context, &user_param, &user_comm)) {
fprintf(stderr, " Couldn't get context for the device\n");
dealloc_comm_struct(&user_comm, &user_param);
goto free_devname;
}
MAIN_ALLOC(my_dest, struct pingpong_dest, user_param.num_of_qps, free_rdma_params);
memset(my_dest, 0, sizeof(struct pingpong_dest) * user_param.num_of_qps);
MAIN_ALLOC(rem_dest, struct pingpong_dest, user_param.num_of_qps, free_my_dest);
memset(rem_dest, 0, sizeof(struct pingpong_dest) * user_param.num_of_qps);
if (alloc_ctx(&ctx, &user_param)) {
fprintf(stderr, "Couldn't allocate context\n");
goto free_mem;
}
if (user_param.work_rdma_cm == ON) {
rc = create_rdma_cm_connection(&ctx, &user_param, &user_comm, my_dest, rem_dest);
if (rc) {
fprintf(stderr, "Failed to create RDMA CM connection with resources.\n");
dealloc_ctx(&ctx, &user_param);
goto free_mem;
}
} else {
if (ctx_init(&ctx, &user_param)) {
fprintf(stderr, " Couldn't create IB resources\n");
dealloc_ctx(&ctx, &user_param);
goto free_mem;
}
}
if (set_up_connection(&ctx, &user_param, my_dest)) {
fprintf(stderr, " Unable to set up socket connection\n");
goto destroy_context;
}
ctx_print_test_info(&user_param);
for (i = 0; i < user_param.num_of_qps; i++) {
if (ctx_hand_shake(&user_comm, &my_dest[i], &rem_dest[i])) {
fprintf(stderr, " Failed to exchange data between server and clients\n");
goto destroy_context;
}
}
if (user_param.work_rdma_cm == OFF) {
if (ctx_check_gid_compatibility(&my_dest[0], &rem_dest[0])) {
fprintf(stderr, "\n Found Incompatibility issue with GID types.\n");
fprintf(stderr, " Please Try to use a different IP version.\n\n");
goto destroy_context;
}
}
if (user_param.work_rdma_cm == OFF) {
if (ctx_connect(&ctx, rem_dest, &user_param, my_dest)) {
fprintf(stderr, " Unable to Connect the HCA's through the link\n");
goto destroy_context;
}
}
if (user_param.connection_type == DC) {
if (set_up_connection(&ctx, &user_param, my_dest)) {
fprintf(stderr, " Unable to set up socket connection\n");
goto destroy_context;
}
}
for (i = 0; i < user_param.num_of_qps; i++)
ctx_print_pingpong_data(&my_dest[i], &user_comm);
user_comm.rdma_params->side = REMOTE;
for (i = 0; i < user_param.num_of_qps; i++) {
if (ctx_hand_shake(&user_comm, &my_dest[i], &rem_dest[i])) {
fprintf(stderr, " Failed to exchange data between server and clients\n");
goto destroy_context;
}
ctx_print_pingpong_data(&rem_dest[i], &user_comm);
}
if (ctx_hand_shake(&user_comm, &my_dest[0], &rem_dest[0])) {
fprintf(stderr, " Failed to exchange data between server and clients\n");
goto destroy_context;
}
if (user_param.output == FULL_VERBOSITY) {
if (user_param.report_per_port) {
printf(RESULT_LINE_PER_PORT);
printf((user_param.report_fmt == MBS ? RESULT_FMT_PER_PORT : RESULT_FMT_G_PER_PORT));
} else {
printf(RESULT_LINE);
printf((user_param.report_fmt == MBS ? RESULT_FMT : RESULT_FMT_G));
}
printf((user_param.cpu_util_data.enable ? RESULT_EXT_CPU_UTIL : RESULT_EXT));
}
if (user_param.machine == SERVER && user_param.verb == WRITE && !user_param.duplex) {
#ifdef HRN
if (user_param.qu_buf_dma_mode == INDEP_UB) {
int err = set_lb(&ctx, &user_param, user_param.num_of_qps);
if (err != 0) {
fprintf(stderr, "set_lb failed, err=%d\n", err);
goto free_mem;
}
}
#endif
if (ctx_hand_shake(&user_comm, &my_dest[0], &rem_dest[0])) {
fprintf(stderr, " Failed to exchange data between server and clients\n");
goto free_mem;
}
xchg_bw_reports(&user_comm, &my_bw_rep, &rem_bw_rep, atof(user_param.rem_version));
if (user_param.test_method != RUN_INFINITELY) {
print_full_bw_report(&user_param, &rem_bw_rep, NULL);
} else {
printf(" Client closed connection\n");
}
if (ctx_close_connection(&user_comm, &my_dest[0], &rem_dest[0])) {
fprintf(stderr, "Failed to close connection between server and client\n");
goto free_mem;
}
if (user_param.output == FULL_VERBOSITY) {
if (user_param.report_per_port)
printf(RESULT_LINE_PER_PORT);
else
printf(RESULT_LINE);
}
if (user_param.work_rdma_cm == ON) {
if (destroy_ctx(&ctx, &user_param)) {
fprintf(stderr, "Failed to destroy resources\n");
goto destroy_cm_context;
}
user_comm.rdma_params->work_rdma_cm = OFF;
free(my_dest);
free(rem_dest);
free(user_param.ib_devname);
if (destroy_ctx(user_comm.rdma_ctx, user_comm.rdma_params)) {
free(user_comm.rdma_params);
free(user_comm.rdma_ctx);
return FAILURE;
}
free(user_comm.rdma_params);
free(user_comm.rdma_ctx);
return SUCCESS;
}
free(my_dest);
free(rem_dest);
free(user_param.ib_devname);
if (destroy_ctx(&ctx, &user_param)) {
free(user_comm.rdma_params);
return FAILURE;
}
free(user_comm.rdma_params);
return SUCCESS;
}
if (user_param.test_method == RUN_ALL) {
for (i = 1; i < 24; ++i) {
user_param.size = (uint64_t)1 << i;
if (user_param.machine == CLIENT || user_param.duplex) {
ctx_set_send_wqes(&ctx, &user_param, rem_dest);
}
if (user_param.verb == WRITE_IMM && !user_param.use_unsolicited_write &&
(user_param.machine == SERVER || user_param.duplex)) {
if (ctx_set_recv_wqes(&ctx, &user_param)) {
fprintf(stderr, " Failed to post receive recv_wqes\n");
goto free_mem;
}
}
if (user_param.perform_warm_up && !user_param.is_use_direct_async) {
if (user_param.verb == WRITE_IMM) {
fprintf(stderr, "Warm up not supported for WRITE_IMM verb.\n");
fprintf(stderr, "Skipping\n");
} else if (perform_warm_up(&ctx, &user_param)) {
fprintf(stderr, "Problems with warm up\n");
goto free_mem;
}
}
if (user_param.duplex || user_param.verb == WRITE_IMM) {
if (ctx_hand_shake(&user_comm, &my_dest[0], &rem_dest[0])) {
fprintf(stderr, "Failed to sync between server and client between different msg sizes\n");
goto free_mem;
}
}
if (user_param.duplex && user_param.verb == WRITE_IMM) {
if (run_iter_bi(&ctx, &user_param)) {
fprintf(stderr, " Failed to complete run_iter_bi function successfully\n");
goto free_mem;
}
} else if (user_param.machine == CLIENT || user_param.verb != WRITE_IMM) {
if (user_param.is_use_direct_async) {
#ifdef CUST
if (user_param.qu_buf_dma_mode == INDEP_UB) {
int warmupLoops = user_param.perform_warm_up == 0 ? 0 : 10;
if (run_iter_bw_nda_for_ubnic(&ctx, &user_param, my_dest, rem_dest, 0, 0, 0, true, 0,
KERNEL_TRANS_TYPE_BW, warmupLoops)) {
fprintf(stderr, " Failed to complete run_iter_bw_nda function successfully\n");
goto free_mem;
}
} else {
int warmupLoops = user_param.perform_warm_up == 0 ? 0 : 10;
if (run_iter_bw_nda_multi_qp(&ctx, &user_param, my_dest, rem_dest, 0, 0, true, 0,
KERNEL_TRANS_TYPE_BW, warmupLoops)) {
fprintf(stderr, " Failed to complete run_iter_bw_nda function successfully\n");
goto free_mem;
}
}
#else
fprintf(stderr, "NDA functionality is not available without CANN support\n");
goto free_mem;
#endif
} else {
if (run_iter_bw(&ctx, &user_param)) {
fprintf(stderr, " Failed to complete run_iter_bw function successfully\n");
goto free_mem;
}
}
} else if (user_param.machine == SERVER) {
if (run_iter_bw_server(&ctx, &user_param)) {
fprintf(stderr, " Failed to complete run_iter_bw_server function successfully\n");
goto free_mem;
}
}
if (user_param.verb == WRITE_IMM || (user_param.duplex && (atof(user_param.version) >= 4.6))) {
if (ctx_hand_shake(&user_comm, &my_dest[0], &rem_dest[0])) {
fprintf(stderr, "Failed to sync between server and client between different msg sizes\n");
goto free_mem;
}
}
if (user_param.is_use_direct_async) {
#ifdef CANN
print_report_bw_nda(&user_param, &my_bw_rep);
#else
fprintf(stderr, "NDA functionality is not available without CANN support\n");
goto free_mem;
#endif
} else {
print_report_bw(&user_param, &my_bw_rep);
}
if (user_param.duplex && (user_param.verb != WRITE_IMM || user_param.test_type != DURATION)) {
xchg_bw_reports(&user_comm, &my_bw_rep, &rem_bw_rep, atof(user_param.rem_version));
print_full_bw_report(&user_param, &my_bw_rep, &rem_bw_rep);
}
}
} else if (user_param.test_method == RUN_REGULAR) {
if ((user_param.machine == CLIENT || user_param.duplex))
ctx_set_send_wqes(&ctx, &user_param, rem_dest);
if (user_param.verb == WRITE_IMM && (user_param.machine == SERVER || user_param.duplex)) {
if (ctx_set_recv_wqes(&ctx, &user_param)) {
fprintf(stderr, " Failed to post receive recv_wqes\n");
goto free_mem;
}
}
if (user_param.verb != SEND && user_param.verb != WRITE_IMM) {
if (user_param.perform_warm_up && !user_param.is_use_direct_async) {
if (perform_warm_up(&ctx, &user_param)) {
fprintf(stderr, "Problems with warm up\n");
goto free_mem;
}
}
}
if (user_param.duplex || user_param.verb == WRITE_IMM) {
if (ctx_hand_shake(&user_comm, &my_dest[0], &rem_dest[0])) {
fprintf(stderr, "Failed to sync between server and client between different msg sizes\n");
goto free_mem;
}
}
if (user_param.duplex && user_param.verb == WRITE_IMM) {
if (run_iter_bi(&ctx, &user_param)) {
fprintf(stderr, " Failed to complete run_iter_bi function successfully\n");
goto free_mem;
}
} else if (user_param.machine == CLIENT || user_param.verb != WRITE_IMM) {
if (user_param.is_use_direct_async) {
#ifdef CUST
if (user_param.qu_buf_dma_mode == INDEP_UB) {
int warmupLoops = user_param.perform_warm_up == 0 ? 0 : 10;
if (run_iter_bw_nda_for_ubnic(&ctx, &user_param, my_dest, rem_dest, 0, 0, 0, true, 0,
KERNEL_TRANS_TYPE_BW, warmupLoops)) {
fprintf(stderr, " Failed to complete run_iter_bw_nda function successfully\n");
goto free_mem;
}
} else {
int warmupLoops = user_param.perform_warm_up == 0 ? 0 : 10;
if (run_iter_bw_nda_multi_qp(&ctx, &user_param, my_dest, rem_dest, 0, 0, true, 0,
KERNEL_TRANS_TYPE_BW, warmupLoops)) {
fprintf(stderr, " Failed to complete run_iter_bw_nda function successfully\n");
goto free_mem;
}
}
#else
fprintf(stderr, "NDA functionality is not available without CANN support\n");
goto free_mem;
#endif
} else {
if (run_iter_bw(&ctx, &user_param)) {
fprintf(stderr, " Failed to complete run_iter_bw function successfully\n");
goto free_mem;
}
}
} else if (user_param.machine == SERVER) {
if (run_iter_bw_server(&ctx, &user_param)) {
fprintf(stderr, " Failed to complete run_iter_bw_server function successfully\n");
goto free_mem;
}
}
if (user_param.is_use_direct_async) {
#ifdef CANN
print_report_bw_nda(&user_param, &my_bw_rep);
#else
fprintf(stderr, "NDA functionality is not available without CANN support\n");
goto free_mem;
#endif
} else {
print_report_bw(&user_param, &my_bw_rep);
}
if (user_param.duplex && (user_param.verb != WRITE_IMM || user_param.test_type != DURATION)) {
xchg_bw_reports(&user_comm, &my_bw_rep, &rem_bw_rep, atof(user_param.rem_version));
print_full_bw_report(&user_param, &my_bw_rep, &rem_bw_rep);
}
if (user_param.report_both && user_param.duplex) {
printf(RESULT_LINE);
printf("\n Local results: \n");
printf(RESULT_LINE);
printf((user_param.report_fmt == MBS ? RESULT_FMT : RESULT_FMT_G));
printf((user_param.cpu_util_data.enable ? RESULT_EXT_CPU_UTIL : RESULT_EXT));
print_full_bw_report(&user_param, &my_bw_rep, NULL);
printf(RESULT_LINE);
printf("\n Remote results: \n");
printf(RESULT_LINE);
printf((user_param.report_fmt == MBS ? RESULT_FMT : RESULT_FMT_G));
printf((user_param.cpu_util_data.enable ? RESULT_EXT_CPU_UTIL : RESULT_EXT));
print_full_bw_report(&user_param, &rem_bw_rep, NULL);
}
} else if (user_param.test_method == RUN_INFINITELY) {
if (user_param.machine == CLIENT || user_param.duplex)
ctx_set_send_wqes(&ctx, &user_param, rem_dest);
else if (user_param.machine == SERVER && user_param.verb == WRITE_IMM) {
if (ctx_set_recv_wqes(&ctx, &user_param)) {
fprintf(stderr, " Failed to post receive recv_wqes\n");
goto free_mem;
}
}
if (user_param.verb == WRITE_IMM) {
if (ctx_hand_shake(&user_comm, &my_dest[0], &rem_dest[0])) {
fprintf(stderr, "Failed to exchange data between server and clients\n");
goto free_mem;
}
}
if (user_param.machine == CLIENT || user_param.verb == WRITE) {
if (user_param.is_use_direct_async) {
#ifdef CUST
if (user_param.qu_buf_dma_mode == INDEP_UB) {
if (run_iter_bw_nda_for_ubnic_run_infinitely(&ctx, &user_param, my_dest, rem_dest, 0, 0, 0, true, 0,
KERNEL_TRANS_TYPE_BW, &my_bw_rep)) {
fprintf(stderr,
" Failed to complete run_iter_bw_nda_for_ubnic_run_infinitely function successfully\n");
goto free_mem;
}
} else {
if (run_iter_bw_nda_infinitely(&ctx, &user_param, my_dest, rem_dest, 0, 0, 0, true, 0,
KERNEL_TRANS_TYPE_BW, &my_bw_rep)) {
fprintf(stderr, " Failed to complete run_iter_bw_nda_infinitely function successfully\n");
goto free_mem;
}
}
#else
fprintf(stderr, "NDA functionality is not available without CANN support\n");
goto free_mem;
#endif
} else {
if (run_iter_bw_infinitely(&ctx, &user_param)) {
fprintf(stderr, " Error occurred while running infinitely! aborting ...\n");
goto free_mem;
}
}
} else if (user_param.machine == SERVER && user_param.verb == WRITE_IMM) {
if (run_iter_bw_infinitely_server(&ctx, &user_param)) {
fprintf(stderr, " Error occurred while running infinitely on server! aborting ...\n");
goto free_mem;
}
}
}
if (user_param.output == FULL_VERBOSITY) {
if (user_param.report_per_port)
printf(RESULT_LINE_PER_PORT);
else
printf(RESULT_LINE);
}
if (user_param.machine == CLIENT && user_param.verb == WRITE && !user_param.duplex) {
if (ctx_hand_shake(&user_comm, &my_dest[0], &rem_dest[0])) {
fprintf(stderr, " Failed to exchange data between server and clients\n");
goto free_mem;
}
xchg_bw_reports(&user_comm, &my_bw_rep, &rem_bw_rep, atof(user_param.rem_version));
}
if (ctx_close_connection(&user_comm, &my_dest[0], &rem_dest[0])) {
fprintf(stderr, "Failed to close connection between server and client\n");
goto free_mem;
}
if (!user_param.is_bw_limit_passed && (user_param.is_limit_bw == ON)) {
fprintf(stderr, "Error: BW result is below bw limit\n");
goto destroy_context;
}
if (!user_param.is_msgrate_limit_passed && (user_param.is_limit_bw == ON)) {
fprintf(stderr, "Error: Msg rate is below msg_rate limit\n");
goto destroy_context;
}
if (user_param.work_rdma_cm == ON) {
if (destroy_ctx(&ctx, &user_param)) {
fprintf(stderr, "Failed to destroy resources\n");
goto destroy_cm_context;
}
user_comm.rdma_params->work_rdma_cm = OFF;
free(rem_dest);
free(my_dest);
free(user_param.ib_devname);
if (destroy_ctx(user_comm.rdma_ctx, user_comm.rdma_params)) {
free(user_comm.rdma_params);
free(user_comm.rdma_ctx);
return FAILURE;
}
free(user_comm.rdma_params);
free(user_comm.rdma_ctx);
return SUCCESS;
}
free(rem_dest);
free(my_dest);
free(user_param.ib_devname);
if (destroy_ctx(&ctx, &user_param)) {
free(user_comm.rdma_params);
return FAILURE;
}
free(user_comm.rdma_params);
return SUCCESS;
destroy_context:
if (destroy_ctx(&ctx, &user_param))
fprintf(stderr, "Failed to destroy resources\n");
destroy_cm_context:
if (user_param.work_rdma_cm == ON) {
rdma_cm_flow_destroyed = 1;
user_comm.rdma_params->work_rdma_cm = OFF;
destroy_ctx(user_comm.rdma_ctx, user_comm.rdma_params);
}
free_mem:
free(rem_dest);
free_my_dest:
free(my_dest);
free_rdma_params:
if (user_param.use_rdma_cm == ON && rdma_cm_flow_destroyed == 0)
dealloc_comm_struct(&user_comm, &user_param);
else {
if (user_param.use_rdma_cm == ON)
free(user_comm.rdma_ctx);
free(user_comm.rdma_params);
}
free_devname:
free(user_param.ib_devname);
return_error:
return FAILURE;
}