update to 1.5.0

2 files changed, 13 insertions, 10 deletions

diff --git a/.SRCINFO b/.SRCINFO
index bdfdc6261654..d306c5ee0595 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,6 +1,6 @@
 pkgbase = python-onnx
 	pkgdesc = Open Neural Network Exchange
-	pkgver = 1.4.1
+	pkgver = 1.5.0
 	pkgrel = 1
 	url = https://onnx.ai
 	arch = x86_64
@@ -15,8 +15,8 @@ pkgbase = python-onnx
 	depends = pybind11
 	depends = python-numpy
 	depends = python-six
-	source = onnx-1.4.1.tar.gz::https://github.com/onnx/onnx/archive/v1.4.1.tar.gz
-	sha512sums = 35d189b32e553bc306c453bc3d94bc8ebe2f13202c4c0a8de353e2dba60d15033629e699c5ef38f6c366df6080606b9f79a4f11ab7dcef31d865ce0df957ca92
+	source = onnx-1.5.0.tar.gz::https://github.com/onnx/onnx/archive/v1.5.0.tar.gz
+	sha512sums = 0dd8ac3210331352f0c97383a36c7c9943f1885491ccb3e2c54ebcfb472ff41b192fe701ba82a1a33251158d9b99ff626b429e33b48921151302de8d0e81b82e
 
 pkgname = python-onnx
 
diff --git a/PKGBUILD b/PKGBUILD
index d3cd940f5242..e20413254256 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,7 +1,8 @@
 # Maintainer: Butui Hu <hot123tea123@gmail.com>
 
 pkgname=python-onnx
-pkgver=1.4.1
+_pkgname=onnx
+pkgver=1.5.0
 pkgrel=1
 pkgdesc='Open Neural Network Exchange'
 arch=('x86_64')
@@ -10,11 +11,11 @@ license=('MIT')
 depends=(protobuf pybind11 python-numpy python-six)
 makedepends=(python-setuptools pybind11 cmake)
 checkdepends=(python-nose python-pytest python-scipy)
-source=("onnx-$pkgver.tar.gz"::"https://github.com/onnx/onnx/archive/v$pkgver.tar.gz")
-sha512sums=('35d189b32e553bc306c453bc3d94bc8ebe2f13202c4c0a8de353e2dba60d15033629e699c5ef38f6c366df6080606b9f79a4f11ab7dcef31d865ce0df957ca92')
+source=("${_pkgname}-${pkgver}.tar.gz::https://github.com/onnx/onnx/archive/v${pkgver}.tar.gz")
+sha512sums=('0dd8ac3210331352f0c97383a36c7c9943f1885491ccb3e2c54ebcfb472ff41b192fe701ba82a1a33251158d9b99ff626b429e33b48921151302de8d0e81b82e')
 
 build() {
-  cd onnx-$pkgver
+  cd "${_pkgname}-${pkgver}"
   python setup.py build
 }
 
@@ -24,7 +25,9 @@ build() {
 #}
 
 package() {
-  cd onnx-$pkgver
-  python setup.py install --root=$pkgdir --optimize=1 --skip-build
-  install -Dm644 LICENSE -t "$pkgdir"/usr/share/licenses/$pkgname
+  cd "${_pkgname}-${pkgver}"
+  export CC=/opt/cuda/bin/gcc
+  export CXX=/opt/cuda/bin/g++
+  python setup.py install --root="${pkgdir}" --optimize=1 --skip-build
+  install -Dm644 LICENSE -t "${pkgdir}/usr/share/licenses/${pkgname}"
 }