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
72f90d4f
Commit
72f90d4f
authored
Sep 14, 2022
by
Po-Yen, Chen
Browse files
Rename functions
parent
67e84aed
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
10 additions
and
9 deletions
+10
-9
example/37_permute/permute_1xHxW_fp16.cpp
example/37_permute/permute_1xHxW_fp16.cpp
+2
-2
example/37_permute/permute_HxWx4_fp16.cpp
example/37_permute/permute_HxWx4_fp16.cpp
+2
-2
example/37_permute/permute_NxHxW_fp16.cpp
example/37_permute/permute_NxHxW_fp16.cpp
+2
-2
example/37_permute/run_permute_element_example.inc
example/37_permute/run_permute_element_example.inc
+4
-3
No files found.
example/37_permute/permute_1xHxW_fp16.cpp
View file @
72f90d4f
...
@@ -15,6 +15,6 @@ using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute
...
@@ -15,6 +15,6 @@ using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute
<
ADataType
,
BDataType
,
PassThrough
,
3
,
256
,
1
,
32
,
32
,
3
,
S
<
1
,
32
,
8
>
,
S
<
0
,
1
,
2
>
,
2
,
1
,
2
,
1
>
;
<
ADataType
,
BDataType
,
PassThrough
,
3
,
256
,
1
,
32
,
32
,
3
,
S
<
1
,
32
,
8
>
,
S
<
0
,
1
,
2
>
,
2
,
1
,
2
,
1
>
;
// clang-format on
// clang-format on
#include "run_permute_example.inc"
#include "run_permute_
element_
example.inc"
int
main
()
{
return
!
run_permute_example
({
1
,
16000
,
80
},
{
0
,
2
,
1
});
}
int
main
()
{
return
!
run_permute_
element_
example
({
1
,
16000
,
80
},
{
0
,
2
,
1
});
}
example/37_permute/permute_HxWx4_fp16.cpp
View file @
72f90d4f
...
@@ -18,6 +18,6 @@ using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute
...
@@ -18,6 +18,6 @@ using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute
#define NUM_ELEMS_IN_BUNDLE 4
#define NUM_ELEMS_IN_BUNDLE 4
static_assert
(
std
::
is_same_v
<
detail
::
get_bundled_t
<
F64
,
NUM_ELEMS_IN_BUNDLE
>
,
F16
>
);
static_assert
(
std
::
is_same_v
<
detail
::
get_bundled_t
<
F64
,
NUM_ELEMS_IN_BUNDLE
>
,
F16
>
);
#include "run_permute_example.inc"
#include "run_permute_
element_
example.inc"
int
main
()
{
return
!
run_permute_example
({
1
,
80
,
16000
},
{
0
,
2
,
1
});
}
int
main
()
{
return
!
run_permute_
element_
example
({
1
,
80
,
16000
},
{
0
,
2
,
1
});
}
example/37_permute/permute_NxHxW_fp16.cpp
View file @
72f90d4f
...
@@ -15,6 +15,6 @@ using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute
...
@@ -15,6 +15,6 @@ using DevicePermuteInstance = ck::tensor_operation::device::DevicePermute
<
ADataType
,
BDataType
,
PassThrough
,
3
,
128
,
4
,
16
,
8
,
6
,
S
<
2
,
16
,
4
>
,
S
<
0
,
1
,
2
>
,
2
,
1
,
2
,
1
>
;
<
ADataType
,
BDataType
,
PassThrough
,
3
,
128
,
4
,
16
,
8
,
6
,
S
<
2
,
16
,
4
>
,
S
<
0
,
1
,
2
>
,
2
,
1
,
2
,
1
>
;
// clang-format on
// clang-format on
#include "run_permute_example.inc"
#include "run_permute_
element_
example.inc"
int
main
()
{
return
!
run_permute_example
({
121
,
768
,
80
},
{
0
,
2
,
1
});
}
int
main
()
{
return
!
run_permute_
element_
example
({
121
,
768
,
80
},
{
0
,
2
,
1
});
}
example/37_permute/run_permute_example.inc
→
example/37_permute/run_permute_
element_
example.inc
View file @
72f90d4f
...
@@ -7,7 +7,7 @@
...
@@ -7,7 +7,7 @@
#define NUM_ELEMS_IN_BUNDLE 1
#define NUM_ELEMS_IN_BUNDLE 1
#endif
#endif
bool
run_permute
(
const
Problem
&
problem
)
bool
run_permute
_element
(
const
Problem
&
problem
)
{
{
#if 1 < NUM_ELEMS_IN_BUNDLE
#if 1 < NUM_ELEMS_IN_BUNDLE
static_assert
(
std
::
is_same_v
<
ADataType
,
BDataType
>
&&
static_assert
(
std
::
is_same_v
<
ADataType
,
BDataType
>
&&
...
@@ -106,7 +106,8 @@ bool run_permute(const Problem& problem)
...
@@ -106,7 +106,8 @@ bool run_permute(const Problem& problem)
#endif
#endif
}
}
bool
run_permute_example
(
const
Problem
::
Shape
&
default_shape
,
const
Problem
::
Axes
&
default_axes
)
bool
run_permute_element_example
(
const
Problem
::
Shape
&
default_shape
,
const
Problem
::
Axes
&
default_axes
)
{
{
return
run_permute
(
Problem
{
default_shape
,
default_axes
});
return
run_permute
_element
(
Problem
{
default_shape
,
default_axes
});
}
}
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