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
0016f6ac
Unverified
Commit
0016f6ac
authored
Feb 10, 2023
by
ltqin
Committed by
GitHub
Feb 10, 2023
Browse files
Merge pull request #574 from ROCmSoftwarePlatform/attn-fwd-train-dropout
fixed bug in philox_rand.hpp
parents
0e7aeef5
cb2d4dbb
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
6 deletions
+3
-6
include/ck/utility/philox_rand.hpp
include/ck/utility/philox_rand.hpp
+3
-6
No files found.
include/ck/utility/philox_rand.hpp
View file @
0016f6ac
...
...
@@ -109,12 +109,9 @@ class philox
__device__
uint2
u32_high_low_multi
(
const
unsigned
int
a
,
const
unsigned
int
b
)
{
uint2
*
res
;
uint2
tmp_res
;
asm
(
"v_mul_hi_u32 %0, %2, %3
\n\t
"
"v_mul_lo_u32 %1, %2, %3
\n\t
"
:
"=v"
(
tmp_res
.
x
),
"=v"
(
tmp_res
.
y
)
:
"v"
(
a
),
"v"
(
b
));
res
=
&
tmp_res
;
unsigned
long
long
tmp
;
tmp
=
static_cast
<
unsigned
long
long
>
(
a
)
*
b
;
res
=
reinterpret_cast
<
uint2
*>
(
&
tmp
);
return
*
res
;
}
...
...
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