Skip to content

Commit 6bc16ff

Browse files
authored
Merge pull request #13186 from seiriosPlus/selected_rows_size_fix
bug fix
2 parents 3b6090e + ff618a2 commit 6bc16ff

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/paddle/fluid/transpiler/distribute_transpiler.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1096,7 +1096,8 @@ def _create_table_optimize_block(self, pserver_index, pserver_program,
10961096
self.table_name]
10971097

10981098
zero_dim = int(
1099-
math.ceil(origin_param_var.shape[0] / len(self.pserver_endpoints)))
1099+
math.ceil(origin_param_var.shape[0] / float(
1100+
len(self.pserver_endpoints))))
11001101
table_shape = list(origin_param_var.shape)
11011102
table_shape[0] = zero_dim
11021103

0 commit comments

Comments
 (0)