You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@singa.apache.org by ch...@apache.org on 2020/06/18 09:57:46 UTC

[singa] branch dev updated: fix onnx backend cuda cases

This is an automated email from the ASF dual-hosted git repository.

chrishkchris pushed a commit to branch dev
in repository https://gitbox.apache.org/repos/asf/singa.git


The following commit(s) were added to refs/heads/dev by this push:
     new 4e6f9e5  fix onnx backend cuda cases
     new 9dd09fa  Merge pull request #742 from joddiy/fix-cuda-cases
4e6f9e5 is described below

commit 4e6f9e5f6d8bdbc3d042c0bb12bfa9153309bf20
Author: joddiy <jo...@qq.com>
AuthorDate: Thu Jun 18 17:42:54 2020 +0800

    fix onnx backend cuda cases
---
 test/python/test_onnx_backend.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/test/python/test_onnx_backend.py b/test/python/test_onnx_backend.py
index debfd85..a2a2b43 100644
--- a/test/python/test_onnx_backend.py
+++ b/test/python/test_onnx_backend.py
@@ -100,6 +100,10 @@ for name in sonnx.SingaBackend._rename_operators.keys():
 for pattern in _exclude_nodes_patterns:
     backend_test.exclude(pattern)
 
+# exclude the cuda cases
+if not singa.USE_CUDA:
+    backend_test.exclude(r'(cuda)')
+
 # import all test cases at global scope to make them visible to python.unittest
 # print(backend_test.enable_report().test_cases)
 test_cases = {