Skip to content

GitLab

Commit 907413ac authored by guangzlu's avatar guangzlu
Browse files

fixed bug in dropout with global pos

parent 95a9bacf
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
include/ck/tensor_operation/gpu/block/blockwise_dropout.hpp
View file @ 907413ac
......@@ -69,7 +69,7 @@ struct BlockwiseDropout
ushort tmp[tmp_size];
for(int i = 0; i < philox_calls; i++)
{
ph.get_random_8x16((tmp + i * 8), element_global_1d_id);
ph.get_random_8x16((tmp + i * 8), element_global_1d_id + i * 8);
}
block_sync_lds();
......@@ -142,7 +142,7 @@ struct BlockwiseDropout
ushort tmp[tmp_size];
for(int i = 0; i < philox_calls; i++)
{
ph.get_random_8x16((tmp + i * 8), element_global_1d_id + philox_calls * 8);
ph.get_random_8x16((tmp + i * 8), element_global_1d_id + i * 8);
}
block_sync_lds();
......@@ -184,7 +184,7 @@ struct BlockwiseDropout
ushort tmp[tmp_size];
for(int i = 0; i < philox_calls; i++)
{
ph.get_random_8x16((tmp + i * 8), element_global_1d_id);
ph.get_random_8x16((tmp + i * 8), element_global_1d_id + i * 8);
}
block_sync_lds();
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment