Commit 48920afb authored by Paul's avatar Paul
Browse files

Use different seeds for different parameters

parent 3755e41e
...@@ -98,9 +98,10 @@ migraph::argument run_cpu() ...@@ -98,9 +98,10 @@ migraph::argument run_cpu()
auto_print pp{p, 0}; auto_print pp{p, 0};
compile_check(p, migraph::cpu::cpu_target{}); compile_check(p, migraph::cpu::cpu_target{});
migraph::program::parameter_map m; migraph::program::parameter_map m;
int seed = 0;
for(auto&& x : p.get_parameter_shapes()) for(auto&& x : p.get_parameter_shapes())
{ {
m[x.first] = migraph::generate_argument(x.second); m[x.first] = migraph::generate_argument(x.second, seed++);
} }
return p.eval(m); return p.eval(m);
} }
...@@ -114,9 +115,10 @@ migraph::argument run_gpu() ...@@ -114,9 +115,10 @@ migraph::argument run_gpu()
compile_check(p, migraph::gpu::target{}); compile_check(p, migraph::gpu::target{});
migraph::program::parameter_map m; migraph::program::parameter_map m;
int seed = 0;
for(auto&& x : p.get_parameter_shapes()) for(auto&& x : p.get_parameter_shapes())
{ {
m[x.first] = migraph::gpu::to_gpu(migraph::generate_argument(x.second)); m[x.first] = migraph::gpu::to_gpu(migraph::generate_argument(x.second, seed++));
} }
return migraph::gpu::from_gpu(p.eval(m)); return migraph::gpu::from_gpu(p.eval(m));
......
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