Diff Coverage

Diff: origin/r1.0.0...HEAD, staged and unstaged changes

Source File Diff Coverage (%) Missing Lines
hyper_parallel/platform/mindspore/activation_checkpoint/activation_swap.py 40.0% 363,366,371
hyper_parallel/platform/torch/activation_checkpoint/activation_swap.py 40.0% 188,191,196
hyper_parallel/platform/mindspore/activation_checkpoint/activation_swap.py
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
        swap_manager = SwapManager()

        def pack_to_cpu(tensor: ms.Tensor):
            if not base_check_fn(tensor):
                return tensor.detach()
            if policy_fn is not None:
                if policy_fn(tensor) == CheckpointPolicy.MUST_SAVE:
                    return tensor.detach()
                if policy_fn(tensor) != CheckpointPolicy.MUST_SWAP:
                    raise RuntimeError(f"Swap :set an invalid policy {policy_fn(tensor)}")
            group_name = swap_manager.get_current_group_name()
            if not group_name:
                return tensor.detach()
            if not self.add_to_storage:
                swap_manager.add_storage(group_name, self.storage)
                self.add_to_storage = True
            funcname = f"{group_name}::{tensor.shape}"
hyper_parallel/platform/torch/activation_checkpoint/activation_swap.py
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
        swap_manager = SwapManager()

        def pack_to_cpu(tensor: torch.Tensor):
            if not base_check_fn(tensor):
                return tensor.detach()
            if policy_fn is not None:
                if policy_fn(tensor) == CheckpointPolicy.MUST_SAVE:
                    return tensor.detach()
                if policy_fn(tensor) != CheckpointPolicy.MUST_SWAP:
                    raise RuntimeError(f"Swap :set an invalid policy {policy_fn(tensor)}")
            group_name = swap_manager.get_current_group_name()
            if not group_name:
                return tensor.detach()
            if not self.add_to_storage:
                swap_manager.add_storage(group_name, self.storage)
                self.add_to_storage = True
            funcname = f"{group_name}::{tensor.shape}"