(self, server_state)
| 192 | ) = state |
| 193 | |
| 194 | def process_request(self, server_state): |
| 195 | kv_store = server_state.kv_store |
| 196 | dtype = F.data_type_dict[self.dtype] |
| 197 | |
| 198 | # We should see requests from multiple clients. We need to ignore the duplicated |
| 199 | # reqeusts. |
| 200 | if self.name in kv_store.data_store: |
| 201 | assert tuple(F.shape(kv_store.data_store[self.name])) == tuple( |
| 202 | self.shape |
| 203 | ) |
| 204 | assert ( |
| 205 | F.reverse_data_type_dict[ |
| 206 | F.dtype(kv_store.data_store[self.name]) |
| 207 | ] |
| 208 | == self.dtype |
| 209 | ) |
| 210 | assert kv_store.part_policy[self.name].policy_str == self.policy_str |
| 211 | else: |
| 212 | if not kv_store.is_backup_server(): |
| 213 | data_tensor = self.init_func(self.shape, dtype) |
| 214 | kv_store.init_data( |
| 215 | name=self.name, |
| 216 | policy_str=self.policy_str, |
| 217 | data_tensor=data_tensor, |
| 218 | ) |
| 219 | else: |
| 220 | kv_store.init_data(name=self.name, policy_str=self.policy_str) |
| 221 | res = InitDataResponse(INIT_MSG) |
| 222 | return res |
| 223 | |
| 224 | |
| 225 | BARRIER = 901234 |
nothing calls this directly
no test coverage detected