Commit 72f90d4f authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Rename functions

parent 67e84aed
...@@ -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}); }
...@@ -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}); }
...@@ -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}); }
...@@ -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});
} }
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