[Caffe2] Skip some tests to unbreak CI (pytorch#7804)

* Skip some tests to unbreak CI * Pass the opset_version to run_node * Remove the stale check_graph call, caffe2_net_to_onnx_model will invoke check_model
acwyzhou · May 24, 2018 · f9633b9 · f9633b9
1 parent fdabc02
commit f9633b9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/caffe2/python/onnx/backend.py b/caffe2/python/onnx/backend.py
@@ -207,7 +207,8 @@ def dummy_name(cls):
     # opset_version if you don't want this to version.
     @classmethod
     def run_node(cls, node, inputs, device='CPU', opset_version=_known_opset_version, outputs_info=None):
-        super(Caffe2Backend, cls).run_node(node, inputs, device=device, outputs_info=outputs_info)
+        super(Caffe2Backend, cls).run_node(node, inputs, device=device,
+                                           outputs_info=outputs_info, opset_version=opset_version)
 
         device_option = get_device_option(Device(device))
         ws = Workspace()

diff --git a/caffe2/python/onnx/frontend.py b/caffe2/python/onnx/frontend.py
@@ -265,7 +265,6 @@ def caffe2_net_to_onnx_graph(cls,
             for name in predict_net.external_output
             if name in all_output)
 
-        checker.check_graph(graph_def)
         return graph_def
 
     @classmethod

diff --git a/caffe2/python/onnx/tests/onnx_backend_test.py b/caffe2/python/onnx/tests/onnx_backend_test.py
@@ -35,6 +35,7 @@
                      '|test_simple_rnn.*'  # Seems simple RNN case has some problem
                      '|test_gru.*'  # Seems GRU case has some problem
                      '|test_upsample.*'  # Upsample is redesigned in ONNX
+                     '|test_prelu.*'  # PRelu is not compliant with ONNX yet
                      '|test_operator_repeat.*'  # Tile is not compliant with ONNX yet
                      '|test_.*pool_.*same.*'  # Does not support pool same.
                      ')')
@@ -43,6 +44,7 @@
 backend_test.exclude('(test_cast_.*'
                      '|test_Conv1d_.*cuda'
                      '|test_Conv3d_groups_cuda'
+                     '|test_rnn_seq_length'
                      '|test_operator_add.*_cuda'
                      '|test_operator_lstm_cuda'
                      '|test_operator_rnn.*_cuda)')