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
77d8310e
Commit
77d8310e
authored
Mar 24, 2023
by
guangzlu
Browse files
modified blockwise_dropout.hpp
parent
3a9ab7a7
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
10 deletions
+2
-10
include/ck/tensor_operation/gpu/block/blockwise_dropout.hpp
include/ck/tensor_operation/gpu/block/blockwise_dropout.hpp
+2
-10
No files found.
include/ck/tensor_operation/gpu/block/blockwise_dropout.hpp
View file @
77d8310e
...
@@ -83,20 +83,12 @@ struct BlockwiseDropout
...
@@ -83,20 +83,12 @@ struct BlockwiseDropout
constexpr
int
tmp_size
=
MRepeat
*
KRepeat
;
constexpr
int
tmp_size
=
MRepeat
*
KRepeat
;
int
hiprand_calls
=
tmp_size
/
8
;
int
hiprand_calls
=
tmp_size
;
float
tmp
[
tmp_size
];
float
tmp
[
tmp_size
];
for
(
int
i
=
0
;
i
<
hiprand_calls
;
i
++
)
for
(
int
i
=
0
;
i
<
hiprand_calls
;
i
++
)
{
{
float
tmp_rand
=
hiprand_uniform
(
&
state
);
tmp
[
i
]
=
hiprand_uniform
(
&
state
);
tmp
[
i
]
=
tmp_rand
;
tmp
[
i
+
1
]
=
tmp_rand
;
tmp
[
i
+
2
]
=
tmp_rand
;
tmp
[
i
+
3
]
=
tmp_rand
;
tmp
[
i
+
4
]
=
tmp_rand
;
tmp
[
i
+
5
]
=
tmp_rand
;
tmp
[
i
+
6
]
=
tmp_rand
;
tmp
[
i
+
7
]
=
tmp_rand
;
}
}
block_sync_lds
();
block_sync_lds
();
...
...
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