From 8b445e9e34a33cfab08d9bc21ff31c07eece4889 Mon Sep 17 00:00:00 2001
From: zhouzj <41366441+zzjjay@users.noreply.github.com>
Date: Tue, 27 Sep 2022 09:49:47 +0800
Subject: [PATCH] Fix tests of latencyPredictor. (#1433)

---
 paddleslim/analysis/_utils.py            | 6 +++++-
 paddleslim/analysis/latency_predictor.py | 2 +-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/paddleslim/analysis/_utils.py b/paddleslim/analysis/_utils.py
index 82bedee5..b8226539 100644
--- a/paddleslim/analysis/_utils.py
+++ b/paddleslim/analysis/_utils.py
@@ -91,6 +91,9 @@ def opt_model(opt="paddle_lite_opt",
         model_out = os.path.join(model_out, 'model')
     else:
         model_out = model_out + '.nb'
+    assert os.path.exists(
+        model_out
+    ), 'There is an error during \'opt\' conversion model, please check the above error message.'
     return model_out
 
 
@@ -133,7 +136,8 @@ def save_cls_model(model, input_shape, save_dir, data_type):
             batch_size=input_shape[0],
             batch_nums=1,
             weight_bits=8,
-            activation_bits=8)
+            activation_bits=8,
+            quantizable_op_type=["conv2d", "depthwise_conv2d"])
 
         model_file = os.path.join(quantize_model_path, 'model.pdmodel')
         param_file = os.path.join(quantize_model_path, 'model.pdiparams')
diff --git a/paddleslim/analysis/latency_predictor.py b/paddleslim/analysis/latency_predictor.py
index f55b413a..8d4a716a 100644
--- a/paddleslim/analysis/latency_predictor.py
+++ b/paddleslim/analysis/latency_predictor.py
@@ -13,7 +13,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
+import time
 import os
 import pip
 import platform
-- 
GitLab