Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel
Commits
df43a051
Commit
df43a051
authored
May 18, 2023
by
rocking
Browse files
Fix typo
parent
376a8047
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
16 deletions
+5
-16
library/include/ck/library/tensor_operation_instance/gpu/pool3d_fwd.hpp
...e/ck/library/tensor_operation_instance/gpu/pool3d_fwd.hpp
+0
-1
profiler/include/profiler/profile_pool3d_fwd_impl.hpp
profiler/include/profiler/profile_pool3d_fwd_impl.hpp
+1
-11
profiler/src/profile_max_pool3d_fwd.cpp
profiler/src/profile_max_pool3d_fwd.cpp
+4
-4
No files found.
library/include/ck/library/tensor_operation_instance/gpu/pool3d_fwd.hpp
View file @
df43a051
// 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
...
...
profiler/include/profiler/profile_pool3d_fwd_impl.hpp
View file @
df43a051
...
@@ -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
];
...
...
profiler/src/profile_max_pool3d_fwd.cpp
View file @
df43a051
...
@@ -45,7 +45,7 @@ struct maxPoolFwdArgParser
...
@@ -45,7 +45,7 @@ struct maxPoolFwdArgParser
}
}
};
};
void
print_help_max_pool_fwd
()
void
print_help_max_pool
3d
_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_pool
3d
_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_pool
3d
_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_pool
3d
_fwd"
,
"max_pool
3d
fwd"
,
profile_max_pool
3d
_fwd
);
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment