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 ...@@ -69,7 +69,7 @@ struct BlockwiseDropout
ushort tmp[tmp_size]; ushort tmp[tmp_size];
for(int i = 0; i < philox_calls; i++) 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(); block_sync_lds();
...@@ -142,7 +142,7 @@ struct BlockwiseDropout ...@@ -142,7 +142,7 @@ struct BlockwiseDropout
ushort tmp[tmp_size]; ushort tmp[tmp_size];
for(int i = 0; i < philox_calls; i++) 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(); block_sync_lds();
...@@ -184,7 +184,7 @@ struct BlockwiseDropout ...@@ -184,7 +184,7 @@ struct BlockwiseDropout
ushort tmp[tmp_size]; ushort tmp[tmp_size];
for(int i = 0; i < philox_calls; i++) 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(); 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