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
2280e5bb
"vscode:/vscode.git/clone" did not exist on "6ca9c4af05d7263ef26f277330487fc7c9af0878"
Commit
2280e5bb
authored
May 17, 2023
by
rocking
Browse files
Refine naming
parent
21ee92f6
Changes
8
Hide whitespace changes
Inline
Side-by-side
Showing
8 changed files
with
12 additions
and
12 deletions
+12
-12
library/src/tensor_operation_instance/gpu/pooling_fwd/device_avg_pooling2d_fwd_nhwc_f16_instance.cpp
...ooling_fwd/device_avg_pooling2d_fwd_nhwc_f16_instance.cpp
+1
-1
library/src/tensor_operation_instance/gpu/pooling_fwd/device_avg_pooling2d_fwd_nhwc_f32_instance.cpp
...ooling_fwd/device_avg_pooling2d_fwd_nhwc_f32_instance.cpp
+1
-1
library/src/tensor_operation_instance/gpu/pooling_fwd/device_avg_pooling3d_fwd_ndhwc_f16_instance.cpp
...oling_fwd/device_avg_pooling3d_fwd_ndhwc_f16_instance.cpp
+1
-1
library/src/tensor_operation_instance/gpu/pooling_fwd/device_avg_pooling3d_fwd_ndhwc_f32_instance.cpp
...oling_fwd/device_avg_pooling3d_fwd_ndhwc_f32_instance.cpp
+1
-1
library/src/tensor_operation_instance/gpu/pooling_fwd/device_max_pooling2d_fwd_nhwc_f16_instance.cpp
...ooling_fwd/device_max_pooling2d_fwd_nhwc_f16_instance.cpp
+2
-2
library/src/tensor_operation_instance/gpu/pooling_fwd/device_max_pooling2d_fwd_nhwc_f32_instance.cpp
...ooling_fwd/device_max_pooling2d_fwd_nhwc_f32_instance.cpp
+2
-2
library/src/tensor_operation_instance/gpu/pooling_fwd/device_max_pooling3d_fwd_ndhwc_f16_instance.cpp
...oling_fwd/device_max_pooling3d_fwd_ndhwc_f16_instance.cpp
+2
-2
library/src/tensor_operation_instance/gpu/pooling_fwd/device_max_pooling3d_fwd_ndhwc_f32_instance.cpp
...oling_fwd/device_max_pooling3d_fwd_ndhwc_f32_instance.cpp
+2
-2
No files found.
library/src/tensor_operation_instance/gpu/pooling_fwd/device_avg_pooling2d_fwd_nhwc_f16_instance.cpp
View file @
2280e5bb
...
@@ -10,7 +10,7 @@ namespace instance {
...
@@ -10,7 +10,7 @@ namespace instance {
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
AVG
;
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
AVG
;
void
add_device_
avg_
pooling2d_fwd_nhwc_f16_instances
(
void
add_device_pooling2d_fwd_nhwc_f16_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
false
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
false
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
...
...
library/src/tensor_operation_instance/gpu/pooling_fwd/device_avg_pooling2d_fwd_nhwc_f32_instance.cpp
View file @
2280e5bb
...
@@ -10,7 +10,7 @@ namespace instance {
...
@@ -10,7 +10,7 @@ namespace instance {
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
AVG
;
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
AVG
;
void
add_device_
avg_
pooling2d_fwd_nhwc_f32_instances
(
void
add_device_pooling2d_fwd_nhwc_f32_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
false
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
false
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
...
...
library/src/tensor_operation_instance/gpu/pooling_fwd/device_avg_pooling3d_fwd_ndhwc_f16_instance.cpp
View file @
2280e5bb
...
@@ -10,7 +10,7 @@ namespace instance {
...
@@ -10,7 +10,7 @@ namespace instance {
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
AVG
;
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
AVG
;
void
add_device_
avg_
pooling3d_fwd_ndhwc_f16_instances
(
void
add_device_pooling3d_fwd_ndhwc_f16_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
false
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
false
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
...
...
library/src/tensor_operation_instance/gpu/pooling_fwd/device_avg_pooling3d_fwd_ndhwc_f32_instance.cpp
View file @
2280e5bb
...
@@ -10,7 +10,7 @@ namespace instance {
...
@@ -10,7 +10,7 @@ namespace instance {
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
AVG
;
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
AVG
;
void
add_device_
avg_
pooling3d_fwd_ndhwc_f32_instances
(
void
add_device_pooling3d_fwd_ndhwc_f32_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
false
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
false
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
...
...
library/src/tensor_operation_instance/gpu/pooling_fwd/device_max_pooling2d_fwd_nhwc_f16_instance.cpp
View file @
2280e5bb
...
@@ -10,14 +10,14 @@ namespace instance {
...
@@ -10,14 +10,14 @@ namespace instance {
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
MAX
;
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
MAX
;
void
add_device_
max_
pooling2d_fwd_nhwc_f16_instances
(
void
add_device_pooling2d_fwd_nhwc_f16_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
false
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
false
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
instances
,
device_pooling2d_fwd_nhwc_instances
<
F16
,
F16
,
I32
,
F16
,
ReduceOpId
,
false
>
{});
instances
,
device_pooling2d_fwd_nhwc_instances
<
F16
,
F16
,
I32
,
F16
,
ReduceOpId
,
false
>
{});
}
}
void
add_device_
max_
pooling2d_fwd_nhwc_index_f16_instances
(
void
add_device_pooling2d_fwd_nhwc_index_f16_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
true
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
true
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
...
...
library/src/tensor_operation_instance/gpu/pooling_fwd/device_max_pooling2d_fwd_nhwc_f32_instance.cpp
View file @
2280e5bb
...
@@ -10,14 +10,14 @@ namespace instance {
...
@@ -10,14 +10,14 @@ namespace instance {
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
MAX
;
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
MAX
;
void
add_device_
max_
pooling2d_fwd_nhwc_f32_instances
(
void
add_device_pooling2d_fwd_nhwc_f32_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
false
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
false
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
instances
,
device_pooling2d_fwd_nhwc_instances
<
F32
,
F32
,
I32
,
F32
,
ReduceOpId
,
false
>
{});
instances
,
device_pooling2d_fwd_nhwc_instances
<
F32
,
F32
,
I32
,
F32
,
ReduceOpId
,
false
>
{});
}
}
void
add_device_
max_
pooling2d_fwd_nhwc_index_f32_instances
(
void
add_device_pooling2d_fwd_nhwc_index_f32_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
true
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
4
,
2
,
ReduceOpId
,
true
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
...
...
library/src/tensor_operation_instance/gpu/pooling_fwd/device_max_pooling3d_fwd_ndhwc_f16_instance.cpp
View file @
2280e5bb
...
@@ -10,14 +10,14 @@ namespace instance {
...
@@ -10,14 +10,14 @@ namespace instance {
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
MAX
;
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
MAX
;
void
add_device_
max_
pooling3d_fwd_ndhwc_f16_instances
(
void
add_device_pooling3d_fwd_ndhwc_f16_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
false
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
false
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
instances
,
device_pooling3d_fwd_ndhwc_instances
<
F16
,
F16
,
I32
,
F16
,
ReduceOpId
,
false
>
{});
instances
,
device_pooling3d_fwd_ndhwc_instances
<
F16
,
F16
,
I32
,
F16
,
ReduceOpId
,
false
>
{});
}
}
void
add_device_
max_
pooling3d_fwd_ndhwc_index_f16_instances
(
void
add_device_pooling3d_fwd_ndhwc_index_f16_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
true
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
true
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
...
...
library/src/tensor_operation_instance/gpu/pooling_fwd/device_max_pooling3d_fwd_ndhwc_f32_instance.cpp
View file @
2280e5bb
...
@@ -10,14 +10,14 @@ namespace instance {
...
@@ -10,14 +10,14 @@ namespace instance {
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
MAX
;
static
constexpr
auto
ReduceOpId
=
ck
::
ReduceTensorOp
::
MAX
;
void
add_device_
max_
pooling3d_fwd_ndhwc_f32_instances
(
void
add_device_pooling3d_fwd_ndhwc_f32_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
false
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
false
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
instances
,
device_pooling3d_fwd_ndhwc_instances
<
F32
,
F32
,
I32
,
F32
,
ReduceOpId
,
false
>
{});
instances
,
device_pooling3d_fwd_ndhwc_instances
<
F32
,
F32
,
I32
,
F32
,
ReduceOpId
,
false
>
{});
}
}
void
add_device_
max_
pooling3d_fwd_ndhwc_index_f32_instances
(
void
add_device_pooling3d_fwd_ndhwc_index_f32_instances
(
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
true
>>>&
instances
)
std
::
vector
<
std
::
unique_ptr
<
DevicePoolFwd
<
5
,
3
,
ReduceOpId
,
true
>>>&
instances
)
{
{
add_device_operation_instances
(
add_device_operation_instances
(
...
...
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