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
a6ea41de
Commit
a6ea41de
authored
Jun 02, 2022
by
Chao Liu
Browse files
update examle
parent
4746e11b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
example/23_contraction/contraction_xdl_fp32.cpp
example/23_contraction/contraction_xdl_fp32.cpp
+3
-3
No files found.
example/23_contraction/contraction_xdl_fp32.cpp
View file @
a6ea41de
...
@@ -215,13 +215,13 @@ int main(int argc, char* argv[])
...
@@ -215,13 +215,13 @@ int main(int argc, char* argv[])
}
}
// A[M0, M1, K0, K1]
// A[M0, M1, K0, K1]
std
::
vector
<
ck
::
index_t
>
a_ms_ks_lengths
{
30
,
128
,
32
,
128
};
std
::
vector
<
ck
::
index_t
>
a_ms_ks_lengths
{
30
,
128
,
16
,
128
};
std
::
vector
<
ck
::
index_t
>
a_ms_ks_strides
{
524288
,
4096
,
128
,
1
};
std
::
vector
<
ck
::
index_t
>
a_ms_ks_strides
{
524288
,
4096
,
128
,
1
};
// B[K0, K1, N0, N1]
// B[K0, K1, N0, N1]
std
::
vector
<
ck
::
index_t
>
b_ks_ns_lengths
{
32
,
128
,
32
,
128
};
std
::
vector
<
ck
::
index_t
>
b_ks_ns_lengths
{
16
,
128
,
16
,
128
};
std
::
vector
<
ck
::
index_t
>
b_ks_ns_strides
{
128
,
1
,
524288
,
4096
};
std
::
vector
<
ck
::
index_t
>
b_ks_ns_strides
{
128
,
1
,
524288
,
4096
};
// C[M0, M1, N0, N1]
// C[M0, M1, N0, N1]
std
::
vector
<
ck
::
index_t
>
c_ms_ns_lengths
{
30
,
128
,
32
,
128
};
std
::
vector
<
ck
::
index_t
>
c_ms_ns_lengths
{
30
,
128
,
16
,
128
};
std
::
vector
<
ck
::
index_t
>
c_ms_ns_strides
{
524288
,
4096
,
128
,
1
};
std
::
vector
<
ck
::
index_t
>
c_ms_ns_strides
{
524288
,
4096
,
128
,
1
};
Tensor
<
ADataType
>
a_ms_ks
(
Tensor
<
ADataType
>
a_ms_ks
(
...
...
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