| 229 | } |
| 230 | |
| 231 | Maybe<Scope> InstructionsBuilder::BuildInitialScope( |
| 232 | int64_t session_id, const JobConfigProto& job_conf, const std::string& device_tag, |
| 233 | const std::vector<std::string>& machine_device_ids, const std::shared_ptr<Shape>& hierarchy, |
| 234 | bool is_local) { |
| 235 | ScopeProto scope_proto; |
| 236 | scope_proto.set_session_id(session_id); |
| 237 | std::shared_ptr<JobDesc> job_conf_sym = JUST(GetJobConfSymbol(job_conf)); |
| 238 | scope_proto.set_job_desc_symbol_id(JUST(job_conf_sym->symbol_id())); |
| 239 | std::shared_ptr<ParallelConf> parallel_conf = |
| 240 | JUST(MakeParallelConf(device_tag, machine_device_ids, hierarchy)); |
| 241 | std::shared_ptr<ParallelDesc> device_parallel_desc_sym = |
| 242 | JUST(GetParallelDescSymbol(*parallel_conf)); |
| 243 | scope_proto.set_device_parallel_desc_symbol_id(JUST(device_parallel_desc_sym->symbol_id())); |
| 244 | parallel_conf = JUST(MakeParallelConf("cpu", machine_device_ids, hierarchy)); |
| 245 | std::shared_ptr<ParallelDesc> host_parallel_desc_sym = |
| 246 | JUST(GetParallelDescSymbol(*parallel_conf)); |
| 247 | scope_proto.set_host_parallel_desc_symbol_id(JUST(host_parallel_desc_sym->symbol_id())); |
| 248 | if (is_local) { |
| 249 | scope_proto.mutable_opt_local_parallel_conf()->mutable_local_parallel(); |
| 250 | } else { |
| 251 | scope_proto.mutable_opt_local_parallel_conf()->clear_local_parallel(); |
| 252 | } |
| 253 | return GetScopeSymbol(scope_proto); |
| 254 | } |
| 255 | |
| 256 | Maybe<Scope> InstructionsBuilder::BuildInitialScopeWithPlacement(int64_t session_id, |
| 257 | const JobConfigProto& job_conf, |
no test coverage detected