[Frontend] Make onnx dependency optional (#33)

make onnx dependency optional

python/hidet/graph/frontend/onnx/__init__.py

@@ -0,0 +1,4 @@
+from .availability import available
+
+if available():
+    from .onnx import from_onnx

python/hidet/graph/frontend/onnx/availability.py

@@ -0,0 +1,19 @@
+try:
+    import onnx  # pylint: disable=unused-import
+except ImportError:
+    onnx = None
+    _available = False
+else:
+    _available = True
+
+
+def available():
+    """
+    Check if onnx is installed.
+
+    Returns
+    -------
+    ret: bool
+        True if torch is installed.
+    """
+    return _available

requirements-dev.txt

@@ -13,4 +13,5 @@ torchvision
 transformers
 
 # check the correctness with onnxruntime
+onnx
 onnxruntime-gpu

requirements.txt

@@ -20,9 +20,6 @@ tqdm
 # in Nsight System.
 nvtx
 
-# for onnx frontend
-onnx
-
 # for print python ast node
 astunparse
 

scripts/build_wheel.sh

@@ -1,5 +1,7 @@
 #!/bin/bash
 
+set -e  # exit immediately if a command exits with a non-zero status.
+
 ###############################################################################
 # This script builds a wheel for the current platform and Python version.
 ###############################################################################

setup.py

@@ -19,7 +19,6 @@ def is_pure(self):
     package_dir={"": "python"},
     include_package_data=True,
     install_requires=[
-        "onnx",
         "numpy",
         "psutil",
         "tqdm",