fix shape type in uniform_random_op.cu

7 years ago · cb1ccc710b
parent d8b697357f
commit cb1ccc710b
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/uniform_random_op.cu
+++ b/paddle/fluid/operators/uniform_random_op.cu
@ -48,7 +48,7 @@ class GPUUniformRandomKernel : public framework::OpKernel<T> {
    if (out_var->IsType<framework::LoDTensor>()) {
      tensor = out_var->GetMutable<framework::LoDTensor>();
    } else if (out_var->IsType<framework::SelectedRows>()) {
-      auto shape = context.Attr<std::vector<int>>("shape");
+      auto shape = context.Attr<std::vector<int64_t>>("shape");
      tensor = out_var->GetMutable<framework::SelectedRows>()->mutable_value();
      tensor->Resize(framework::make_ddim(shape));
    } else {