initial commit

3 files changed, 169 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..a790bfd8a8ac
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,45 @@
+pkgbase = ctranslate2
+	pkgdesc = A C++ and Python library for efficient inference with Transformer models.
+	pkgver = 3.23.0
+	pkgrel = 1
+	url = https://opennmt.net/CTranslate2
+	arch = x86_64
+	license = MIT
+	makedepends = cmake
+	makedepends = cuda
+	makedepends = git
+	makedepends = intel-oneapi-mkl
+	makedepends = onednn
+	makedepends = pybind11
+	makedepends = python-build
+	makedepends = python-installer
+	makedepends = python-setuptools
+	makedepends = python-wheel
+	source = git+https://github.com/OpenNMT/CTranslate2.git#commit=83caf67972751ac9ce41b4fe17e8bf1a42b95307
+	source = git+https://github.com/jarro2783/cxxopts.git
+	source = git+https://github.com/NVIDIA/thrust.git
+	source = git+https://github.com/google/googletest.git
+	source = git+https://github.com/google/cpu_features.git
+	source = git+https://github.com/gabime/spdlog.git
+	source = git+https://github.com/google/ruy.git
+	source = git+https://github.com/pytorch/cpuinfo.git
+	sha256sums = SKIP
+	sha256sums = SKIP
+	sha256sums = SKIP
+	sha256sums = SKIP
+	sha256sums = SKIP
+	sha256sums = SKIP
+	sha256sums = SKIP
+	sha256sums = SKIP
+
+pkgname = ctranslate2
+	pkgdesc = A C++ library for efficient inference with Transformer models.
+	depends = onednn
+	provides = libctranslate2.so=3
+
+pkgname = python-ctranslate2
+	pkgdesc = A Python library for efficient inference with Transformer models.
+	depends = ctranslate2
+	depends = python-numpy
+	depends = python-setuptools
+	depends = python-yaml
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 000000000000..4dab8d6386e3
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,7 @@
+# Ignore everything
+*
+
+# But not these files...
+!.gitignore
+!PKGBUILD
+!.SRCINFO
diff --git a/PKGBUILD b/PKGBUILD
new file mode 100644
index 000000000000..5c67cd81f277
--- /dev/null
+++ b/PKGBUILD
@@ -0,0 +1,117 @@
+# Maintainer: Mark Wagie <mark dot wagie at proton dot me>
+pkgname=('ctranslate2' 'python-ctranslate2')
+pkgbase=ctranslate2
+pkgver=3.23.0
+pkgrel=1
+pkgdesc="A C++ and Python library for efficient inference with Transformer models."
+arch=('x86_64')
+url="https://opennmt.net/CTranslate2"
+license=('MIT')
+makedepends=(
+  'cmake'
+  'cuda'
+#  'cudnn'
+  'git'
+  'intel-oneapi-mkl'
+  'onednn'
+#  'openblas'
+  'pybind11'
+  'python-build'
+  'python-installer'
+  'python-setuptools'
+  'python-wheel'
+)
+#checkdepends=(
+#  'python-opennmt-tf'  # TODO
+#  'python-numpy'
+#  'python-pytest'
+#  'python-yaml'
+#)
+_commit=83caf67972751ac9ce41b4fe17e8bf1a42b95307  # tags/3.23.0^0
+source=("git+https://github.com/OpenNMT/CTranslate2.git#commit=$_commit"
+        'git+https://github.com/jarro2783/cxxopts.git'
+        'git+https://github.com/NVIDIA/thrust.git'
+        'git+https://github.com/google/googletest.git'
+        'git+https://github.com/google/cpu_features.git'
+        'git+https://github.com/gabime/spdlog.git'
+        'git+https://github.com/google/ruy.git'
+        'git+https://github.com/pytorch/cpuinfo.git')
+sha256sums=('SKIP'
+            'SKIP'
+            'SKIP'
+            'SKIP'
+            'SKIP'
+            'SKIP'
+            'SKIP'
+            'SKIP')
+
+pkgver() {
+  cd CTranslate2
+  git describe --tags | sed 's/^v//;s/-/+/g'
+}
+
+prepare() {
+  cd CTranslate2
+  git submodule init
+  for submodule in cxxopts thrust googletest cpu_features spdlog ruy; do
+    git config submodule.src/third_party/"$submodule".url "$srcdir/$submodule"
+  done
+  git -c protocol.file.allow=always submodule update
+
+  cd third_party/ruy
+  git submodule init
+  git config submodule.src/third_party/cpuinfo.url "$srcdir/cpuinfo"
+  git config submodule.src/third_party/googletest.url "$srcdir/googletest"
+  git -c protocol.file.allow=always submodule update
+}
+
+build() {
+
+  ## WITH_CUDA
+  ## CUDA_DYNAMIC_LOADING
+  ## WITH_CUDNN
+  # v12 not supported, fails to build
+  # https://github.com/OpenNMT/CTranslate2/issues/1250
+
+  ## WITH_OPENBLAS
+  # CMake Error at CMakeLists.txt:396 (message):
+  # OpenBLAS include directory not found
+
+  cmake -B build -S CTranslate2 \
+    -DCMAKE_BUILD_TYPE='Release' \
+    -DCMAKE_INSTALL_PREFIX='/usr' \
+    -DOPENMP_RUNTIME='COMP' \
+    -DWITH_MKL='ON' \
+    -DWITH_DNNL='ON' \
+    -DWITH_RUY='ON' \
+    -Wno-dev
+  cmake --build build
+
+  cd CTranslate2/python
+  CTRANSLATE2_ROOT=.. LIBRARY_PATH="$srcdir/build" python -m build --wheel --no-isolation
+}
+
+#check() {
+#  cd CTranslate2
+#  PYTHONPATH=python pytest python/tests/
+#}
+
+package_ctranslate2() {
+  pkgdesc="A C++ library for efficient inference with Transformer models."
+  depends=('onednn')
+  provides=('libctranslate2.so=3')
+
+  DESTDIR="$pkgdir" cmake --install build
+
+  install -Dm644 CTranslate2/LICENSE -t "$pkgdir/usr/share/licenses/$pkgname/"
+}
+
+package_python-ctranslate2() {
+  pkgdesc="A Python library for efficient inference with Transformer models."
+  depends=('ctranslate2' 'python-numpy' 'python-setuptools' 'python-yaml')
+
+  cd CTranslate2/python
+  python -m installer --destdir="$pkgdir" dist/*.whl
+
+  install -Dm644 ../LICENSE -t "$pkgdir/usr/share/licenses/$pkgname/"
+}