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
653f86ff
Commit
653f86ff
authored
Sep 05, 2021
by
Chao Liu
Browse files
refactor
parent
29a118c6
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
6 deletions
+6
-6
composable_kernel/include/tensor_description/multi_index_transform_helper.hpp
...clude/tensor_description/multi_index_transform_helper.hpp
+6
-6
No files found.
composable_kernel/include/tensor_description/multi_index_transform_helper.hpp
View file @
653f86ff
...
@@ -53,13 +53,9 @@ template <typename LowLengths>
...
@@ -53,13 +53,9 @@ template <typename LowLengths>
__host__
__device__
constexpr
auto
make_merge_transform
(
const
LowLengths
&
low_lengths
)
__host__
__device__
constexpr
auto
make_merge_transform
(
const
LowLengths
&
low_lengths
)
{
{
#if CK_EXPERIMENTAL_MERGE_USE_MAGIC_DIVISION
#if CK_EXPERIMENTAL_MERGE_USE_MAGIC_DIVISION
#if 1
return
make_merge_transform_v2_magic_division
(
low_lengths
);
return
Merge_v2_magic_division
<
LowLengths
>
{
low_lengths
};
#else
return
Merge_v2r2_magic_division
<
LowLengths
>
{
low_lengths
};
#endif
#else
#else
return
Merge
_v1_carry_check
<
LowLengths
>
{
low_lengths
}
;
return
make_merge_transform
_v1_carry_check
(
low_lengths
)
;
#endif
#endif
}
}
...
@@ -74,7 +70,11 @@ template <typename LowLengths>
...
@@ -74,7 +70,11 @@ template <typename LowLengths>
__host__
__device__
constexpr
auto
__host__
__device__
constexpr
auto
make_merge_transform_v2_magic_division
(
const
LowLengths
&
low_lengths
)
make_merge_transform_v2_magic_division
(
const
LowLengths
&
low_lengths
)
{
{
#if 1
return
Merge_v2_magic_division
<
LowLengths
>
{
low_lengths
};
return
Merge_v2_magic_division
<
LowLengths
>
{
low_lengths
};
#else
return
Merge_v2r2_magic_division
<
LowLengths
>
{
low_lengths
};
#endif
}
}
template
<
typename
LowLengths
>
template
<
typename
LowLengths
>
...
...
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