(group_id)
| 1026 | |
| 1027 | |
| 1028 | def _compute_group_hash(group_id): |
| 1029 | hashed_id = hashlib.sha256(group_id.encode("utf-8")).hexdigest() |
| 1030 | # first 16 characters for a reasonably short but unique name |
| 1031 | return hashed_id[:16] |
| 1032 | |
| 1033 | |
| 1034 | def _maybe_remove_and_reapply_group_offloading(module: torch.nn.Module) -> None: |
searching dependent graphs…