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
414f3ac9
"git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "5489bda9bcc9fed83fe45832de033e85bb31649f"
Commit
414f3ac9
authored
Oct 02, 2023
by
Jing Zhang
Browse files
merge develop
parents
a60cd1d9
9d58c421
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
0 deletions
+2
-0
example/60_gemm_multi_ABD/gemm_multi_ABD_xdl_fp16.cpp
example/60_gemm_multi_ABD/gemm_multi_ABD_xdl_fp16.cpp
+1
-0
include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_abd_xdl_cshuffle.hpp
...tion/gpu/grid/gridwise_gemm_multiple_abd_xdl_cshuffle.hpp
+1
-0
No files found.
example/60_gemm_multi_ABD/gemm_multi_ABD_xdl_fp16.cpp
View file @
414f3ac9
...
@@ -74,6 +74,7 @@ struct AddScale
...
@@ -74,6 +74,7 @@ struct AddScale
a
=
scale
*
(
a0
+
a1
);
a
=
scale
*
(
a0
+
a1
);
}
}
// this attribute will force copy_function applying element_wise with vector_type
static
constexpr
ck
::
index_t
vec_len
=
4
;
static
constexpr
ck
::
index_t
vec_len
=
4
;
float
scale
=
1.0
;
float
scale
=
1.0
;
...
...
include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_abd_xdl_cshuffle.hpp
View file @
414f3ac9
...
@@ -427,6 +427,7 @@ struct GridwiseGemmMultipleABD_xdl_cshuffle
...
@@ -427,6 +427,7 @@ struct GridwiseGemmMultipleABD_xdl_cshuffle
return
generate_tuple
(
return
generate_tuple
(
[
&
](
auto
i
)
{
[
&
](
auto
i
)
{
using
ALayout
=
remove_cvref_t
<
tuple_element_t
<
i
.
value
,
AsLayout
>>
;
using
ALayout
=
remove_cvref_t
<
tuple_element_t
<
i
.
value
,
AsLayout
>>
;
return
MakeAGridDescriptor_M_K
<
ALayout
,
GemmSpec
>
(
MRaws
[
i
],
KRaws
[
i
],
AsStride
[
i
]);
return
MakeAGridDescriptor_M_K
<
ALayout
,
GemmSpec
>
(
MRaws
[
i
],
KRaws
[
i
],
AsStride
[
i
]);
},
},
Number
<
NumATensor
>
{});
Number
<
NumATensor
>
{});
...
...
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