| 37 | } |
| 38 | |
| 39 | TaskStreamIndexManager::stream_index_t TaskStreamIndexManager::GetTaskStreamIndex( |
| 40 | TaskType task_type, const DeviceId& device_id) { |
| 41 | auto* generator = GetGenerator(device_id); |
| 42 | auto stream_index = CHECK_JUST(TaskStreamIndexGetterRegistry::Instance().Dispatch( |
| 43 | device_id.device_type(), task_type, generator)); |
| 44 | return stream_index; |
| 45 | } |
| 46 | |
| 47 | TaskStreamIndexManager::stream_index_t TaskStreamIndexManager::GetComputeTaskStreamIndex( |
| 48 | const DeviceId& device_id) { |
no test coverage detected