Commit 7c42995b authored by rocking's avatar rocking
Browse files

Refine naming

parent 084588c8
...@@ -11,7 +11,7 @@ ...@@ -11,7 +11,7 @@
using ck::index_t; using ck::index_t;
struct maxPoolFwdArgParser struct avgPoolFwdArgParser
{ {
std::unordered_map<std::string, std::vector<int>> long_opts = { std::unordered_map<std::string, std::vector<int>> long_opts = {
{"length", {}}, {"wsize", {}}, {"wstride", {}}, {"pad1", {}}, {"pad2", {}}}; {"length", {}}, {"wsize", {}}, {"wstride", {}}, {"pad1", {}}, {"pad2", {}}};
...@@ -90,7 +90,7 @@ int profile_avg_pool2d_fwd(int argc, char* argv[]) ...@@ -90,7 +90,7 @@ int profile_avg_pool2d_fwd(int argc, char* argv[])
time_kernel = std::stoi(argv[6]); time_kernel = std::stoi(argv[6]);
// parse the long options // parse the long options
maxPoolFwdArgParser arg_parser; avgPoolFwdArgParser arg_parser;
arg_parser(argc, argv); arg_parser(argc, argv);
in_length = arg_parser.long_opts["length"]; in_length = arg_parser.long_opts["length"];
wsize = arg_parser.long_opts["wsize"]; wsize = arg_parser.long_opts["wsize"];
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment