repo-sync-2024-12-19T16:16:02+0800 (#160)

secretflow · Dec 19, 2024 · b7b0e31 · b7b0e31
1 parent 16a1910
commit b7b0e31
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/heu/library/algorithms/paillier_clustar_fpga/fpga_engine/driver/driver.cc b/heu/library/algorithms/paillier_clustar_fpga/fpga_engine/driver/driver.cc
@@ -797,9 +797,10 @@ int fpga_fedai_operator_accl(fpga_config *cfg, char *para, char *data1,
       }
       // Check whether data_size is supported
       unsigned long long max_task_memspace = 0;
-      max_task_memspace = memory_total_size[fpga_id] / taskid_counts[fpga_id] *
-                              1024 * 1024 * 1024 -
-                          16 * 1024;
+      max_task_memspace =
+          static_cast<unsigned long long>(memory_total_size[fpga_id]) /
+              taskid_counts[fpga_id] * 1024 * 1024 * 1024 -
+          16 * 1024;  // Updated by Ant Group
       // fprintf(stdout,"cmd_size %ld, para_size %ld, data_size %ld, total_size
       // %ld, max_task_space %ld\n",16*1024,cfg->para_data_size,
       // cfg->data1_size,16*1024+cfg->para_data_size+cfg->data1_size,max_task_memspace+16*1024);

diff --git a/heu/pylib/version.py b/heu/pylib/version.py
@@ -13,4 +13,4 @@
 # limitations under the License.
 
 
-__version__ = "0.6.0.dev20241106"
+__version__ = "0.6.0.dev0"
Original file line number	Diff line number	Diff line change
Expand Up		@@ -13,4 +13,4 @@
		# limitations under the License.


		__version__ = "0.6.0.dev20241106"
		__version__ = "0.6.0.dev0"