From 3cd601dcd65ea3bd91bbf32dffd739a85028a2b8 Mon Sep 17 00:00:00 2001
From: hlu1 <14827759+hlu1@users.noreply.github.com>
Date: Fri, 12 Apr 2019 22:53:20 -0700
Subject: [PATCH] Add caffe2 nnvm frontend to CI

---
 nnvm/tests/python/frontend/caffe2/test_forward.py | 13 ++++++-------
 tests/scripts/task_python_frontend.sh             |  7 +++++--
 2 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/nnvm/tests/python/frontend/caffe2/test_forward.py b/nnvm/tests/python/frontend/caffe2/test_forward.py
index 9f3c5c9fa85a..2a216314ba1a 100644
--- a/nnvm/tests/python/frontend/caffe2/test_forward.py
+++ b/nnvm/tests/python/frontend/caffe2/test_forward.py
@@ -45,7 +45,6 @@ def get_tvm_output(model,
     graph, lib, params = nnvm.compiler.build(
         sym, target, shape=shape_dict, dtype=dtype_dict, params=params)
 
-    ctx = tvm.cpu(0)
     m = graph_runtime.create(graph, lib, ctx)
 
     # set inputs
@@ -89,21 +88,21 @@ def verify_caffe2_forward_impl(model, data_shape, out_shape):
         tvm.testing.assert_allclose(c2_out, tvm_out, rtol=1e-5, atol=1e-5)
 
 
-def verify_squeezenet1_1():
+def test_squeezenet1_1():
     verify_caffe2_forward_impl(c2_squeezenet, (1, 3, 224, 224),
                                (1, 1000, 1, 1))
 
 
-def verify_resnet50():
+def test_resnet50():
     verify_caffe2_forward_impl(c2_resnet50, (1, 3, 224, 224),
                                (1, 1000))
 
 
-def verify_vgg19():
+def test_vgg19():
     verify_caffe2_forward_impl(c2_vgg19, (1, 3, 224, 224), (1, 1000))
 
 
 if __name__ == '__main__':
-    verify_squeezenet1_1()
-    verify_resnet50()
-    verify_vgg19()
+    test_squeezenet1_1()
+    test_resnet50()
+    test_vgg19()
diff --git a/tests/scripts/task_python_frontend.sh b/tests/scripts/task_python_frontend.sh
index 1679595e712b..37159dbc9a58 100755
--- a/tests/scripts/task_python_frontend.sh
+++ b/tests/scripts/task_python_frontend.sh
@@ -6,9 +6,9 @@
 # to you under the Apache License, Version 2.0 (the
 # "License"); you may not use this file except in compliance
 # with the License.  You may obtain a copy of the License at
-# 
+#
 #   http://www.apache.org/licenses/LICENSE-2.0
-# 
+#
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
@@ -50,6 +50,9 @@ python3 -m nose -v nnvm/tests/python/frontend/tensorflow
 echo "Running nnvm CoreML frontend test..."
 python3 -m nose -v nnvm/tests/python/frontend/coreml
 
+echo "Running nnvm Caffe2 frontend test..."
+python3 -m nose -v nnvm/tests/python/frontend/caffe2
+
 echo "Running nnvm DarkNet frontend test..."
 python3 -m nose -v nnvm/tests/python/frontend/darknet || exit -1