fix bugs in unittest
diff --git a/.github/workflows/macOS.yaml b/.github/workflows/macOS.yaml
index 79e1d2a..795beb6 100644
--- a/.github/workflows/macOS.yaml
+++ b/.github/workflows/macOS.yaml
@@ -31,15 +31,17 @@
with:
python-version: "3.7"
- name: install-build-dependencies
- run: brew install protobuf swig opencv glog lmdb numpy && pip3 install numpy
+ run: brew install protobuf swig opencv glog lmdb numpy && pip3 install numpy \
+ && wget https://github.com/oneapi-src/oneDNN/releases/download/v1.1.3/dnnl_mac_1.1.3_cpu_iomp.tgz -P /tmp && tar zxf /tmp/dnnl_mac_1.1.3_cpu_iomp.tgz -C .
- name: configure
- run: mkdir build && cd build && cmake -DUSE_PYTHON3=YES -DENABLE_TEST=YES ..
+ run: mkdir build && cd build && cmake -DUSE_PYTHON3=YES -DENABLE_TEST=YES -DUSE_DNNL=YES ..
env:
CMAKE_INCLUDE_PATH: /usr/local/opt/openblas/include:$CMAKE_INCLUDE_PATH
CMAKE_LIBRARY_PATH: /usr/local/opt/openblas/lib:$CMAKE_LIBRARY_PATH
+ DNNL_ROOT: `pwd`/dnnl_mac_1.1.3_cpu_iomp
- name: build
run: cd build && make
env:
CXXFLAGS: -I /Users/runner/hostedtoolcache/Python/3.7.8/x64/lib/python3.7/site-packages/numpy/core/include $CXXFLAGS
- name: C++ test
- run: build/bin/test_singa
+ run: build/bin/test_singa
\ No newline at end of file
diff --git a/python/singa/sonnx.py b/python/singa/sonnx.py
index ce4cdc8..fb8197f 100755
--- a/python/singa/sonnx.py
+++ b/python/singa/sonnx.py
@@ -20,7 +20,9 @@
from __future__ import division
import numpy as np
+
import onnx
+import onnx.utils
from onnx.backend.base import Backend, BackendRep
from onnx import (checker, helper, numpy_helper, GraphProto, NodeProto,
TensorProto, OperatorSetIdProto, optimizer, mapping,
