(self, name, ndarray)
| 436 | _save(opt_dict, optim_path) |
| 437 | |
| 438 | def _set_var(self, name, ndarray): |
| 439 | t = global_scope().find_var(name).get_tensor() |
| 440 | p = t._place() |
| 441 | if p.is_cpu_place(): |
| 442 | place = paddle.base.CPUPlace() |
| 443 | elif p.is_cuda_pinned_place(): |
| 444 | place = paddle.base.CUDAPinnedPlace() |
| 445 | elif p.is_xpu_place(): |
| 446 | p = paddle.base.core.Place() |
| 447 | p.set_place(t._place()) |
| 448 | place = paddle.base.XPUPlace(p.xpu_device_id()) |
| 449 | elif p.is_custom_place(): |
| 450 | p = paddle.base.core.Place() |
| 451 | p.set_place(t._place()) |
| 452 | place = paddle.base.CustomPlace( |
| 453 | paddle.device.get_device().split(':')[0], p.custom_device_id() |
| 454 | ) |
| 455 | else: |
| 456 | p = paddle.base.core.Place() |
| 457 | p.set_place(t._place()) |
| 458 | place = paddle.base.CUDAPlace(p.gpu_device_id()) |
| 459 | |
| 460 | t.set(ndarray, place) |
| 461 | |
| 462 | def load(self, param_state_pairs, optim_state): |
| 463 | if self._executor is None: |
no test coverage detected