fix dim_test.cu bug

8 years ago · 857cd24a35
parent 11163dfc7a
commit 857cd24a35
1 changed files with 1 additions and 1 deletions
--- a/paddle/framework/dim_test.cu
+++ b/paddle/framework/dim_test.cu
@ -8,7 +8,7 @@ __global__ void test(paddle::framework::Dim<2>* o) {
  o[0] = paddle::framework::make_dim(5, 6);
 }

-__global__ void dyn_idx_gpu(int* o) {
+__global__ void dyn_idx_gpu(int64_t* o) {
  auto d = paddle::framework::make_dim(5, 6);
  o[0] = d[1];
 }