Add options & unit tests for cibuildwheel (#162)

li-plus · Oct 30, 2023 · 972b0de · 972b0de
1 parent 2552da6
commit 972b0de
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 54 deletions.
diff --git a/.github/workflows/python_compile_wheel.yml b/.github/workflows/python_compile_wheel.yml
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
@@ -20,6 +20,11 @@ jobs:
 
       - name: Build wheels
         uses: pypa/[email protected]
+        env:
+          CIBW_BUILD: cp*
+          CIBW_SKIP: "*-win32 *_i686 *musllinux*"
+          CIBW_TEST_REQUIRES: pytest
+          CIBW_TEST_COMMAND: pytest {package}/tests/test_chatglm_cpp.py
 
       - uses: actions/upload-artifact@v3
         with:

diff --git a/README.md b/README.md
@@ -249,6 +249,8 @@ pip install git+https://github.com/li-plus/chatglm.cpp.git@main
 pip install .
 ```
 
+Pre-built wheels for CPU backend on Linux / MacOS / Windows are published on [release](https://github.com/li-plus/chatglm.cpp/releases). For CUDA / Metal backends, please compile from source code or source distribution.
+
 **Using pre-converted ggml models**
 
 Here is a simple demo that uses `chatglm_cpp.Pipeline` to load the GGML model and chat with it. First enter the examples folder (`cd examples`) and launch a Python interactive shell: