BugFix

llvm/projects/hpvm-tensor-rt/bin/tuner_src/profiling.py

@@ -2,19 +2,18 @@
 import time
 
 profiled_ops = {}
-start = 0
 
 def startProfile(op_id):
   start = time.time()
+  return start
   
 
-def stopProfile(op_id):
+def stopProfile(op_id, start):
   end = time.time()
   total_time = end - start
  
   profiled_ops[op_id] = total_time
-  start = 0
-
+ 
   
 def dumpProfiles(file_name):
 

llvm/projects/hpvm-tensor-rt/bin/tuner_src/run_autotuner.py

@@ -57,20 +57,38 @@ def computeLayerSwings():
 def runPromiseTuner():
 
   #runPromiseBench(bench_tuner_data["vgg16_cifar100"])
-  startProfile("Lenet")
+  start = startProfile("LeNet")  
   runPromiseBench(bench_tuner_data["lenet_keras"])
-  stopProfile("Lenet")
-
-  dumpProfiles("time_profile.txt")
+  stopProfile("LeNet", start)
   
-  #runPromiseBench(bench_tuner_data["mobilenet_cifar10"])
-  #runPromiseBench(bench_tuner_data["alexnet_cifar10"])
-  #runPromiseBench(bench_tuner_data["vgg16_cifar10"])
+  start = startProfile("VGG16_100")  
+  runPromiseBench(bench_tuner_data["vgg16_cifar100"])
+  stopProfile("VGG16_100", start)
 
-  
-  #runPromiseBench("alexnet2_cifar10")
-  #runPromiseBench("resnet18_cifar10")
-  #runPromiseBench("mobilenet_shallow")
+  start = startProfile("VGG16_10")  
+  runPromiseBench(bench_tuner_data["vgg16_cifar10"])
+  stopProfile("VGG16_10", start)  
+
+  start = startProfile("Alexnet2")  
+  runPromiseBench(bench_tuner_data["alexnet2_cifar10"])
+  stopProfile("Alexnet2", start)  
+
+  start = startProfile("Alexnet")  
+  runPromiseBench(bench_tuner_data["alexnet_cifar10"])
+  stopProfile("Alexnet", start)
+
+  start = startProfile("ResNet")  
+  runPromiseBench(bench_tuner_data["resnet18_cifar10"])
+  stopProfile("ResNet", start)  
+
+  start = startProfile("MobileNet")  
+  runPromiseBench(bench_tuner_data["mobilenet_shallow"])
+  stopProfile("MobileNet", start)
+
+  start = startProfile("MobileNet-SH")  
+  runPromiseBench(bench_tuner_data["mobilenet_cifar10"])
+  stopProfile("MobileNet-SH", start)  
+   
  
 
   #runPSNRPromiseBench("pipeline_GEOM")
@@ -156,3 +174,6 @@ if __name__ == "__main__":
   #runPromiseValidation()
 
   #runSensAnalysis()
+
+    
+  dumpProfiles("time_profile.txt")