fix a potential bug

denghuilu · denghuilu · commit 51de5ec5eb1d · 2021-01-27T11:37:59.000+08:00
diff --git a/source/op/descrpt_se_a_multi_device.cc b/source/op/descrpt_se_a_multi_device.cc
@@ -268,7 +268,7 @@ class DescrptSeAOp : public OpKernel {
             cudaErrcheck(cudaMemcpy(jlist,  jlist_host,  sizeof(int) * mesh_host[3], cudaMemcpyHostToDevice));
             
             max_nbor_size = 1024;
-            for(int ii = 0; ii < mesh_host[2] - 1; ii++) {
+            for(int ii = 0; ii < mesh_host[2]; ii++) {
                 max_nbor_size = (jrange_host[ii + 1] - jrange_host[ii]) > max_nbor_size ? (jrange_host[ii + 1] - jrange_host[ii]) : max_nbor_size;
             }
         }
diff --git a/source/op/descrpt_se_r_multi_device.cc b/source/op/descrpt_se_r_multi_device.cc
@@ -257,7 +257,7 @@ class DescrptSeROp : public OpKernel {
             cudaErrcheck(cudaMemcpy(jlist,  jlist_host,  sizeof(int) * mesh_host[3], cudaMemcpyHostToDevice));
 
             max_nbor_size = 1024;
-            for(int ii = 0; ii < mesh_host[2] - 1; ii++) {
+            for(int ii = 0; ii < mesh_host[2]; ii++) {
                 max_nbor_size = (jrange_host[ii + 1] - jrange_host[ii]) > max_nbor_size ? (jrange_host[ii + 1] - jrange_host[ii]) : max_nbor_size;
             }
         }

Original file line number	Diff line number	Diff line change
`@@ -268,7 +268,7 @@ class DescrptSeAOp : public OpKernel {`
`268`	`268`	`cudaErrcheck(cudaMemcpy(jlist, jlist_host, sizeof(int) * mesh_host[3], cudaMemcpyHostToDevice));`
`269`	`269`
`270`	`270`	`max_nbor_size = 1024;`
`271`		`- for(int ii = 0; ii < mesh_host[2] - 1; ii++) {`
	`271`	`+ for(int ii = 0; ii < mesh_host[2]; ii++) {`
`272`	`272`	`max_nbor_size = (jrange_host[ii + 1] - jrange_host[ii]) > max_nbor_size ? (jrange_host[ii + 1] - jrange_host[ii]) : max_nbor_size;`
`273`	`273`	`}`
`274`	`274`	`}`
Original file line number	Diff line number	Diff line change
`@@ -257,7 +257,7 @@ class DescrptSeROp : public OpKernel {`
`257`	`257`	`cudaErrcheck(cudaMemcpy(jlist, jlist_host, sizeof(int) * mesh_host[3], cudaMemcpyHostToDevice));`
`258`	`258`
`259`	`259`	`max_nbor_size = 1024;`
`260`		`- for(int ii = 0; ii < mesh_host[2] - 1; ii++) {`
	`260`	`+ for(int ii = 0; ii < mesh_host[2]; ii++) {`
`261`	`261`	`max_nbor_size = (jrange_host[ii + 1] - jrange_host[ii]) > max_nbor_size ? (jrange_host[ii + 1] - jrange_host[ii]) : max_nbor_size;`
`262`	`262`	`}`
`263`	`263`	`}`