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
c84e739d
Commit
c84e739d
authored
Jul 06, 2022
by
Chao Liu
Browse files
rename
parent
ed3feb4d
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
6 deletions
+6
-6
include/ck/tensor_operation/gpu/device/device_contraction_multiple_d_xdl_cshuffle.hpp
...gpu/device/device_contraction_multiple_d_xdl_cshuffle.hpp
+6
-6
No files found.
include/ck/tensor_operation/gpu/device/device_contraction_multiple_d_xdl_cshuffle.hpp
View file @
c84e739d
...
...
@@ -172,8 +172,8 @@ struct DeviceContractionMultipleD_Xdl_CShuffle
assert
(
a_ms_ks_lengths_vec
.
size
()
==
NumDimM
+
NumDimK
&&
a_ms_ks_strides_vec
.
size
()
==
NumDimM
+
NumDimK
);
const
auto
to_tuple
=
[
&
](
auto
&
vec
,
auto
N
um
)
{
return
generate_tuple
([
&
](
auto
i
)
{
return
vec
[
i
];
},
N
um
);
const
auto
to_tuple
=
[
&
](
auto
&
vec
,
auto
n
um
)
{
return
generate_tuple
([
&
](
auto
i
)
{
return
vec
[
i
];
},
n
um
);
};
const
auto
a_ms_ns_lengths
=
to_tuple
(
a_ms_ks_lengths_vec
,
Number
<
NumDimM
+
NumDimK
>
{});
...
...
@@ -301,8 +301,8 @@ struct DeviceContractionMultipleD_Xdl_CShuffle
assert
(
b_ns_ks_lengths_vec
.
size
()
==
NumDimN
+
NumDimK
&&
b_ns_ks_strides_vec
.
size
()
==
NumDimN
+
NumDimK
);
const
auto
to_tuple
=
[
&
](
auto
&
vec
,
auto
N
um
)
{
return
generate_tuple
([
&
](
auto
i
)
{
return
vec
[
i
];
},
N
um
);
const
auto
to_tuple
=
[
&
](
auto
&
vec
,
auto
n
um
)
{
return
generate_tuple
([
&
](
auto
i
)
{
return
vec
[
i
];
},
n
um
);
};
const
auto
b_ns_ks_lengths
=
to_tuple
(
b_ns_ks_lengths_vec
,
Number
<
NumDimN
+
NumDimK
>
{});
...
...
@@ -430,8 +430,8 @@ struct DeviceContractionMultipleD_Xdl_CShuffle
assert
(
e_ms_ns_lengths_vec
.
size
()
==
NumDimM
+
NumDimN
&&
e_ms_ns_strides_vec
.
size
()
==
NumDimM
+
NumDimN
);
const
auto
to_tuple
=
[
&
](
auto
&
vec
,
auto
N
um
)
{
return
generate_tuple
([
&
](
auto
i
)
{
return
vec
[
i
];
},
N
um
);
const
auto
to_tuple
=
[
&
](
auto
&
vec
,
auto
n
um
)
{
return
generate_tuple
([
&
](
auto
i
)
{
return
vec
[
i
];
},
n
um
);
};
const
auto
e_ms_ns_lengths
=
to_tuple
(
e_ms_ns_lengths_vec
,
Number
<
NumDimM
+
NumDimN
>
{});
...
...
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