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 = {