fix arcface performance

Change-Id: Iddefa969142b4fbf769a524020beb11b19977996
sophgo · Nov 3, 2024 · 5a602cf · 5a602cf
1 parent 6ce150c
commit 5a602cf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/Dialect/Tpu/Transforms/LayerGroup/CycleCalculator.cpp b/lib/Dialect/Tpu/Transforms/LayerGroup/CycleCalculator.cpp
@@ -323,6 +323,7 @@ int64_t CycleCalculator::getGroupCycle(BasicTimeStepPtr &time_step,
   });
   }
 
+  DEBUG_WITH_TYPE("mc_lg_refactor", {
     if(num_core == 8){
       loop_num = loop_num / num_core + (loop_num % num_core > 0);
           DEBUG_WITH_TYPE("cycle_calc", {
@@ -331,7 +332,7 @@ int64_t CycleCalculator::getGroupCycle(BasicTimeStepPtr &time_step,
           << "; loop_num = " << loop_num << "\n";
       });
     }
-
+  });
 
   int64_t filling_cycle = 0, kernel_cycle = 0, draining_cycle = 0;
   int64_t total_layer_cycle = 0, total_gdma_cycle = 0;