refactor: implement suggestions by dreieck

1 files changed, 17 insertions, 4 deletions

diff --git a/PKGBUILD b/PKGBUILD
index dd91daff424b..d57418052b58 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -3,12 +3,12 @@
 
 pkgname="llm"
 pkgver=0.1.1
-pkgrel=4
+pkgrel=5
 epoch=
 pkgdesc="Run inference for Large Language Models on CPU, with Rust 🦀🚀🦙"
 arch=(any)
 url="https://github.com/rustformers/${pkgname}"
-license=('MIT' 'APACHE')
+license=('Apache-2.0 OR MIT')
 depends=(glibc gcc-libs)
 makedepends=(git cargo)
 provides=(llm)
@@ -17,6 +17,7 @@ source=("git+${url}#tag=v${pkgver}"
         "git+https://github.com/ggerganov/ggml"
         )
 sha256sums=('SKIP' 'SKIP')
+options+=('!lto')
 
 
 prepare() {
@@ -24,12 +25,24 @@ prepare() {
     git submodule init
     git config submodule.crates/ggml/sys/ggml.url "${srcdir}/ggml"
     git -c protocol.file.allow=always submodule update
+
+    cat LICENSE-* > LICENSE
+
+    export RUSTUP_TOOLCHAIN=stable
+    cargo fetch --locked --target "$(rustc -vV | sed -n 's/host: //p')"
 }
 
 build() {
     cd "${srcdir}/${pkgname}"
-    cat LICENSE-* > LICENSE
-    cargo build --release
+    export RUSTUP_TOOLCHAIN=stable
+    export CARGO_TARGET_DIR=target
+    cargo build --frozen --release
+}
+
+check() {
+    cd "${srcdir}/${pkgname}"
+    export RUSTUP_TOOLCHAIN=stable
+    cargo test --frozen --all-features
 }
 
 package() {