| 827 | backend(_backend) {} |
| 828 | |
| 829 | uint64_t PackAllReduceReplacePass::GroupInfo::hash(uint64_t extra) const { |
| 830 | DTypeEnum ev = dtype.enumv(); |
| 831 | const std::string& server_addr = group_client->get_addr(); |
| 832 | return XXHash() |
| 833 | .update(&extra, sizeof(uint64_t)) |
| 834 | .update(&device, sizeof(int)) |
| 835 | .update(&ev, sizeof(DTypeEnum)) |
| 836 | .update(&nr_devices, sizeof(size_t)) |
| 837 | .update(&is_root, sizeof(bool)) |
| 838 | .update(&rank, sizeof(int)) |
| 839 | .update(server_addr.c_str(), server_addr.size()) |
| 840 | .update(backend.c_str(), backend.size()) |
| 841 | .digest(); |
| 842 | } |
| 843 | |
| 844 | uint64_t PackAllReduceReplacePass::collect_groups( |
| 845 | OperatorNodeBase* opr, |