TVM 0.7.0 dev1

sjlee25 · Aug 2, 2020 · 1679f8e · 1679f8e
1 parent 35ab4cf
commit 1679f8e
Show file tree

Hide file tree

Showing 9 changed files with 3 additions and 3 deletions.
diff --git a/Inference.py b/Inference.py
@@ -91,7 +91,7 @@ def pushResult(self):
         global result
         string = '%4s %d = %7.2f ms (%6.2f ms)' % (self.dev_type.upper(), self.idx, \
             self.result_time.exec_time + self.result_time.io_time, self.result_time.io_time)
-        if self.predict_time > 0:
+        if self.predict_time is not None:
             string += ' | %7.2f ms' % (self.predict_time)
         string += ' [%3d]\n' % (self.batch_size)
         result += string

diff --git a/Partition.py b/Partition.py
@@ -127,12 +127,12 @@ def estimateDevTime(self, dev, batch_size):
         yp_exec = [dev_perf[xp[0]].exec_time, dev_perf[xp[1]].exec_time]
 
         intp_time = np.interp(batch_size, xp, yp_exec) * batch_size
+        intp_io_time = 0.
         if self.with_io_time:
             yp_io = [dev_perf[xp[0]].io_time, dev_perf[xp[1]].io_time]
             intp_io_time = np.interp(batch_size, xp, yp_io) * batch_size
-            intp_time += intp_io_time
 
-        return intp_time
+        return intp_time + intp_io_time
 
     def getMaxDiffDev(self):
         ret_dev = None

diff --git a/table_densenet-121 b/table_densenet-121
diff --git a/table_inception_v3 b/table_inception_v3
diff --git a/table_resnet-18 b/table_resnet-18
diff --git a/table_resnet-34 b/table_resnet-34
diff --git a/table_resnet-50 b/table_resnet-50
diff --git a/table_squeezenet_v1.0 b/table_squeezenet_v1.0
diff --git a/table_squeezenet_v1.1 b/table_squeezenet_v1.1