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
54f52881
"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "6131a93b969f87d171148bd367fd9990d5a49b6b"
Commit
54f52881
authored
Sep 30, 2022
by
wangshaojie6
Browse files
remove ignore in generating tuple of vector
parent
cabe8d80
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
8 deletions
+4
-8
include/ck/tensor_operation/gpu/grid/gridwise_layernorm_welford_variance.hpp
...peration/gpu/grid/gridwise_layernorm_welford_variance.hpp
+4
-8
No files found.
include/ck/tensor_operation/gpu/grid/gridwise_layernorm_welford_variance.hpp
View file @
54f52881
...
...
@@ -126,8 +126,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk
p_y_global
,
y_grid_desc_m_k
.
GetElementSpaceSize
());
auto
x_thread_buf
=
generate_tuple
(
[
&
](
auto
i
)
{
ignore
=
i
;
[
&
](
auto
)
{
return
StaticBuffer
<
AddressSpaceEnum
::
Vgpr
,
AccDataType
,
MThreadSliceSize
*
XSrcVectorSize
,
...
...
@@ -136,8 +135,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk
Number
<
XThreadBufferNumber
>
{});
auto
gamma_thread_buf
=
generate_tuple
(
[
&
](
auto
i
)
{
ignore
=
i
;
[
&
](
auto
)
{
return
StaticBuffer
<
AddressSpaceEnum
::
Vgpr
,
AccDataType
,
MThreadSliceSize
*
GammaSrcVectorSize
,
...
...
@@ -146,8 +144,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk
Number
<
GammaThreadBufferNumber
>
{});
auto
beta_thread_buf
=
generate_tuple
(
[
&
](
auto
i
)
{
ignore
=
i
;
[
&
](
auto
)
{
return
StaticBuffer
<
AddressSpaceEnum
::
Vgpr
,
AccDataType
,
MThreadSliceSize
*
BetaSrcVectorSize
,
...
...
@@ -156,8 +153,7 @@ struct GridwiseLayernormWelfordVariance_mk_to_mk
Number
<
BetaThreadBufferNumber
>
{});
auto
y_thread_buf
=
generate_tuple
(
[
&
](
auto
i
)
{
ignore
=
i
;
[
&
](
auto
)
{
return
StaticBuffer
<
AddressSpaceEnum
::
Vgpr
,
AccDataType
,
MThreadSliceSize
*
YDstVectorSize
,
...
...
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