Skip to content

Commit 90f1057

Browse files
committed
fix bug in rc-fixer buffer allocation
1 parent 6afc136 commit 90f1057

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

mpisppy/cylinders/reduced_costs_spoke.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,8 @@ def register_send_fields(self) -> None:
6060
scenario_buffer_len = 0
6161
for s in self.opt.local_scenarios.values():
6262
scenario_buffer_len += len(s._mpisppy_data.nonant_indices)
63-
self._scenario_rc_buffer = np.zeros(self.nonant_length)
63+
self._scenario_rc_buffer = np.zeros(scenario_buffer_len)
64+
self._scenario_rc_len = scenario_buffer_len # used in an assert
6465

6566
self.initialize_bound_fields()
6667
self.create_integer_variable_where()
@@ -194,6 +195,7 @@ def extract_and_store_reduced_costs(self):
194195
rc[ci] = np.nan
195196

196197
self._scenario_rc_buffer.fill(0)
198+
assert self._scenario_rc_buffer.size == self._scenario_rc_len
197199
ci = 0 # buffer index
198200
for sub in self.opt.local_subproblems.values():
199201
for sn in sub.scen_list:

0 commit comments

Comments
 (0)