From 5a602cf0a28bf3a9b8f9a966730a51d599961559 Mon Sep 17 00:00:00 2001
From: "haozhe.yang" <haozhe.yang@sophgo.com>
Date: Sun, 3 Nov 2024 23:03:51 +0800
Subject: [PATCH] fix arcface performance

Change-Id: Iddefa969142b4fbf769a524020beb11b19977996
---
 lib/Dialect/Tpu/Transforms/LayerGroup/CycleCalculator.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/Dialect/Tpu/Transforms/LayerGroup/CycleCalculator.cpp b/lib/Dialect/Tpu/Transforms/LayerGroup/CycleCalculator.cpp
index 5db057d67..626ea440b 100644
--- 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;