Set schedule when parsing a single instruction as a module.

PiperOrigin-RevId: 285867393
Change-Id: Ia9bbb5933a157c053cd26357160bdb729b7b5f90

tensorflow/compiler/xla/service/hlo_parser.cc

@@ -4386,6 +4386,7 @@ bool HloParserImpl::ParseSingleInstruction(HloModule* module) {
   for (auto& comp : computations_) {
     module->AddEmbeddedComputation(std::move(comp));
   }
+  TF_CHECK_OK(module->set_schedule(ScheduleFromInstructionOrder(module)));
   return true;
 }