| 241 | } |
| 242 | |
| 243 | Status RingAlg::InitializeCollectiveContext(CollectiveContext* col_ctx) { |
| 244 | DCHECK(col_ctx->dev_mgr); |
| 245 | col_ctx_ = col_ctx; |
| 246 | col_params_ = &col_ctx->col_params; |
| 247 | return collective_util::InitializeDeviceAndLocality( |
| 248 | col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, |
| 249 | &col_ctx->device_locality); |
| 250 | } |
| 251 | |
| 252 | string RingAlg::TensorDebugString(const Tensor& tensor) { |
| 253 | const DeviceBase::GpuDeviceInfo* gpu_device_info = |