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
ea9d7396
"...composable_kernel.git" did not exist on "9d3f634a3cc37599db9f3ebfeb1f9a3c45d9a673"
Commit
ea9d7396
authored
Sep 22, 2023
by
Jing Zhang
Browse files
update example
parent
f60ad8b9
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
7 deletions
+4
-7
example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp
..._contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp
+4
-7
No files found.
example/61_contraction_multi_ABD/contraction_multi_ABD_xdl_fp16.cpp
View file @
ea9d7396
...
...
@@ -25,9 +25,6 @@ using S = ck::Sequence<Is...>;
using
F16
=
ck
::
half_t
;
using
F32
=
float
;
using
Row
=
ck
::
tensor_layout
::
gemm
::
RowMajor
;
using
Col
=
ck
::
tensor_layout
::
gemm
::
ColumnMajor
;
using
PassThrough
=
ck
::
tensor_operation
::
element_wise
::
PassThrough
;
using
A0DataType
=
F16
;
...
...
@@ -130,19 +127,19 @@ int main(int argc, char* argv[])
// A0[M0, M1, K0, K1]
std
::
vector
<
ck
::
index_t
>
a0_ms_ks_lengths
{
30
,
128
,
32
,
64
};
std
::
vector
<
ck
::
index_t
>
a0_ms_ks_strides
{
524288
,
4096
,
128
,
1
};
std
::
vector
<
ck
::
index_t
>
a0_ms_ks_strides
{
128
*
32
*
64
,
32
*
64
,
64
,
1
};
// A1[M1, K1] -> A1[M0, M1, K0, K1]
std
::
vector
<
ck
::
index_t
>
a1_ms_ks_lengths
{
30
,
128
,
32
,
64
};
std
::
vector
<
ck
::
index_t
>
a1_ms_ks_strides
{
0
,
64
,
0
,
1
};
// B[N0, N1, K0, K1]
std
::
vector
<
ck
::
index_t
>
b_ns_ks_lengths
{
32
,
64
,
32
,
64
};
std
::
vector
<
ck
::
index_t
>
b_ns_ks_strides
{
524288
,
4096
,
128
,
1
};
std
::
vector
<
ck
::
index_t
>
b_ns_ks_strides
{
64
*
32
*
64
,
32
*
64
,
64
,
1
};
// D[M0, M1, N0, N1]
std
::
vector
<
ck
::
index_t
>
d_ms_ns_lengths
{
30
,
128
,
32
,
64
};
std
::
vector
<
ck
::
index_t
>
d_ms_ns_strides
{
524288
,
4096
,
128
,
1
};
std
::
vector
<
ck
::
index_t
>
d_ms_ns_strides
{
128
*
32
*
64
,
32
*
64
,
64
,
1
};
// E[M0, M1, N0, N1]
std
::
vector
<
ck
::
index_t
>
e_ms_ns_lengths
{
30
,
128
,
32
,
64
};
std
::
vector
<
ck
::
index_t
>
e_ms_ns_strides
{
524288
,
4096
,
128
,
1
};
std
::
vector
<
ck
::
index_t
>
e_ms_ns_strides
{
128
*
32
*
64
,
32
*
64
,
64
,
1
};
if
(
argc
==
1
)
{
...
...
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