Skip to content

Commit

Permalink
TOOLS: expose -f param
Browse files Browse the repository at this point in the history
  • Loading branch information
@x41lakazam
x41lakazam committed Aug 19, 2024
1 parent 5ced790 commit 5f83325
Showing 1 changed file with 115 additions and 123 deletions.
238 changes: 115 additions & 123 deletions tools/perf/ucc_pt_config.cc
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/**
* Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* Copyright (c) 2022-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
*
* See file LICENSE for terms.
*/
Expand All @@ -9,7 +9,8 @@ BEGIN_C_DECLS
#include "utils/ucc_string.h"
END_C_DECLS

ucc_pt_config::ucc_pt_config() {
ucc_pt_config::ucc_pt_config()
{
bootstrap.bootstrap = UCC_PT_BOOTSTRAP_MPI;
bench.op_type = UCC_PT_OP_TYPE_ALLREDUCE;
bench.min_count = 128;
Expand Down Expand Up @@ -66,140 +67,131 @@ const std::map<std::string, ucc_memory_type_t> ucc_pt_memtype_map = {
};

const std::map<std::string, ucc_datatype_t> ucc_pt_datatype_map = {
{"int8", UCC_DT_INT8},
{"uint8", UCC_DT_UINT8},
{"int16", UCC_DT_INT16},
{"uint16", UCC_DT_UINT16},
{"float16", UCC_DT_FLOAT16},
{"bfloat16", UCC_DT_BFLOAT16},
{"int32", UCC_DT_INT32},
{"uint32", UCC_DT_UINT32},
{"float32", UCC_DT_FLOAT32},
{"float32_complex", UCC_DT_FLOAT32_COMPLEX},
{"int64", UCC_DT_INT64},
{"uint64", UCC_DT_UINT64},
{"float64", UCC_DT_FLOAT64},
{"float64_complex", UCC_DT_FLOAT64_COMPLEX},
{"int128", UCC_DT_INT128},
{"uint128", UCC_DT_UINT128},
{"float128", UCC_DT_FLOAT64},
{"float128_complex", UCC_DT_FLOAT128_COMPLEX},
{"int8", UCC_DT_INT8}, {"uint8", UCC_DT_UINT8},
{"int16", UCC_DT_INT16}, {"uint16", UCC_DT_UINT16},
{"float16", UCC_DT_FLOAT16}, {"bfloat16", UCC_DT_BFLOAT16},
{"int32", UCC_DT_INT32}, {"uint32", UCC_DT_UINT32},
{"float32", UCC_DT_FLOAT32}, {"float32_complex", UCC_DT_FLOAT32_COMPLEX},
{"int64", UCC_DT_INT64}, {"uint64", UCC_DT_UINT64},
{"float64", UCC_DT_FLOAT64}, {"float64_complex", UCC_DT_FLOAT64_COMPLEX},
{"int128", UCC_DT_INT128}, {"uint128", UCC_DT_UINT128},
{"float128", UCC_DT_FLOAT64}, {"float128_complex", UCC_DT_FLOAT128_COMPLEX},
};

ucc_status_t ucc_pt_config::process_args(int argc, char *argv[])
{
int c;
int c;
ucc_status_t st;

while ((c = getopt(argc, argv, "c:b:e:d:m:n:w:o:N:r:S:iphFT")) != -1) {
while ((c = getopt(argc, argv, "c:b:e:f:d:m:n:w:o:N:r:S:iphFT")) != -1) {
switch (c) {
case 'c':
if (ucc_pt_op_map.count(optarg) == 0) {
std::cerr << "invalid operation: " << optarg
<< std::endl;
return UCC_ERR_INVALID_PARAM;
}
bench.op_type = ucc_pt_op_map.at(optarg);
break;
case 'o':
if (ucc_pt_reduction_op_map.count(optarg) == 0) {
std::cerr << "invalid reduction operation: " << optarg
<< std::endl;
return UCC_ERR_INVALID_PARAM;
}
bench.op = ucc_pt_reduction_op_map.at(optarg);
break;
case 'm':
if (ucc_pt_memtype_map.count(optarg) == 0) {
std::cerr << "invalid memory type: " << optarg
<<std::endl;
return UCC_ERR_INVALID_PARAM;
}
bench.mt = ucc_pt_memtype_map.at(optarg);
comm.mt = bench.mt;
break;
case 'd':
if (ucc_pt_datatype_map.count(optarg) == 0) {
std::cerr << "invalid datatype:" << optarg
<< std::endl;
return UCC_ERR_INVALID_PARAM;
}
bench.dt = ucc_pt_datatype_map.at(optarg);
break;
case 'b':
st = ucc_str_to_memunits(optarg, (void*)&bench.min_count);
if (st != UCC_OK) {
std::cerr << "failed to parse min count" << std::endl;
return st;
}
break;
case 'e':
st = ucc_str_to_memunits(optarg, (void*)&bench.max_count);
if (st != UCC_OK) {
std::cerr << "failed to parse max count" << std::endl;
return st;
}
break;
case 'r':
std::stringstream(optarg) >> bench.root;
break;
case 'S':
std::stringstream(optarg) >> bench.root_shift;
break;
case 'n':
std::stringstream(optarg) >> bench.n_iter_small;
bench.n_iter_large = bench.n_iter_small;
break;
case 'w':
std::stringstream(optarg) >> bench.n_warmup_small;
bench.n_warmup_large = bench.n_warmup_small;
break;
case 'f':
std::stringstream(optarg) >> bench.mult_factor;
break;
case 'N':
std::stringstream(optarg) >> bench.n_bufs;
break;
case 'i':
bench.inplace = true;
break;
case 'p':
bench.persistent = true;
break;
case 'T':
bench.triggered = true;
break;
case 'F':
bench.full_print = true;
break;
case 'h':
default:
print_help();
std::exit(0);
case 'c':
if (ucc_pt_op_map.count(optarg) == 0) {
std::cerr << "invalid operation: " << optarg << std::endl;
return UCC_ERR_INVALID_PARAM;
}
bench.op_type = ucc_pt_op_map.at(optarg);
break;
case 'o':
if (ucc_pt_reduction_op_map.count(optarg) == 0) {
std::cerr << "invalid reduction operation: " << optarg
<< std::endl;
return UCC_ERR_INVALID_PARAM;
}
bench.op = ucc_pt_reduction_op_map.at(optarg);
break;
case 'm':
if (ucc_pt_memtype_map.count(optarg) == 0) {
std::cerr << "invalid memory type: " << optarg << std::endl;
return UCC_ERR_INVALID_PARAM;
}
bench.mt = ucc_pt_memtype_map.at(optarg);
comm.mt = bench.mt;
break;
case 'd':
if (ucc_pt_datatype_map.count(optarg) == 0) {
std::cerr << "invalid datatype:" << optarg << std::endl;
return UCC_ERR_INVALID_PARAM;
}
bench.dt = ucc_pt_datatype_map.at(optarg);
break;
case 'b':
st = ucc_str_to_memunits(optarg, (void *)&bench.min_count);
if (st != UCC_OK) {
std::cerr << "failed to parse min count" << std::endl;
return st;
}
break;
case 'e':
st = ucc_str_to_memunits(optarg, (void *)&bench.max_count);
if (st != UCC_OK) {
std::cerr << "failed to parse max count" << std::endl;
return st;
}
break;
case 'r':
std::stringstream(optarg) >> bench.root;
break;
case 'S':
std::stringstream(optarg) >> bench.root_shift;
break;
case 'n':
std::stringstream(optarg) >> bench.n_iter_small;
bench.n_iter_large = bench.n_iter_small;
break;
case 'w':
std::stringstream(optarg) >> bench.n_warmup_small;
bench.n_warmup_large = bench.n_warmup_small;
break;
case 'f':
std::stringstream(optarg) >> bench.mult_factor;
break;
case 'N':
std::stringstream(optarg) >> bench.n_bufs;
break;
case 'i':
bench.inplace = true;
break;
case 'p':
bench.persistent = true;
break;
case 'T':
bench.triggered = true;
break;
case 'F':
bench.full_print = true;
break;
case 'h':
default:
print_help();
std::exit(0);
}
}
return UCC_OK;
}

void ucc_pt_config::print_help()
{
std::cout << "Usage: ucc_perftest [options]"<<std::endl;
std::cout << " -c <collective name>: Collective type"<<std::endl;
std::cout << " -b <count>: Min number of elements"<<std::endl;
std::cout << " -e <count>: Max number of elements"<<std::endl;
std::cout << " -i: inplace collective"<<std::endl;
std::cout << " -p: persistent collective"<<std::endl;
std::cout << " -d <dt name>: datatype"<<std::endl;
std::cout << " -o <op name>: reduction operation type"<<std::endl;
std::cout << " -r <number>: root for rooted collectives"<<std::endl;
std::cout << " -m <mtype name>: memory type"<<std::endl;
std::cout << " -n <number>: number of iterations"<<std::endl;
std::cout << " -w <number>: number of warmup iterations"<<std::endl;
std::cout << " -f <number>: multiplication factor between sizes. Default : 2."<<std::endl;
std::cout << " -N <number>: number of buffers"<<std::endl;
std::cout << " -T: triggered collective"<<std::endl;
std::cout << " -F: enable full print"<<std::endl;
std::cout << " -S: <number>: root shift for rooted collectives"<<std::endl;
std::cout << " -h: show this help message"<<std::endl;
std::cout << "Usage: ucc_perftest [options]" << std::endl;
std::cout << " -c <collective name>: Collective type" << std::endl;
std::cout << " -b <count>: Min number of elements" << std::endl;
std::cout << " -e <count>: Max number of elements" << std::endl;
std::cout << " -i: inplace collective" << std::endl;
std::cout << " -p: persistent collective" << std::endl;
std::cout << " -d <dt name>: datatype" << std::endl;
std::cout << " -o <op name>: reduction operation type" << std::endl;
std::cout << " -r <number>: root for rooted collectives" << std::endl;
std::cout << " -m <mtype name>: memory type" << std::endl;
std::cout << " -n <number>: number of iterations" << std::endl;
std::cout << " -w <number>: number of warmup iterations" << std::endl;
std::cout
<< " -f <number>: multiplication factor between sizes. Default : 2."
<< std::endl;
std::cout << " -N <number>: number of buffers" << std::endl;
std::cout << " -T: triggered collective" << std::endl;
std::cout << " -F: enable full print" << std::endl;
std::cout << " -S: <number>: root shift for rooted collectives"
<< std::endl;
std::cout << " -h: show this help message" << std::endl;
std::cout << std::endl;
}

0 comments on commit 5f83325

Please sign in to comment.