Commit de43a6d8 authored by guangzlu's avatar guangzlu
Browse files

fixed philox generator

parent 8185c3d3
...@@ -219,7 +219,8 @@ int run(int argc, char* argv[]) ...@@ -219,7 +219,8 @@ int run(int argc, char* argv[])
b1_element_op, b1_element_op,
c_element_op, c_element_op,
0, // dropout ratio 0, // dropout ratio
{0, 448}); // dropout random seed and offset {0, 448}); // dropout random seed and offset, offset should be
// at least the number of elements on a thread
// specify workspace for problem_desc // specify workspace for problem_desc
DeviceMem problem_desc_workspace(gemm.GetWorkSpaceSize(&argument)); DeviceMem problem_desc_workspace(gemm.GetWorkSpaceSize(&argument));
......
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