|
|
|
@ -26,6 +26,7 @@ namespace hybrid {
|
|
|
|
|
namespace {
|
|
|
|
|
const int kDataOutputIndex = 0;
|
|
|
|
|
const size_t kMinimumPiplineStages = 2;
|
|
|
|
|
const int kDefaultLoopCount = 10;
|
|
|
|
|
}
|
|
|
|
|
HybridModelAsyncExecutor::HybridModelAsyncExecutor(HybridModel *model)
|
|
|
|
|
: model_(model), run_flag_(false) {
|
|
|
|
@ -150,7 +151,7 @@ Status HybridModelAsyncExecutor::RunInternal() {
|
|
|
|
|
GELOGI("HybridModel will execute in pipeline mode");
|
|
|
|
|
auto iter_per_run = std::getenv("ITER_NUM");
|
|
|
|
|
if (iter_per_run) {
|
|
|
|
|
args.num_loops = static_cast<int>(strtol(iter_per_run, nullptr, 10));
|
|
|
|
|
args.num_loops = static_cast<int>(strtol(iter_per_run, nullptr, kDefaultLoopCount));
|
|
|
|
|
}
|
|
|
|
|
ret = pipe_executor_->Execute(args);
|
|
|
|
|
} else {
|
|
|
|
|