Skip to content

Commit

Permalink
Revert "consider 1690 multi-core cycle calc"
Browse files Browse the repository at this point in the history
This reverts commit be3c8c6.

Change-Id: I9a5f1651597a46429b409d5a05b089ada15e799c
  • Loading branch information
charlesxzb committed May 29, 2024
1 parent 5bb7a49 commit a085169
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions lib/Dialect/Tpu/Transforms/LayerGroup/CycleCalculator.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -99,10 +99,9 @@ int64_t CycleCalculator::getGroupCycle(BasicTimeStepPtr &time_step,
std::vector<layer_cycle_info_t> layer_cycle;
std::vector<gdma_cycle_info_t> gdma_cycle;
bool enable_multi_core = false;
auto num_core = module::getCoreNum();
if (num_core > 1 && (loop_num / shape_secs.wsecs) > 1) {
if (module::getCoreNum() == 2 && loop_num > 1) {
enable_multi_core = true;
loop_num = loop_num / num_core + loop_num % num_core;
loop_num = loop_num / 2 + loop_num % 2;
}

int64_t filling_cycle = 0, kernel_cycle = 0, draining_cycle = 0;
Expand Down

0 comments on commit a085169

Please sign in to comment.