PKGBUILD


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

# Maintainer: piernov <piernov@piernov.org>

pkgname=python-tensorflow-model-optimization-git
pkgver=r327.c791831
pkgrel=1
pkgdesc="A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning."
url="https://github.com/tensorflow/model-optimization"
arch=('x86_64')
license=('APACHE')
conflicts=('python-tensorflow-model-optimization')
provides=('python-tensorflow-model-optimization')
depends=('python-tensorflow' 'python-numpy' 'python-six' 'python-scipy' 'python-mock' 'python-dm-tree')
makedepends=('python-setuptools' 'bazel')
source=("git+https://github.com/tensorflow/model-optimization.git")
sha256sums=('SKIP')

pkgver() {
	cd model-optimization
	printf "r%s.%s" "$(git rev-list --count HEAD)" "$(git rev-parse --short HEAD)"
}

build() {
	# Force OpenJDK-11 
	export JAVA_HOME=/usr/lib/jvm/java-11-openjdk 

	cd model-optimization
	bazel build --copt=-O3 --copt=-march=native :pip_pkg
	./bazel-bin/pip_pkg .
}

package() {
	cd model-optimization
	PIP_CONFIG_FILE=/dev/null pip install --isolated --root="$pkgdir" --ignore-installed --no-deps tf_model_optimization*.whl
}