Add caffe2 nnvm frontend to CI (#3018)

diff --git a/nnvm/tests/python/frontend/caffe2/test_forward.py b/nnvm/tests/python/frontend/caffe2/test_forward.py
index 9f3c5c9fa85a79801a6a9fa67752a61146e180dd..2a216314ba1a8951f1cf6ab1f95c88d2c2cf1152 100644 (file)
--- 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 1679595e712be9ae986b2fe973a5e695968fc314..37159dbc9a58078725d5df86ed032ef18f1b0519 100755 (executable)
--- 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