[TOOLS] fix send_bw test for UD.
[mirror/winof/.git] / tools / perftests / user / write_lat / write_lat.c
index 85921e3..fe1c128 100644 (file)
@@ -388,7 +388,8 @@ static int pp_connect_ctx(struct pingpong_context *ctx, int port, int my_psn,
 
 
 static SOCKET pp_open_port(struct pingpong_context *ctx, const char * servername,
-                       int ib_port, int port, struct pingpong_dest **p_rem_dest,struct user_parameters *user_parm)
+                       int ib_port, int port, struct pingpong_dest **p_my_dest,
+                       struct pingpong_dest **p_rem_dest,struct user_parameters *user_parm)
 {
        struct pingpong_dest    *my_dest;
        struct pingpong_dest    *rem_dest;
@@ -439,7 +440,7 @@ static SOCKET pp_open_port(struct pingpong_context *ctx, const char * servername
                my_dest[i].rkey = ctx->rkey;
                my_dest[i].vaddr = (uintptr_t)ctx->buf + ctx->size;
 
-               printf("  local address:  LID %#04x, QPN %#06x, PSN %#06x "
+               printf("  local address:  LID %#04x, QPN %#06x, PSN %#06x, "
                "RKey %#08x VAddr %#016Lx\n",
                my_dest[i].lid, my_dest[i].qpn, my_dest[i].psn,
                my_dest[i].rkey, my_dest[i].vaddr);
@@ -463,6 +464,7 @@ static SOCKET pp_open_port(struct pingpong_context *ctx, const char * servername
                        return INVALID_SOCKET;
        }
        *p_rem_dest = rem_dest;
+       *p_my_dest = my_dest;
        return sockfd;
 }
 
@@ -648,7 +650,7 @@ int run_iter(struct pingpong_context *ctx, struct user_parameters *user_param,
 
                        ++ccnt;
                }
-//printf("ccnt = %d \n",ccnt);
+               PERF_DEBUG("ccnt = %d \n",ccnt);
        }
        return(0);
 }
@@ -665,6 +667,7 @@ int __cdecl main(int argc, char *argv[])
 
 
        struct pingpong_context *ctx;
+       struct pingpong_dest     *my_dest;
        struct pingpong_dest     *rem_dest;
        struct user_parameters  user_param;
        char                            *ib_devname = NULL;
@@ -836,7 +839,7 @@ int __cdecl main(int argc, char *argv[])
        ctx = pp_init_ctx( size, ib_port,&user_param);
        if (!ctx)
                return 8;
-       sockfd = pp_open_port(ctx, user_param.servername, ib_port, port, &rem_dest,&user_param);
+       sockfd = pp_open_port(ctx, user_param.servername, ib_port, port,&my_dest,&rem_dest,&user_param);
        if (sockfd == INVALID_SOCKET)
                return 9;
        printf("------------------------------------------------------------------\n");