Skip to content

Commit 196c9f9

Browse files
authored
Merge pull request #13192 from seiriosPlus/cherry-pick-selected_rows_size_fix
Cherry-pick-SelectedRows-Size-Fix
2 parents 2e8a95c + 3745143 commit 196c9f9

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
@@ -1060,7 +1060,8 @@ def _create_table_optimize_block(self, pserver_index, pserver_program,
10601060
self.table_name]
10611061

10621062
zero_dim = int(
1063-
math.ceil(origin_param_var.shape[0] / len(self.pserver_endpoints)))
1063+
math.ceil(origin_param_var.shape[0] / float(
1064+
len(self.pserver_endpoints))))
10641065
table_shape = list(origin_param_var.shape)
10651066
table_shape[0] = zero_dim
10661067

0 commit comments

Comments
 (0)