| 440 | } |
| 441 | |
| 442 | bool Member::MergeGpuStreamIdx(const Member& other) { |
| 443 | if (gpu_stream_idx_ == -1) { |
| 444 | gpu_stream_idx_ = other.gpu_stream_idx_; |
| 445 | } else if (other.gpu_stream_idx_ != -1 && |
| 446 | other.gpu_stream_idx_ != gpu_stream_idx_) { |
| 447 | return false; |
| 448 | } |
| 449 | |
| 450 | return true; |
| 451 | } |
| 452 | |
| 453 | void Member::AssignGpuStreamIdx(Node *node) { |
| 454 | if (gpu_stream_idx_ != -1) |