forked from mindspore-Ecosystem/mindspore
!32136 [MD][AT] Autotune config num wrks fix
Merge pull request !32136 from harshvardhangupta/na_at_with_0_wrks
This commit is contained in:
commit
e7fe64e304
|
@ -153,8 +153,8 @@ Status AutoTune::SummarizeTreeConfiguration(std::vector<std::string> *out) {
|
|||
if (!op->inlined() && op->Name() != "DeviceQueueOp") {
|
||||
std::stringstream s;
|
||||
s << std::left << std::setw(op_name_width) << op->NameWithID() << "(num_parallel_workers:" << std::right
|
||||
<< std::setw(val_width) << op->NumWorkers() << ", prefetch_size:" << std::setw(val_width)
|
||||
<< op->ConnectorCapacity() << ")";
|
||||
<< std::setw(val_width) << (op->NumWorkers() == 0 ? "NA" : std::to_string(op->NumWorkers()))
|
||||
<< ", prefetch_size:" << std::setw(val_width) << op->ConnectorCapacity() << ")";
|
||||
(void)out->emplace_back(s.str());
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue