diff --git a/paddle/framework/dim_test.cu b/paddle/framework/dim_test.cu
index f42b1eeadf30428ad26e6b714e002d5935d24cbc..0a6a87669c900de6cb507dd48f0cfc871defe279 100644
--- 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];
 }