(self, include: Container[OffloadStateTypeEnum])
| 5559 | return self._is_compiled |
| 5560 | |
| 5561 | def _refine_include_states(self, include: Container[OffloadStateTypeEnum]) -> Container[OffloadStateTypeEnum]: |
| 5562 | if include is None: |
| 5563 | include = list(OffloadStateTypeEnum) |
| 5564 | |
| 5565 | if self.zero_use_cpu_optimizer(): |
| 5566 | exclude_states = [OffloadStateTypeEnum.hp_params, OffloadStateTypeEnum.optim_states] |
| 5567 | if self.zero_optimization_partition_weights(): |
| 5568 | exclude_states.append(OffloadStateTypeEnum.lp_grads) |
| 5569 | include = [x for x in include if x not in exclude_states] |
| 5570 | |
| 5571 | return include |
| 5572 | |
| 5573 | def offload_states(self, |
| 5574 | include: Container[OffloadStateTypeEnum] = None, |
no test coverage detected