Fix compiling warning.

qingqing01 · qingqing01 · commit e88e1964eb79 · 2017-10-31T18:10:58.000+08:00
diff --git a/paddle/operators/nccl_op_test.cu b/paddle/operators/nccl_op_test.cu
@@ -185,7 +185,7 @@ TEST_F(NCCLTester, ncclAllReduceOp) {
         recv_tensor.numel() * sizeof(float),
         static_cast<p::CUDADeviceContext *>(dev_ctxs[i])->stream());
 
-    for (size_t j = 0; j < f::product(kDims); ++j) {
+    for (int64_t j = 0; j < f::product(kDims); ++j) {
       ASSERT_NEAR(ct[j], result, 1e-5);
     }
   }
@@ -234,7 +234,7 @@ TEST_F(NCCLTester, ncclReduceOp) {
       recv_tensor.numel() * sizeof(float),
       static_cast<p::CUDADeviceContext *>(dev_ctxs[kRoot])->stream());
 
-  for (int j = 0; j < f::product(kDims); ++j) {
+  for (int64_t j = 0; j < f::product(kDims); ++j) {
     ASSERT_NEAR(ct[j], result, 1e-5);
   }
 }
@@ -282,7 +282,7 @@ TEST_F(NCCLTester, ncclBcastOp) {
       recv_tensor.numel() * sizeof(float),
       static_cast<p::CUDADeviceContext *>(dev_ctxs[idx])->stream());
 
-  for (size_t j = 0; j < f::product(kDims); ++j) {
+  for (int64_t j = 0; j < f::product(kDims); ++j) {
     ASSERT_NEAR(ct[j], result, 1e-5);
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -185,7 +185,7 @@ TEST_F(NCCLTester, ncclAllReduceOp) {`
`185`	`185`	`recv_tensor.numel() * sizeof(float),`
`186`	`186`	`static_cast<p::CUDADeviceContext *>(dev_ctxs[i])->stream());`
`187`	`187`
`188`		`- for (size_t j = 0; j < f::product(kDims); ++j) {`
	`188`	`+ for (int64_t j = 0; j < f::product(kDims); ++j) {`
`189`	`189`	`ASSERT_NEAR(ct[j], result, 1e-5);`
`190`	`190`	`}`
`191`	`191`	`}`
`@@ -234,7 +234,7 @@ TEST_F(NCCLTester, ncclReduceOp) {`
`234`	`234`	`recv_tensor.numel() * sizeof(float),`
`235`	`235`	`static_cast<p::CUDADeviceContext *>(dev_ctxs[kRoot])->stream());`
`236`	`236`
`237`		`- for (int j = 0; j < f::product(kDims); ++j) {`
	`237`	`+ for (int64_t j = 0; j < f::product(kDims); ++j) {`
`238`	`238`	`ASSERT_NEAR(ct[j], result, 1e-5);`
`239`	`239`	`}`
`240`	`240`	`}`
`@@ -282,7 +282,7 @@ TEST_F(NCCLTester, ncclBcastOp) {`
`282`	`282`	`recv_tensor.numel() * sizeof(float),`
`283`	`283`	`static_cast<p::CUDADeviceContext *>(dev_ctxs[idx])->stream());`
`284`	`284`
`285`		`- for (size_t j = 0; j < f::product(kDims); ++j) {`
	`285`	`+ for (int64_t j = 0; j < f::product(kDims); ++j) {`
`286`	`286`	`ASSERT_NEAR(ct[j], result, 1e-5);`
`287`	`287`	`}`
`288`	`288`	`}`