Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
add nightly tests, fix sanity
Browse files Browse the repository at this point in the history
  • Loading branch information
Wei Chu committed Apr 1, 2021
1 parent 16bbe1e commit 600e10c
Show file tree
Hide file tree
Showing 3 changed files with 22 additions and 4 deletions.
12 changes: 12 additions & 0 deletions ci/docker/runtime_functions.sh
Original file line number Diff line number Diff line change
Expand Up @@ -1594,6 +1594,18 @@ nightly_estimator() {
nosetests test_sentiment_rnn.py
}

nightly_onnx_operator_tests() {
set -ex
export PYTHONPATH=./python/
export MXNET_SUBGRAPH_VERBOSE=0
export DMLC_LOG_STACK_TRACE_DEPTH=10
COV_ARG="--cov=./ --cov-report=xml --cov-append"
pip3 install onnx==1.8.1
pytest $COV_ARG --verbose tests/python-pytest/onnx/test_operators.py
pip3 install onnx==1.7.0
pytest $COV_ARG --verbose tests/python-pytest/onnx/test_operators.py
}

nightly_onnx_cv_tests() {
set -ex
export PYTHONPATH=./python/
Expand Down
6 changes: 2 additions & 4 deletions python/mxnet/contrib/onnx/mx2onnx/export_onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,10 +89,8 @@ def convert_layer(node, **kwargs):
+ "Instructions to install - https://github.com/onnx/onnx")

op = str(node["op"])
opset_version = onnx_opset_version()
if 'opset_version' in kwargs:
opset_version = kwargs["opset_version"]
# fallback to older opset versions if op is not registered in current version
opset_version = kwargs.get("opset_version", onnx_opset_version())
# fallback to older opset versions if op is not registered in current version
for op_version in range(opset_version, 10, -1):
if op_version not in MXNetGraph.registry_ or op not in MXNetGraph.registry_[op_version]:
if opset_version == 11:
Expand Down
8 changes: 8 additions & 0 deletions tests/nightly/JenkinsfileForBinaries
Original file line number Diff line number Diff line change
Expand Up @@ -94,6 +94,14 @@ core_logic: {
}
}
},
'ONNX-operator: CPU': {
node(NODE_LINUX_CPU) {
ws('workspace/onnx-operator-test-cpu') {
utils.unpack_and_init('cpu_int64', mx_cmake_lib)
utils.docker_run('ubuntu_nightly_cpu', 'nightly_onnx_operator_tests', false)
}
}
},
'ONNX-CV: CPU': {
node(NODE_LINUX_CPU) {
ws('workspace/onnx-cv-test-cpu') {
Expand Down

0 comments on commit 600e10c

Please sign in to comment.