Merge pull request #424 from denghuilu/master

amcadmus · web-flow · commit 3a59596dbd3b · 2021-03-20T00:53:44.000+08:00
fix bug of max_nbor_size error
diff --git a/source/op/descrpt_se_a_multi_device.cc b/source/op/descrpt_se_a_multi_device.cc
@@ -261,7 +261,7 @@ class DescrptSeAOp : public OpKernel {
             cudaErrcheck(cudaMemcpy(jlist,  jlist_host,  sizeof(int) * mesh_host[3], cudaMemcpyHostToDevice));
             
             max_nbor_size = 0;
-            for(int ii = 0; ii < mesh_host[2]; ii++) {
+            for(int ii = 0; ii < mesh_host[2] - 1; ii++) {
                 max_nbor_size = (jrange_host[ii + 1] - jrange_host[ii]) > max_nbor_size ? (jrange_host[ii + 1] - jrange_host[ii]) : max_nbor_size;
             }
             assert(max_nbor_size <= GPU_MAX_NBOR_SIZE);
diff --git a/source/op/descrpt_se_r_multi_device.cc b/source/op/descrpt_se_r_multi_device.cc
@@ -250,7 +250,7 @@ class DescrptSeROp : public OpKernel {
             cudaErrcheck(cudaMemcpy(jlist,  jlist_host,  sizeof(int) * mesh_host[3], cudaMemcpyHostToDevice));
 
             max_nbor_size = 0;
-            for(int ii = 0; ii < mesh_host[2]; ii++) {
+            for(int ii = 0; ii < mesh_host[2] - 1; ii++) {
                 max_nbor_size = (jrange_host[ii + 1] - jrange_host[ii]) > max_nbor_size ? (jrange_host[ii + 1] - jrange_host[ii]) : max_nbor_size;
             }
             assert(max_nbor_size <= GPU_MAX_NBOR_SIZE);

Original file line number	Diff line number	Diff line change
`@@ -261,7 +261,7 @@ class DescrptSeAOp : public OpKernel {`
`261`	`261`	`cudaErrcheck(cudaMemcpy(jlist, jlist_host, sizeof(int) * mesh_host[3], cudaMemcpyHostToDevice));`
`262`	`262`
`263`	`263`	`max_nbor_size = 0;`
`264`		`- for(int ii = 0; ii < mesh_host[2]; ii++) {`
	`264`	`+ for(int ii = 0; ii < mesh_host[2] - 1; ii++) {`
`265`	`265`	`max_nbor_size = (jrange_host[ii + 1] - jrange_host[ii]) > max_nbor_size ? (jrange_host[ii + 1] - jrange_host[ii]) : max_nbor_size;`
`266`	`266`	`}`
`267`	`267`	`assert(max_nbor_size <= GPU_MAX_NBOR_SIZE);`
Original file line number	Diff line number	Diff line change
`@@ -250,7 +250,7 @@ class DescrptSeROp : public OpKernel {`
`250`	`250`	`cudaErrcheck(cudaMemcpy(jlist, jlist_host, sizeof(int) * mesh_host[3], cudaMemcpyHostToDevice));`
`251`	`251`
`252`	`252`	`max_nbor_size = 0;`
`253`		`- for(int ii = 0; ii < mesh_host[2]; ii++) {`
	`253`	`+ for(int ii = 0; ii < mesh_host[2] - 1; ii++) {`
`254`	`254`	`max_nbor_size = (jrange_host[ii + 1] - jrange_host[ii]) > max_nbor_size ? (jrange_host[ii + 1] - jrange_host[ii]) : max_nbor_size;`
`255`	`255`	`}`
`256`	`256`	`assert(max_nbor_size <= GPU_MAX_NBOR_SIZE);`