create

4 files changed, 81 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..2ed5348abca6
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,29 @@
+pkgbase = microsoft-nni-git
+	pkgdesc = An open source toolkit for neural architecture search, hyper-parameter optimization and optimizer design
+	pkgver = 0.1.0.r28.f1c2b70
+	pkgrel = 1
+	url = https://github.com/microsoft/nni
+	arch = x86_64
+	license = MIT
+	makedepends = git
+	makedepends = yarn
+	makedepends = typescript
+	makedepends = python-setuptools
+	depends = nodejs-serve
+	depends = python-astor
+	depends = python-json-tricks
+	depends = python-psutil
+	depends = python-yaml
+	depends = python-requests
+	depends = python-scipy
+	provides = microsoft-nni
+	conflicts = microsoft-nni
+	source = git://github.com/microsoft/nni.git
+	source = nnimanager
+	source = nnictl
+	sha1sums = SKIP
+	sha1sums = SKIP
+	sha1sums = SKIP
+
+pkgname = microsoft-nni-git
+
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..2bbbc6458f5c
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,48 @@
+# Maintainer:  Liu Zhe <cruise.pas@gmail.com>
+
+pkgname=microsoft-nni-git
+pkgver=0.1.0.r28.f1c2b70
+pkgrel=1
+pkgdesc='An open source toolkit for neural architecture search, hyper-parameter optimization and optimizer design'
+arch=('x86_64')
+url='https://github.com/microsoft/nni'
+license=('MIT')
+depends=('nodejs-serve' 'python-astor' 'python-json-tricks' 'python-psutil' 'python-yaml' 'python-requests' 'python-scipy')
+makedepends=('git' 'yarn' 'typescript' 'python-setuptools')
+provides=('microsoft-nni')
+conflicts=('microsoft-nni')
+source=('git://github.com/microsoft/nni.git'
+        'nnimanager' 'nnictl')
+sha1sums=('SKIP' 'SKIP' 'SKIP')
+
+pkgver() {
+    cd "$srcdir"/nni
+    echo 0.1.0.r$(git rev-list --count master).$(git rev-parse --short master)
+}
+
+build() {
+    cd "$srcdir"/nni
+    make
+}
+
+package() {
+    mkdir -p "$pkgdir"/usr/bin
+    mkdir -p "$pkgdir"/usr/lib/node_modules/nni
+    mkdir -p "$pkgdir"/usr/doc/nni
+
+    cd "$srcdir"/nni
+
+    cp -rT src/nni_manager/dist "$pkgdir"/usr/lib/node_modules/nni/nni_manager
+    cp -rT src/nni_manager/node_modules "$pkgdir"/usr/lib/node_modules/nni/nni_manager/node_modules
+    cp -rT src/webui/build "$pkgdir"/usr/lib/node_modules/nni/webui
+
+    ( cd src/sdk/pynni && python setup.py install --skip-build -O1 --root="$pkgdir" )
+    ( cd tools && python setup.py install --skip-build -O1 --root="$pkgdir" )
+
+    cp -rT examples "$pkgdir"/usr/doc/nni/examples
+
+    install -Dm644 LICENSE "$pkgdir"/usr/share/licenses/$pkgname/LICENSE
+
+    install -Dm755 "$srcdir"/nnimanager "$pkgdir"/usr/bin/nnimanager
+    install -Dm755 "$srcdir"/nnictl "$pkgdir"/usr/bin/nnictl
+}
diff --git a/nnictl b/nnictl
new file mode 100755
index 000000000000..5c5e36641521
--- /dev/null
+++ b/nnictl
@@ -0,0 +1,2 @@
+#!/bin/sh
+WEB_UI_FOLDER=/usr/lib/node_modules/nni/webui python -m nnicmd.nnictl $@
diff --git a/nnimanager b/nnimanager
new file mode 100755
index 000000000000..9c2696a4ae86
--- /dev/null
+++ b/nnimanager
@@ -0,0 +1,2 @@
+#!/bin/sh
+cd /usr/lib/node_modules/nni/nni_manager && node main.js $@