new package

2 files changed, 67 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..dbf134eed4f2
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,24 @@
+pkgbase = python-onnxoptimizer
+	pkgdesc = ONNX model optimizer
+	pkgver = 0.2.5
+	pkgrel = 1
+	url = https://github.com/onnx/optimizer
+	arch = x86_64
+	license = MIT
+	checkdepends = python-pytest
+	checkdepends = python-nbval
+	checkdepends = python-torchvision
+	checkdepends = python-onnxruntime
+	makedepends = python-setuptools
+	makedepends = cmake
+	makedepends = pybind11
+	makedepends = git
+	depends = python
+	depends = python-onnx
+	source = onnx-optimizer::git+https://github.com/onnx/optimizer.git#tag=v0.2.5
+	source = git+https://github.com/onnx/onnx.git
+	sha256sums = SKIP
+	sha256sums = SKIP
+
+pkgname = python-onnxoptimizer
+
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..b9ae7fe54e29
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,43 @@
+# Maintainer: Chih-Hsuan Yen <yan12125@gmail.com>
+
+pkgname=python-onnxoptimizer
+pkgver=0.2.5
+pkgdesc='ONNX model optimizer'
+pkgrel=1
+arch=(x86_64)
+url='https://github.com/onnx/optimizer'
+license=(MIT)
+depends=(python python-onnx)
+makedepends=(python-setuptools cmake pybind11 git)
+checkdepends=(python-pytest python-nbval python-torchvision python-onnxruntime)
+source=("onnx-optimizer::git+https://github.com/onnx/optimizer.git#tag=v$pkgver"
+        "git+https://github.com/onnx/onnx.git")
+sha256sums=('SKIP'
+            'SKIP')
+
+prepare() {
+  cd onnx-optimizer
+  sed -i '/pytest-runner/d' setup.py
+
+  git submodule init
+  git config submodule.third_party/onnx.url "$srcdir"/onnx
+  git submodule update third_party/onnx
+}
+
+build() {
+  cd onnx-optimizer
+  # https://github.com/onnx/optimizer/issues/38
+  CMAKE_ARGS="-DONNX_USE_LITE_PROTO=ON" python setup.py build
+}
+
+check() {
+  cd onnx-optimizer
+  pyver=$(python -c 'import sys; print(str(sys.version_info[0]) + "." + str(sys.version_info[1]))')
+  PYTHONPATH="$PWD/build/lib.linux-$CARCH-$pyver" pytest
+}
+
+package() {
+  cd onnx-optimizer
+  python setup.py install --root="$pkgdir" --skip-build --optimize=1
+  install -Dm644 LICENSE -t "$pkgdir"/usr/share/licenses/$pkgname
+}