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