Commit df43a051 authored by rocking's avatar rocking
Browse files

Fix typo

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