diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/profiling/alexnet2_profiling_tensors.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/profiling/alexnet2_profiling_tensors.cc
index e9db1ed227291f8aa60c7c919162db26dbd0ea98..f44ba60e2611daf53c6fe959b400d1e1b031804c 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/profiling/alexnet2_profiling_tensors.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/profiling/alexnet2_profiling_tensors.cc
@@ -13,20 +13,9 @@
 #include "../../../tensor_runtime/include/tensor_runtime.h"
 #include "../../include/utils.h"
 
-/*
-void add_to_map(std::unordered_map<std::string, std::pair<double, double> >& total_time_energies, double data, const std::string& key){
-    auto itr = total_time_energies.find(key);
-    if (itr == total_time_energies.end()){
-        total_time_energies.insert(std::make_pair(key, data));
-    } else {
-        itr->second += data;
-    }
-}
-*/
-
 void add_data(std::unordered_map<std::string, std::pair<double, double> >& total_time_energies, Profiler& profiler, const std::string& op_name){
     profiler.pause_profiler();
-    auto time_energy = profiler.get_time_energy_realzzzz();
+    auto time_energy = profiler.get_time_energy();
 
     auto itr = total_time_energies.find(op_name);
     if (itr == total_time_energies.end()){
@@ -228,9 +217,6 @@ void testCifarNet(){
       freeBatchMemory();
     }
   }
-  auto total = profiler.get_time_energy_realzzzz();
-  std::cout << "total = " << total.first << ", " << total.second << std::endl;
-  profiler.get();
   profiler.stop_profiler();
   
   stopProfiling();