Commit df43a051 authored by rocking's avatar rocking
Browse files

Fix typo

parent 376a8047
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.
#pragma once
......
......@@ -34,16 +34,6 @@ bool profile_pool3d_fwd_impl(int do_verification,
std::vector<index_t> input_left_pads,
std::vector<index_t> input_right_pads)
{
ignore = do_verification;
ignore = init_method;
ignore = do_log;
ignore = time_kernel;
ignore = in_length;
ignore = window_spatial_lengths;
ignore = window_strides;
ignore = input_left_pads;
ignore = input_right_pads;
constexpr index_t InOutRank = 5;
constexpr index_t WindowRank = 3;
......@@ -71,7 +61,7 @@ bool profile_pool3d_fwd_impl(int do_verification,
}
int Di = in_length[2];
int Hi = in_length[2];
int Hi = in_length[3];
int Wi = in_length[4];
int Do = out_length[2];
int Ho = out_length[3];
......
......@@ -45,7 +45,7 @@ struct maxPoolFwdArgParser
}
};
void print_help_max_pool_fwd()
void print_help_max_pool3d_fwd()
{
std::cout << "arg1: data type (0: fp16; 1: fp32)\n"
<< "arg2: verification (0: no; 1: yes)\n"
......@@ -61,7 +61,7 @@ void print_help_max_pool_fwd()
<< std::endl;
}
int profile_max_pool_fwd(int argc, char* argv[])
int profile_max_pool3d_fwd(int argc, char* argv[])
{
ck::DataTypeEnum data_type = ck::DataTypeEnum::Half;
bool do_verification = true;
......@@ -78,7 +78,7 @@ int profile_max_pool_fwd(int argc, char* argv[])
if(argc != 2 && argc != 30)
{
print_help_max_pool_fwd();
print_help_max_pool3d_fwd();
return 0;
}
else if(argc == 30)
......@@ -162,4 +162,4 @@ int profile_max_pool_fwd(int argc, char* argv[])
return 0;
}
REGISTER_PROFILER_OPERATION("max_pool_fwd", "max_pool fwd", profile_max_pool_fwd);
REGISTER_PROFILER_OPERATION("max_pool3d_fwd", "max_pool3d fwd", profile_max_pool3d_fwd);
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