-
Notifications
You must be signed in to change notification settings - Fork 376
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
fabtests: New fabtest fi_flood to test over subscription of resources #10427
base: main
Are you sure you want to change the base?
Changes from all commits
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,305 @@ | ||
/* | ||
* Copyright (c) Intel Corporation. All rights reserved. | ||
* | ||
* This software is available to you under the 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. | ||
*/ | ||
|
||
#include <stdio.h> | ||
#include <stdlib.h> | ||
#include <getopt.h> | ||
#include <unistd.h> | ||
|
||
#include <shared.h> | ||
|
||
int sleep_time = 0; | ||
|
||
static ssize_t post_one_tx(struct ft_context *msg) | ||
{ | ||
ssize_t ret; | ||
|
||
if (ft_check_opts(FT_OPT_VERIFY_DATA | FT_OPT_ACTIVE)) { | ||
ret = ft_fill_buf(msg->buf + ft_tx_prefix_size(), | ||
opts.transfer_size); | ||
if (ret) | ||
return ret; | ||
} | ||
|
||
return ft_post_tx_buf(ep, remote_fi_addr, opts.transfer_size, | ||
NO_CQ_DATA, &msg->context, msg->buf, | ||
msg->desc, 0); | ||
} | ||
|
||
static ssize_t wait_check_rx_bufs(void) | ||
{ | ||
ssize_t ret; | ||
int i; | ||
|
||
ret = ft_get_rx_comp(rx_seq); | ||
if (ret) | ||
return ret; | ||
|
||
if (!ft_check_opts(FT_OPT_VERIFY_DATA | FT_OPT_ACTIVE)) | ||
return 0; | ||
|
||
for (i = 0; i < opts.window_size; i++) { | ||
ret = ft_check_buf((char *) rx_ctx_arr[i].buf + | ||
ft_rx_prefix_size(), opts.transfer_size); | ||
if (ret) | ||
return ret; | ||
} | ||
|
||
return 0; | ||
} | ||
|
||
static int post_rx_sync(void) | ||
{ | ||
int ret; | ||
|
||
ret = ft_post_rx(ep, rx_size, &rx_ctx); | ||
if (ret) | ||
return ret; | ||
|
||
if (opts.dst_addr) { | ||
ret = ft_tx(ep, remote_fi_addr, 1, &tx_ctx); | ||
if (ret) | ||
return ret; | ||
|
||
ret = ft_get_rx_comp(rx_seq); | ||
} else { | ||
ret = ft_get_rx_comp(rx_seq); | ||
if (ret) | ||
return ret; | ||
|
||
ret = ft_tx(ep, remote_fi_addr, 1, &tx_ctx); | ||
} | ||
|
||
return ret; | ||
} | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. add line between functions |
||
|
||
static void mr_close_all(struct ft_context *ctx_arr, int window_size) | ||
{ | ||
int i; | ||
for (i = 0; i < window_size; i++) | ||
FT_CLOSE_FID(ctx_arr[i].mr); | ||
} | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. add line between functions |
||
|
||
static int run_seq_mr_send(void) { | ||
|
||
int ret; | ||
int i; | ||
|
||
mr_close_all(tx_ctx_arr, opts.window_size); | ||
mr_close_all(rx_ctx_arr, opts.window_size); | ||
|
||
printf("Sequential memory registration:"); | ||
if (opts.dst_addr) { | ||
for (i = 0; i < opts.window_size; i++) { | ||
ret = ft_reg_mr(fi, tx_ctx_arr[i].buf, tx_mr_size, | ||
ft_info_to_mr_access(fi), | ||
FT_TX_MR_KEY + i, opts.iface, opts.device, | ||
&(tx_ctx_arr[i].mr), &(tx_ctx_arr[i].desc)); | ||
if (ret) | ||
goto err; | ||
|
||
if (ft_check_opts(FT_OPT_VERIFY_DATA | FT_OPT_ACTIVE)) { | ||
ret = ft_fill_buf(tx_ctx_arr[i].buf + ft_tx_prefix_size(), | ||
opts.transfer_size); | ||
if (ret) | ||
goto err; | ||
} | ||
|
||
ft_post_tx_buf(ep, remote_fi_addr, opts.transfer_size, | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Does this return something? There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. always returns 0 |
||
NO_CQ_DATA, &(tx_ctx_arr[i].context), | ||
tx_ctx_arr[i].buf, tx_ctx_arr[i].desc, | ||
ft_tag); | ||
ret = ft_get_tx_comp(tx_seq); | ||
if (ret) | ||
goto err; | ||
FT_CLOSE_FID(tx_ctx_arr[i].mr); | ||
} | ||
} | ||
else { | ||
for (i = 0; i < opts.window_size; i++) { | ||
ret = ft_reg_mr(fi, rx_ctx_arr[i].buf, rx_mr_size, | ||
ft_info_to_mr_access(fi), FT_RX_MR_KEY + i, | ||
opts.iface, opts.device, &(rx_ctx_arr[i].mr), | ||
&(rx_ctx_arr[i].desc)); | ||
if (ret) | ||
goto err; | ||
|
||
ft_post_rx_buf(ep, opts.transfer_size, | ||
&(rx_ctx_arr[i].context), rx_ctx_arr[i].buf, | ||
rx_ctx_arr[i].desc, ft_tag); | ||
ret = wait_check_rx_bufs(); | ||
if (ret) | ||
goto err; | ||
FT_CLOSE_FID(rx_ctx_arr[i].mr); | ||
} | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Add rx buf close here |
||
} | ||
if (opts.options & FT_OPT_OOB_SYNC) | ||
ret = ft_sync(); | ||
else | ||
ret = post_rx_sync(); | ||
err: | ||
printf("%s\n", ret ? "Fail" : "Pass"); | ||
return ret; | ||
} | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. new line |
||
|
||
static int run_batch_mr_send(void) | ||
{ | ||
int ret, i; | ||
|
||
/* Receive side delay is used in order to let the sender | ||
get ahead of the receiver and post multiple sends | ||
before the receiver begins processing them. */ | ||
if (!opts.dst_addr) | ||
sleep(sleep_time); | ||
|
||
printf("Batch memory registration:"); | ||
if (opts.dst_addr) { | ||
for (i = 0; i < opts.window_size; i++) { | ||
ret = post_one_tx(&tx_ctx_arr[i]); | ||
if (ret) | ||
goto err; | ||
} | ||
|
||
ret = ft_get_tx_comp(tx_seq); | ||
if (ret) | ||
goto err; | ||
} else { | ||
for (i = 0; i < opts.window_size; i++) { | ||
ret = ft_post_rx_buf(ep, opts.transfer_size, | ||
&rx_ctx_arr[i].context, | ||
rx_ctx_arr[i].buf, | ||
rx_ctx_arr[i].desc, 0); | ||
if (ret) | ||
goto err; | ||
} | ||
|
||
ret = wait_check_rx_bufs(); | ||
if (ret) | ||
goto err; | ||
} | ||
|
||
if (opts.options & FT_OPT_OOB_SYNC) | ||
ret = ft_sync(); | ||
else | ||
ret = post_rx_sync(); | ||
err: | ||
printf("%s\n", ret ? "Fail" : "Pass"); | ||
return ret; | ||
} | ||
|
||
static int run(void) | ||
{ | ||
int ret; | ||
|
||
ret = hints->ep_attr->type == FI_EP_MSG ? | ||
ft_init_fabric_cm() : ft_init_fabric(); | ||
if (ret) | ||
return ret; | ||
|
||
ret = run_batch_mr_send(); | ||
if (ret) | ||
goto out; | ||
|
||
ret = run_seq_mr_send(); | ||
if (ret) | ||
goto out; | ||
|
||
out: | ||
return ret; | ||
} | ||
|
||
int main(int argc, char **argv) | ||
{ | ||
int op, ret; | ||
|
||
opts = INIT_OPTS; | ||
|
||
hints = fi_allocinfo(); | ||
if (!hints) | ||
return EXIT_FAILURE; | ||
|
||
hints->ep_attr->type = FI_EP_RDM; | ||
|
||
while ((op = getopt(argc, argv, "UW:vT:h" CS_OPTS ADDR_OPTS INFO_OPTS)) != -1) { | ||
switch (op) { | ||
default: | ||
ft_parse_addr_opts(op, optarg, &opts); | ||
ft_parseinfo(op, optarg, hints, &opts); | ||
ft_parsecsopts(op, optarg, &opts); | ||
break; | ||
case 'W': | ||
opts.window_size = atoi(optarg); | ||
break; | ||
case 'U': | ||
hints->tx_attr->op_flags |= FI_DELIVERY_COMPLETE; | ||
break; | ||
case 'v': | ||
opts.options |= FT_OPT_VERIFY_DATA; | ||
break; | ||
case 'T': | ||
sleep_time = atoi(optarg); | ||
break; | ||
case '?': | ||
case 'h': | ||
ft_usage(argv[0], "test to oversubscribe mr cache and receiver with unexpected msgs."); | ||
FT_PRINT_OPTS_USAGE("-T sleep_time", | ||
"Receive side delay before starting"); | ||
FT_PRINT_OPTS_USAGE("-v", "Enable data verification"); | ||
FT_PRINT_OPTS_USAGE("-W window_size", | ||
"Set transmit window size before waiting for completion"); | ||
return EXIT_FAILURE; | ||
} | ||
} | ||
|
||
if (optind < argc) | ||
opts.dst_addr = argv[optind]; | ||
|
||
hints->caps = FI_MSG; | ||
hints->mode = FI_CONTEXT; | ||
hints->domain_attr->mr_mode = opts.mr_mode; | ||
hints->addr_format = opts.address_format; | ||
|
||
opts.options |= FT_OPT_ALLOC_MULT_MR; | ||
opts.options |= FT_OPT_NO_PRE_POSTED_RX; | ||
if (hints->ep_attr->type == FI_EP_DGRAM) { | ||
fprintf(stderr, "This test does not support DGRAM endpoints\n"); | ||
return -FI_EINVAL; | ||
} | ||
|
||
if (opts.options & FT_OPT_VERIFY_DATA) { | ||
hints->tx_attr->msg_order |= FI_ORDER_SAS; | ||
hints->rx_attr->msg_order |= FI_ORDER_SAS; | ||
} | ||
|
||
ret = run(); | ||
|
||
ft_free_res(); | ||
|
||
return ft_exit_code(ret); | ||
} |
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1 @@ | ||
.so man7/fabtests.7 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This test is just adding a new mode to the bw test - I would just replace/rename the bw test and add the new testing mode inside. No need to create a whole new test
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
FWIW, AWS CI has such flood_peer test that reuse fi_bw: https://github.com/ofiwg/libfabric/blob/main/fabtests/pytest/efa/test_flood_peer.py#L6